Zero-Shot Aerial Object Detection with Visual Description Regularization

Authors

  • Zhengqing Zang College of Computer Science, Sichuan University, Chengdu, 610065, P. R. China Engineering Research Center of Machine Learning and Industry Intelligence, Ministry of Education, Chengdu, 610065, P. R. China
  • Chenyu Lin College of Computer Science, Sichuan University, Chengdu, 610065, P. R. China Engineering Research Center of Machine Learning and Industry Intelligence, Ministry of Education, Chengdu, 610065, P. R. China
  • Chenwei Tang College of Computer Science, Sichuan University, Chengdu, 610065, P. R. China Engineering Research Center of Machine Learning and Industry Intelligence, Ministry of Education, Chengdu, 610065, P. R. China
  • Tao Wang College of Computer Science, Sichuan University, Chengdu, 610065, P. R. China Engineering Research Center of Machine Learning and Industry Intelligence, Ministry of Education, Chengdu, 610065, P. R. China
  • Jiancheng Lv College of Computer Science, Sichuan University, Chengdu, 610065, P. R. China Engineering Research Center of Machine Learning and Industry Intelligence, Ministry of Education, Chengdu, 610065, P. R. China

DOI:

https://doi.org/10.1609/aaai.v38i7.28518

Keywords:

CV: Object Detection & Categorization, ML: Transfer, Domain Adaptation, Multi-Task Learning

Abstract

Existing object detection models are mainly trained on large-scale labeled datasets. However, annotating data for novel aerial object classes is expensive since it is time-consuming and may require expert knowledge. Thus, it is desirable to study label-efficient object detection methods on aerial images. In this work, we propose a zero-shot method for aerial object detection named visual Description Regularization, or DescReg. Concretely, we identify the weak semantic-visual correlation of the aerial objects and aim to address the challenge with prior descriptions of their visual appearance. Instead of directly encoding the descriptions into class embedding space which suffers from the representation gap problem, we propose to infuse the prior inter-class visual similarity conveyed in the descriptions into the embedding learning. The infusion process is accomplished with a newly designed similarity-aware triplet loss which incorporates structured regularization on the representation space. We conduct extensive experiments with three challenging aerial object detection datasets, including DIOR, xView, and DOTA. The results demonstrate that DescReg significantly outperforms the state-of-the-art ZSD methods with complex projection designs and generative frameworks, e.g., DescReg outperforms best reported ZSD method on DIOR by 4.5 mAP on unseen classes and 8.1 in HM. We further show the generalizability of DescReg by integrating it into generative ZSD methods as well as varying the detection architecture. Codes will be released at https://github.com/zq-zang/DescReg.

Published

2024-03-24

How to Cite

Zang, Z., Lin, C., Tang, C., Wang, T., & Lv, J. (2024). Zero-Shot Aerial Object Detection with Visual Description Regularization. Proceedings of the AAAI Conference on Artificial Intelligence, 38(7), 6926-6934. https://doi.org/10.1609/aaai.v38i7.28518

Issue

Section

AAAI Technical Track on Computer Vision VI