LDA-AQU: Adaptive Query-guided Upsampling via Local Deformable Attention

TL;DR

LDA-AQU introduces a novel adaptive feature upsampling method using local self-attention and deformation mechanisms, significantly improving performance across multiple dense prediction tasks.

Contribution

The paper proposes LDA-AQU, a lightweight, dynamic kernel-based upsampler leveraging local self-attention and deformation for enhanced feature reassembly in neural networks.

Findings

Outperforms previous state-of-the-art upsamplers in four dense prediction tasks.

Achieves performance improvements of 1.7 AP, 1.5 AP, 2.0 PQ, and 2.5 mIoU.

Easily integrable into various model architectures.

Abstract

Feature upsampling is an essential operation in constructing deep convolutional neural networks. However, existing upsamplers either lack specific feature guidance or necessitate the utilization of high-resolution feature maps, resulting in a loss of performance and flexibility. In this paper, we find that the local self-attention naturally has the feature guidance capability, and its computational paradigm aligns closely with the essence of feature upsampling (\ie feature reassembly of neighboring points). Therefore, we introduce local self-attention into the upsampling task and demonstrate that the majority of existing upsamplers can be regarded as special cases of upsamplers based on local self-attention. Considering the potential semantic gap between upsampled points and their neighboring points, we further introduce the deformation mechanism into the upsampler based on local self-attention, thereby proposing LDA-AQU. As a novel dynamic kernel-based upsampler, LDA-AQU utilizes the feature of queries to guide the model in adaptively adjusting the position and aggregation weight of neighboring points, thereby meeting the upsampling requirements across various complex scenarios. In addition, LDA-AQU is lightweight and can be easily integrated into various model architectures. We evaluate the effectiveness of LDA-AQU across four dense prediction tasks: object detection, instance segmentation, panoptic segmentation, and semantic segmentation. LDA-AQU consistently outperforms previous state-of-the-art upsamplers, achieving performance enhancements of 1.7 AP, 1.5 AP, 2.0 PQ, and 2.5 mIoU compared to the baseline models in the aforementioned four tasks, respectively. Code is available at \url{https://github.com/duzw9311/LDA-AQU}.