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

Polarity Loss for Zero-shot Object Detection

About

Conventional object detection models require large amounts of training data. In comparison, humans can recognize previously unseen objects by merely knowing their semantic description. To mimic similar behaviour, zero-shot object detection aims to recognize and localize 'unseen' object instances by using only their semantic information. The model is first trained to learn the relationships between visual and semantic domains for seen objects, later transferring the acquired knowledge to totally unseen objects. This setting gives rise to the need for correct alignment between visual and semantic concepts, so that the unseen objects can be identified using only their semantic attributes. In this paper, we propose a novel loss function called 'Polarity loss', that promotes correct visual-semantic alignment for an improved zero-shot object detection. On one hand, it refines the noisy semantic embeddings via metric learning on a 'Semantic vocabulary' of related concepts to establish a better synergy between visual and semantic domains. On the other hand, it explicitly maximizes the gap between positive and negative predictions to achieve better discrimination between seen, unseen and background objects. Our approach is inspired by embodiment theories in cognitive science, that claim human semantic understanding to be grounded in past experiences (seen objects), related linguistic concepts (word vocabulary) and visual perception (seen/unseen object images). We conduct extensive evaluations on MS-COCO and Pascal VOC datasets, showing significant improvements over state of the art.

Shafin Rahman, Salman Khan, Nick Barnes• 2018

Related benchmarks

TaskDatasetResultRank
Object DetectionPASCAL VOC 2007 (test)--
821
Zero-shot Object DetectionMS-COCO (65/15)
mAP (IoU=0.5)12.4
20
Object DetectionMS-COCO (48/17)
Recall@100 (IoU=0.5)43.5
19
Generalized Zero-Shot DetectionMS-COCO (65/15)
Seen mAP34.07
16
Generalized Zero-Shot Object DetectionPASCAL VOC (Seen+Unseen)
mAP (Seen)63.5
11
Object DetectionMS-COCO (65/15)
Recall@100 (IoU=0.5)37.7
11
Object DetectionMSCOCO (48/17)--
11
Object DetectionMS-COCO (48/17)
Recall@100 (S)38.2
7
Zero-shot Object DetectionPascal-VOC (Unseen)
mAP62.1
6
Object DetectionMS COCO ZSD
mAP12.4
6
Showing 10 of 15 rows

Other info

Code

Follow for update