Command Palette
Search for a command to run...
Hierarchical Attention Network for Few-Shot Object Detection via Meta-Contrastive Learning
Dongwoo Park Jong-Min Lee

Abstract
Few-shot object detection (FSOD) aims to classify and detect few images of novel categories. Existing meta-learning methods insufficiently exploit features between support and query images owing to structural limitations. We propose a hierarchical attention network with sequentially large receptive fields to fully exploit the query and support images. In addition, meta-learning does not distinguish the categories well because it determines whether the support and query images match. In other words, metric-based learning for classification is ineffective because it does not work directly. Thus, we propose a contrastive learning method called meta-contrastive learning, which directly helps achieve the purpose of the meta-learning strategy. Finally, we establish a new state-of-the-art network, by realizing significant margins. Our method brings 2.3, 1.0, 1.3, 3.4 and 2.4% AP improvements for 1-30 shots object detection on COCO dataset. Our code is available at: https://github.com/infinity7428/hANMCL
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| few-shot-object-detection-on-ms-coco-1-shot | hANMCL | AP: 13.4 |
| few-shot-object-detection-on-ms-coco-10-shot | hANMCL | AP: 22.4 |
| few-shot-object-detection-on-ms-coco-30-shot | hANMCL | AP: 25.0 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.