AP-Loss for Accurate One-Stage Object Detection


Chen, Kean and Lin, Weiyao and Li, Jianguo and See, John Su Yang and Wang, Ji and Zou, Junni (2021) AP-Loss for Accurate One-Stage Object Detection. IEEE Transactions on Pattern Analysis and Machine Intelligence, 43 (11). pp. 3782-3798. ISSN 0162-8828

Full text not available from this repository.


One-stage object detectors are trained by optimizing classification-loss and localization-loss simultaneously, with the former suffering much from extreme foreground-background class imbalance issue due to the large number of anchors. This paper alleviates this issue by proposing a novel framework to replace the classification task in one-stage detectors with a ranking task, and adopting the average-precision loss (AP-loss) for the ranking problem. Due to its non-differentiability and non-convexity, the AP-loss cannot be optimized directly. For this purpose, we develop a novel optimization algorithm, which seamlessly combines the error-driven update scheme in perceptron learning and backpropagation algorithm in deep networks. We provide in-depth analyses on the good convergence property and computational complexity of the proposed algorithm, both theoretically and empirically. Experimental results demonstrate notable improvement in addressing the imbalance issue in object detection over existing AP-based optimization algorithms. An improved state-of-the-art performance is achieved in one-stage detectors based on AP-loss over detectors using classification-losses on various standard benchmarks. The proposed framework is also highly versatile in accommodating different network architectures.

Item Type: Article
Uncontrolled Keywords: Detectors, Task Analysis, Measurement, Optimization, Object Detection, Training, Proposals, Computer Vision, Object Detection, Machine Learning, Ranking Loss
Subjects: T Technology > TK Electrical engineering. Electronics Nuclear engineering > TK7800-8360 Electronics
Divisions: Faculty of Computing and Informatics (FCI)
Depositing User: Ms Nurul Iqtiani Ahmad
Date Deposited: 28 Oct 2021 23:48
Last Modified: 28 Oct 2021 23:48
URII: http://shdl.mmu.edu.my/id/eprint/9725


Downloads per month over past year

View ItemEdit (login required)