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

利用基于 Transformer 的非对称双边 U-Net 促进显著性目标检测

Boosting Salient Object Detection with Transformer-based Asymmetric Bilateral U-Net

Nankai University(南开大学);Agency for Science, Technology and Research (A*STAR)(新加坡科技研究局);Institute for Infocomm Research (I2R)(资讯通信研究院);University of Electronic Science and Technology of China (UESTC)(电子科技大学)

Yu Qiu, Yun Liu, Le Zhang, Jing Xu

AI总结:

本文提出 ABiU-Net,通过 Transformer 与轻量级 CNN 的非对称双边编解码路径及阶段交互,融合全局上下文与局部细节,提升显著性目标检测性能。

摘要:

现有的显著性目标检测(SOD)方法主要依赖带有跳跃连接的 U 形卷积神经网络(CNN),以结合全局上下文和局部空间细节;这两类信息分别对定位显著性目标和细化目标细节至关重要。尽管取得了巨大成功,CNN 学习全局上下文的能力仍然有限。近年来,视觉 Transformer 凭借其对全局依赖关系的强大建模能力,在计算机视觉领域取得了革命性进展。然而,将 Transformer 直接应用于 SOD 并非最优,因为 Transformer 缺乏学习局部空间表示的能力。为此,本文探索结合 Transformer 与 CNN,从而为 SOD 同时学习全局和局部表示。我们提出了一种基于 Transformer 的非对称双边 U-Net(ABiU-Net)。非对称双边编码器包含一条 Transformer 路径和一条轻量级 CNN 路径,两条路径在编码器的每个阶段进行通信,以分别学习互补的全局上下文和局部空间细节。非对称双边解码器同样由两条路径组成,用于处理来自 Transformer 和 CNN 编码器路径的特征,并在解码器的每个阶段进行通信,以分别解码粗略的显著性目标位置和细粒度的目标细节。编码器/解码器两条路径之间的这种通信使 ABiU-Net 能够学习互补的全局和局部表示,并分别利用 Transformer 和 CNN 的天然优势。因此,ABiU-Net 为基于 Transformer 的 SOD 提供了新的视角。大量实验表明,ABiU-Net 的性能优于以往的先进 SOD 方法。代码可在 https://github.com/yuqiuyuqiu/ABiU-Net 获取。

英文摘要:

Existing salient object detection (SOD) methods mainly rely on U-shaped convolution neural networks (CNNs) with skip connections to combine the global contexts and local spatial details that are crucial for locating salient objects and refining object details, respectively. Despite great successes, the ability of CNNs in learning global contexts is limited. Recently, the vision transformer has achieved revolutionary progress in computer vision owing to its powerful modeling of global dependencies. However, directly applying the transformer to SOD is suboptimal because the transformer lacks the ability to learn local spatial representations. To this end, this paper explores the combination of transformers and CNNs to learn both global and local representations for SOD. We propose a transformer-based Asymmetric Bilateral U-Net (ABiU-Net). The asymmetric bilateral encoder has a transformer path and a lightweight CNN path, where the two paths communicate at each encoder stage to learn complementary global contexts and local spatial details, respectively. The asymmetric bilateral decoder also consists of two paths to process features from the transformer and CNN encoder paths, with communication at each decoder stage for decoding coarse salient object locations and fine-grained object details, respectively. Such communication between the two encoder/decoder paths enables AbiU-Net to learn complementary global and local representations, taking advantage of the natural merits of transformers and CNNs, respectively. Hence, ABiU-Net provides a new perspective for transformer-based SOD. Extensive experiments demonstrate that ABiU-Net performs favorably against previous state-of-the-art SOD methods. The code is available at https://github.com/yuqiuyuqiu/ABiU-Net.

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

↑