arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2311.04828cs.CV

SODAWideNet——无需ImageNet预训练的注意力增强型宽编码解码网络显著性目标检测

SODAWideNet -- Salient Object Detection with an Attention augmented Wide Encoder Decoder network without ImageNet pre-training

University of Delaware(特拉华大学)

Rohit Venkata Sai Dulam, Chandra Kambhamettu

AI总结:

本研究针对现有SOD模型依赖ImageNet预训练骨干网络、重训练耗时久的问题,提出无需预训练的宽浅编码解码网络SODAWideNet,结合空洞卷积与注意力机制设计相关模块,两种参数规模变体在五个数据集上达到SOTA水平。

摘要:

开发新型显著性目标检测(Salient Object Detection, SOD)模型通常需要选取经ImageNet预训练的骨干网络,并设计全新的特征细化模块以利用骨干网络特征。然而,在预训练骨干网络中添加新组件需要在ImageNet数据集上重新训练整个网络,耗时极长。因此,我们探索直接从零构建神经网络,无需ImageNet预训练,直接在SOD任务上训练,这种模式为设计任务专属组件提供了完全的自主性。为此,我们提出SODAWideNet,一种用于显著性目标检测的编码-解码风格网络。我们摒弃了常用的窄深卷积模型范式,转而采用宽浅架构,构建出参数高效的深度神经网络。为实现更浅的网络,我们从网络初始阶段就结合空洞卷积与自注意力来扩大感受野。据此,我们提出多感受野特征聚合模块(Multi Receptive Field Feature Aggregation Module, MRFFAM),该模块利用空洞卷积在高分辨率下从更远区域高效获取判别性特征。随后,我们提出多尺度注意力(Multi-Scale Attention, MSA),它构建特征金字塔并高效计算多分辨率间的注意力,以从更大的特征图中提取全局特征。最后,我们提出两种变体:SODAWideNet-S(3.03M参数)和SODAWideNet(9.03M参数),在五个数据集上的性能与当前最优模型相当。

英文摘要:

Developing a new Salient Object Detection (SOD) model involves selecting an ImageNet pre-trained backbone and creating novel feature refinement modules to use backbone features. However, adding new components to a pre-trained backbone needs retraining the whole network on the ImageNet dataset, which requires significant time. Hence, we explore developing a neural network from scratch directly trained on SOD without ImageNet pre-training. Such a formulation offers full autonomy to design task-specific components. To that end, we propose SODAWideNet, an encoder-decoder-style network for Salient Object Detection. We deviate from the commonly practiced paradigm of narrow and deep convolutional models to a wide and shallow architecture, resulting in a parameter-efficient deep neural network. To achieve a shallower network, we increase the receptive field from the beginning of the network using a combination of dilated convolutions and self-attention. Therefore, we propose Multi Receptive Field Feature Aggregation Module (MRFFAM) that efficiently obtains discriminative features from farther regions at higher resolutions using dilated convolutions. Next, we propose Multi-Scale Attention (MSA), which creates a feature pyramid and efficiently computes attention across multiple resolutions to extract global features from larger feature maps. Finally, we propose two variants, SODAWideNet-S (3.03M) and SODAWideNet (9.03M), that achieve competitive performance against state-of-the-art models on five datasets.

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

↑