Adaptive Important Region Selection with Reinforced Hierarchical Search for Dense Object Detection
Dingrong Wang, Hitesh Sapkota, Qi Yu
Abstract
Existing state-of-the-art dense object detection techniques tend to produce a large number of false positive detections on difficult images with complex scenes because they focus on ensuring a high recall. To improve the detection accuracy, we propose an Adaptive Important Region Selection ( AIRS ) framework guided by Evidential Q-learning coupled with a uniquely designed reward function. Inspired by human visual attention, our detection model conducts object search in a top-down, hierarchical fashion. It starts from the top of the hierarchy with the coarsest granularity and then identifies the potential patches likely to contain objects of interest. It then discards non-informative patches and progressively moves downward on the selected ones for a fine-grained search. The proposed evidential Q-learning systematically encodes epistemic uncertainty in its evidential-Q value to encourage the exploration of unknown patches, especially in the early phase of model training. In this way, the proposed model dynamically balances exploration-exploitation to cover both highly valuable and informative patches. Theoretical analysis and extensive experiments on multiple datasets demonstrate that our proposed framework outperforms the SOTA models.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext fa1f926b-2eeb-447d-a222-5b688384bf3dBuilds on14
- Swin Transformer: Hierarchical Vision Transformer using Shifted WindowsZe Liu, Yutong Lin, Yue Cao, Han Hu et al.ICCV 2021 · 31,683 citations
- A ConvNet for the 2020sZhuang Liu, Hanzi Mao, Chao-Yuan Wu, Christoph Feichtenhofer et al.CVPR 2022 · 6,782 citations
- FCOS: Fully Convolutional One-Stage Object DetectionZhi Tian, Chunhua Shen, Hao Chen, Tong HeICCV 2019 · 6,042 citations
- Distance-IoU Loss: Faster and Better Learning for Bounding Box RegressionZhaohui Zheng, Ping Wang, Wei Liu, Jinze Li et al.AAAI 2020 · 4,823 citations
- Generalized Focal Loss: Learning Qualified and Distributed Bounding Boxes for Dense Object DetectionXiang Li, Wenhai Wang, Lijun Wu, Shuo Chen et al.NeurIPS 2020 · 2,118 citations
Related papers
- Active Learning for Object Detection with Evidential Deep Learning and Hierarchical Uncertainty AggregationYounghyun Park, Wonjeong Choi, Soyeong Kim, Dong-Jun Han et al.ICLR 2023
- Progressive Exploration-Conformal Learning for Sparsely Annotated Object Detection in Aerial ImagesZihan Lu, Chenxu Wang, Chunyan Xu, Xiangwei Zheng et al.NeurIPS 2024 · 6 citations
- A Partially-Supervised Reinforcement Learning Framework for Visual Active SearchAnindya Sarkar, Nathan Jacobs, Yevgeniy VorobeychikNeurIPS 2023 · 13 citations
- Reinforced active learning for image segmentationArantxa Casanova, Pedro O. Pinheiro, Negar Rostamzadeh, Christopher J. PalICLR 2020 · 127 citations
- IQDet: Instance-Wise Quality Distribution Sampling for Object DetectionYuchen Ma, Songtao Liu, Zeming Li, Jian SunCVPR 2021
