Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Strengthening Layer Interaction via Dynamic Layer Attention

About

In recent years, employing layer attention to enhance interaction among hierarchical layers has proven to be a significant advancement in building network structures. In this paper, we delve into the distinction between layer attention and the general attention mechanism, noting that existing layer attention methods achieve layer interaction on fixed feature maps in a static manner. These static layer attention methods limit the ability for context feature extraction among layers. To restore the dynamic context representation capability of the attention mechanism, we propose a Dynamic Layer Attention (DLA) architecture. The DLA comprises dual paths, where the forward path utilizes an improved recurrent neural network block, named Dynamic Sharing Unit (DSU), for context feature extraction. The backward path updates features using these shared context representations. Finally, the attention mechanism is applied to these dynamically refreshed feature maps among layers. Experimental results demonstrate the effectiveness of the proposed DLA architecture, outperforming other state-of-the-art methods in image recognition and object detection tasks. Additionally, the DSU block has been evaluated as an efficient plugin in the proposed DLA architecture.The code is available at https://github.com/tunantu/Dynamic-Layer-Attention.

Kaishen Wang, Xun Xia, Jian Liu, Zhang Yi, Tao He• 2024

Related benchmarks

TaskDatasetResultRank
Object DetectionCOCO 2017 (val)--
2930
Image ClassificationCIFAR-100--
362
Object DetectionCOCO 2017
AP (Box)40.9
348
Medical Image SegmentationKvasir-SEG (test)
mIoU76.33
139
Medical Image SegmentationISIC 2018 (test)
Dice Score89.05
83
Image ClassificationImageNet-1K 1.0 (val)
Top-1 Acc78.9
51
Medical Image SegmentationISIC 2017 (test)
Dice88.24
50
Image ClassificationCIFAR-100
Top-1 Accuracy76.82
28
Showing 8 of 8 rows

Other info

Follow for update