Comprehensive Attention Self-Distillation for Weakly-Supervised Object Detection

December 2020

Comprehensive Attention Self-Distillation for Weakly-Supervised Object Detection

Authors:

Zeyi Huang, Yang Zou, Vijayakumar Bhagavatula, and Dong Huang
Conference Paper
Proceedings of (NeurIPS) Neural Information Processing Systems

Abstract:

Weakly Supervised Object Detection (WSOD) has emerged as an effective tool to train object detectors using only the image-level category labels. However, without object-level labels, WSOD detectors are prone to detect bounding boxes on salient objects, clustered objects and discriminative object parts. Moreover, the image-level category labels do not enforce consistent object detection across different transformations of the same images. To address the above issues, we propose a Comprehensive Attention Self-Distillation (CASD) training approach for WSOD. To balance feature learning among all object instances, CASD computes the comprehensive attention aggregated from multiple transformations and feature layers of the same images. To enforce consistent spatial supervision on objects, CASD conducts self-distillation on the WSOD networks, such that the comprehensive attention is approximated simultaneously by multiple transformations and feature layers of the same images. CASD produces new state-of-the-art WSOD results on standard benchmarks such as PASCAL VOC 2007/2012 and MS-COCO.
@conference{Huang and Huang-2020-125332,
author = {Zeyi Huang And Yang Zou And Vijayakumar Bhagavatula And Dong Huang},
title = {Comprehensive Attention Self-Distillation for Weakly-Supervised Object Detection},
booktitle = {Proceedings of (NeurIPS) Neural Information Processing Systems},
year = {2020},
month = {December},
keywords = {Weakly Supervised Learning, Object Detection, Attention-based Training},
}
Copyright notice: This material is presented to ensure timely dissemination of scholarly and technical work. Copyright and all rights therein are retained by authors or by other copyright holders. All persons copying this information are expected to adhere to the terms and constraints invoked by each author's copyright. These works may not be reposted without the explicit permission of the copyright holder.