arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.16645cs.CV

SODAWideNet++:结合注意力与卷积的显著性目标检测

SODAWideNet++: Combining Attention and Convolutions for Salient Object Detection

University of Delaware(特拉华大学)

Rohit Venkata Sai Dulam, Chandra Kambhamettu

AI总结:

提出专为显著性目标检测设计的SODAWideNet++网络,结合注意力与大空洞卷积提取长距离特征,并利用COCO数据集二值化预训练,在五个数据集上表现优异且参数量仅为主流模型的35%。

摘要:

显著性目标检测(SOD)传统上依赖于利用ImageNet预训练骨干网络特征的模块。然而,由于SOD与图像分类的本质差异,这种方法限制了预训练整个网络的可能性。此外,这些最初为图像分类构建的骨干网络架构对于像SOD这样的密集预测任务来说是次优的。为了解决这些问题,我们提出了一种名为SODAWideNet++的新型编码器-解码器风格神经网络,专为SOD设计。受视觉Transformer在初始阶段即可获得全局感受野能力的启发,我们引入了注意力引导长距离特征提取(AGLRFE)模块,该模块结合了大空洞卷积和自注意力机制。具体而言,我们使用注意力特征来引导由多个空洞卷积提取的长距离信息,从而利用卷积操作的归纳偏置和自注意力带来的输入依赖性。与当前ImageNet预训练范式不同,我们通过对COCO语义分割数据集中的118K张标注图像进行二值化处理来修改它们,以端到端地预训练所提出的模型。此外,我们连同前景一起监督背景预测,以推动模型生成准确的显著性预测。SODAWideNet++在五个不同的数据集上表现出竞争力,同时与最先进的模型相比仅包含35%的可训练参数。代码和预计算的显著性图可在https://github.com/VimsLab/SODAWideNetPlusPlus获取。

英文摘要:

Salient Object Detection (SOD) has traditionally relied on feature refinement modules that utilize the features of an ImageNet pre-trained backbone. However, this approach limits the possibility of pre-training the entire network because of the distinct nature of SOD and image classification. Additionally, the architecture of these backbones originally built for Image classification is sub-optimal for a dense prediction task like SOD. To address these issues, we propose a novel encoder-decoder-style neural network called SODAWideNet++ that is designed explicitly for SOD. Inspired by the vision transformers ability to attain a global receptive field from the initial stages, we introduce the Attention Guided Long Range Feature Extraction (AGLRFE) module, which combines large dilated convolutions and self-attention. Specifically, we use attention features to guide long-range information extracted by multiple dilated convolutions, thus taking advantage of the inductive biases of a convolution operation and the input dependency brought by self-attention. In contrast to the current paradigm of ImageNet pre-training, we modify 118K annotated images from the COCO semantic segmentation dataset by binarizing the annotations to pre-train the proposed model end-to-end. Further, we supervise the background predictions along with the foreground to push our model to generate accurate saliency predictions. SODAWideNet++ performs competitively on five different datasets while only containing 35% of the trainable parameters compared to the state-of-the-art models. The code and pre-computed saliency maps are provided at https://github.com/VimsLab/SODAWideNetPlusPlus.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑