arrow
返回

Sparse attention block: Aggregating contextual information for object detection

delete2022-04-01
delete8
PRE
AI
C
Chunlin Chen
J
Jun Yu
Q
Qiang Ling *
DOI:10.1016/j.patcog.2021.108418delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
It is well recognized that the contextual information of surrounding objects is beneficial for object detection. Such contextual information can often be obtained from long-range dependencies. This paper proposes a sparse attention block to capture long-range dependencies in an efficient way. Unlike the conventional non-local block, which generates a dense attention map to characterize the dependency between any two positions of the input feature map, our sparse attention block samples the most representative positions for contextual information aggregation. After searching for local peaks in a heat map of the given input feature map, it adaptively selects a sparse set of positions to represent the relationship between query and key elements. With the obtained sparse positions, our sparse attention block can well model long-range dependencies, and greatly improve the object detection performance at the additional cost of < 2% GPU memory and computation of the conventional non-local block. This sparse attention block can be easily plugged into various object detection frameworks, such as Faster R-CNN, RetinaNet and Mask R-CNN. Experiments on COCO benchmark confirm that our sparse attention block can boost the detection accuracy with significant gains ranging from 1.4% to 1.9% and negligible overhead of computation and memory usage.
Keyword:
Object detection
Self-attention
Convolution neural network

期刊

Pattern Recognition 封面图
Pattern Recognition
IF:
7.6
论文数:
1.3W
被引数:
4.5W

机构

C
chinese academy of sciences
学者数:
56.7W
论文数: 45.0W
被引数: 704
引用论文

引用论文

Comparative analysis of occlusion methods for artificial sphincters人工括约肌闭塞方法的比较分析
err2020-04-07
err0
PREAI
errLeonardo Marziale; Gioia Lucarini; Tommaso Mazzocchi; Leonardo Ricotti; Arianna Menciassi
err分享
err收藏
Multi-model ensemble with rich spatial information for object detection
err2020-03-01
err63
PREAI
errXu, Jie; Wang, Wei; Wang, Hanyuan; Guo, Jinhong
err分享
err收藏
SERVICE-ORIENTED COMPUTING: A RESEARCH ROADMAP
err2012-01-25
err0
errOAAI
errMICHAEL P. PAPAZOGLOU; PAOLO TRAVERSO; SCHAHRAM DUSTDAR; FRANK LEYMANN
err分享
err收藏
err分享
err收藏
err
IF0
err
err0
PREAI
err
err分享
err收藏
学者 查看更多内容