Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

LCTR: On Awakening the Local Continuity of Transformer for Weakly Supervised Object Localization

About

Weakly supervised object localization (WSOL) aims to learn object localizer solely by using image-level labels. The convolution neural network (CNN) based techniques often result in highlighting the most discriminative part of objects while ignoring the entire object extent. Recently, the transformer architecture has been deployed to WSOL to capture the long-range feature dependencies with self-attention mechanism and multilayer perceptron structure. Nevertheless, transformers lack the locality inductive bias inherent to CNNs and therefore may deteriorate local feature details in WSOL. In this paper, we propose a novel framework built upon the transformer, termed LCTR (Local Continuity TRansformer), which targets at enhancing the local perception capability of global features among long-range feature dependencies. To this end, we propose a relational patch-attention module (RPAM), which considers cross-patch information on a global basis. We further design a cue digging module (CDM), which utilizes local features to guide the learning trend of the model for highlighting the weak local responses. Finally, comprehensive experiments are carried out on two widely used datasets, ie, CUB-200-2011 and ILSVRC, to verify the effectiveness of our method.

Zhiwei Chen, Changan Wang, Yabiao Wang, Guannan Jiang, Yunhang Shen, Ying Tai, Chengjie Wang, Wei Zhang, Liujuan Cao• 2021

Related benchmarks

TaskDatasetResultRank
Weakly Supervised Object LocalizationCUB (test)
Top-1 Loc Acc79.2
80
Object LocalizationImageNet-1k (val)
Top-1 Loc Acc56.1
80
Object LocalizationCUB-200-2011 (test)
Top-1 Loc. Accuracy79.2
68
Image ClassificationCUB (test)
Top-1 Accuracy85
31
Weakly Supervised Object LocalizationILSVRC (test)
Top-1 Loc Acc56.1
14
Weakly Supervised Object LocalizationImageNet-1k (val)
Top-1 Loc Acc56.1
10
Image ClassificationILSVRC (test)
Top-1 Accuracy77.1
9
Showing 7 of 7 rows

Other info

Follow for update