跳到主要导航 跳到搜索 跳到主要内容

Learning 1-Bit Tiny Object Detector with Discriminative Feature Refinement

  • Sheng Xu
  • , Mingze Wang
  • , Yanjing Li
  • , Mingbao Lin
  • , Baochang Zhang*
  • , David Doermann
  • , Xiao Sun
  • *此作品的通讯作者
  • Beihang University
  • Skywork AI
  • Zhongguancun Laboratory
  • SUNY Buffalo
  • Shanghai Artificial Intelligence Laboratory

科研成果: 期刊稿件会议文章同行评审

摘要

1-bit detectors show impressive performance comparable to their real-valued counterparts when detecting commonly sized objects while exhibiting significant performance degradation on tiny objects. The challenge stems from the fact that high-level features extracted by 1-bit convolutions seem less compelling to reveal the discriminative foreground features. To address these issues, we introduce a Discriminative Feature Refinement method for 1-bit Detectors (DFR-Det), aiming to enhance the discriminative ability of foreground representation for tiny objects in aerial images. This is accomplished by refining the feature representation using an information bottleneck (IB) to achieve a distinctive representation of tiny objects. Specifically, we introduce a new decoder with a foreground mask, aiming to enhance the discriminative ability of high-level features for the target but suppress the background impact. Additionally, our decoder is simple but effective and can be easily mounted on existing detectors without extra burden added to the inference procedure. Extensive experiments on various tiny object detection (TOD) tasks demonstrate DFR-Det's superiority over state-of-the-art 1-bit detectors. For example, 1-bit FCOS achieved by DFR-Det achieves the 12.8% AP on AI-TOD dataset, approaching the performance of the real-valued counterpart.

源语言英语
页(从-至)55337-55347
页数11
期刊Proceedings of Machine Learning Research
235
出版状态已出版 - 2024
活动41st International Conference on Machine Learning, ICML 2024 - Vienna, 奥地利
期限: 21 7月 202427 7月 2024

学术指纹

探究 'Learning 1-Bit Tiny Object Detector with Discriminative Feature Refinement' 的科研主题。它们共同构成独一无二的学术指纹。

引用此