Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Discriminative Region-based Multi-Label Zero-Shot Learning

About

Multi-label zero-shot learning (ZSL) is a more realistic counter-part of standard single-label ZSL since several objects can co-exist in a natural image. However, the occurrence of multiple objects complicates the reasoning and requires region-specific processing of visual features to preserve their contextual cues. We note that the best existing multi-label ZSL method takes a shared approach towards attending to region features with a common set of attention maps for all the classes. Such shared maps lead to diffused attention, which does not discriminatively focus on relevant locations when the number of classes are large. Moreover, mapping spatially-pooled visual features to the class semantics leads to inter-class feature entanglement, thus hampering the classification. Here, we propose an alternate approach towards region-based discriminability-preserving multi-label zero-shot classification. Our approach maintains the spatial resolution to preserve region-level characteristics and utilizes a bi-level attention module (BiAM) to enrich the features by incorporating both region and scene context information. The enriched region-level features are then mapped to the class semantics and only their class predictions are spatially pooled to obtain image-level predictions, thereby keeping the multi-class features disentangled. Our approach sets a new state of the art on two large-scale multi-label zero-shot benchmarks: NUS-WIDE and Open Images. On NUS-WIDE, our approach achieves an absolute gain of 6.9% mAP for ZSL, compared to the best published results.

Sanath Narayan, Akshita Gupta, Salman Khan, Fahad Shahbaz Khan, Ling Shao, Mubarak Shah• 2021

Related benchmarks

TaskDatasetResultRank
Multi-Label ClassificationNUS-WIDE 925/81 (unseen)
mAP (Mean Average Precision)26.3
43
Multi-Label ClassificationNUS-WIDE
mAP47.8
38
Multi-Label ClassificationOpen Images
mAP84.5
24
Multi-Label ClassificationNUS-WIDE 81 unseen labels (test)
mAP0.478
17
Multi-Label ClassificationOpen Images (test)
mAP85
16
Multi-label recognitionNUS-WIDE seen & unseen
F1 Score @ 316.1
10
Remote Sensing Image ClassificationMultiScene-OV GZSL
F1@314.4
7
Remote Sensing Image ClassificationMLRSNet-OV GZSL
F1@313.5
7
Pedestrian Attribute RecognitionRAP-OV (test)
F1@319.3
6
Pedestrian Attribute RecognitionPA100K OV (test)
F1@39.6
6
Showing 10 of 11 rows

Other info

Code

Follow for update