References & Citations
Computer Science > Computer Vision and Pattern Recognition
Title: Condensing Two-stage Detection with Automatic Object Key Part Discovery
(Submitted on 10 Jun 2020 (v1), last revised 13 Aug 2020 (this version, v3))
Abstract: Modern two-stage object detectors generally require excessively large models for their detection heads to achieve high accuracy. To address this problem, we propose that the model parameters of two-stage detection heads can be condensed and reduced by concentrating on object key parts. To this end, we first introduce an automatic object key part discovery task to make neural networks discover representative sub-parts in each foreground object. With these discovered key parts, we then decompose the object appearance modeling into a key part modeling process and a global modeling process for detection. Key part modeling encodes fine and detailed features from the discovered key parts, and global modeling encodes rough and holistic object characteristics. In practice, such decomposition allows us to significantly abridge model parameters without sacrificing much detection accuracy. Experiments on popular datasets illustrate that our proposed technique consistently maintains original performance while waiving around 50% of the model parameters of common two-stage detection heads, with the performance only deteriorating by 1.5% when waiving around 96% of the original model parameters. Codes are released on: this https URL
Submission history
From: Zhe Chen [view email][v1] Wed, 10 Jun 2020 01:20:47 GMT (1399kb,D)
[v2] Tue, 23 Jun 2020 05:18:56 GMT (2061kb,D)
[v3] Thu, 13 Aug 2020 01:08:50 GMT (2062kb,D)
Link back to: arXiv, form interface, contact.