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

ComPtr:通过简单而通用的互补 Transformer 处理多样化双源密集预测任务

ComPtr: Towards Diverse Bi-source Dense Prediction Tasks via A Simple yet General Complementary Transformer

Dalian University of Technology(大连理工大学)

Youwei Pang, Xiaoqi Zhao, Lihe Zhang, Huchuan Lu

AI总结:

本文提出通用互补 Transformer ComPtr,通过一致性增强与差异感知模块建模双源信息,并在多类密集预测任务中验证了统一架构的有效性。

摘要:

深度学习(DL)推动了密集预测领域的发展,同时也逐渐消解了不同任务之间固有的壁垒。然而,现有大多数工作只关注为特定任务设计架构和构建视觉线索,忽视了 DL 范式所带来的潜在统一性。本文尝试构建一种新颖的互补 Transformer(Complementary transformer),即 ComPtr,用于多样化的双源密集预测任务。具体而言,不同于现有方法过度专精于单一任务或任务子集,ComPtr 从更通用的双源密集预测概念出发。基于对信息互补性的基本依赖,我们提出了一致性增强和差异感知组件,使 ComPtr 能够分别从不同图像源中疏散和收集用于多样化任务的重要视觉语义线索。ComPtr 平等对待不同输入,并在 transformer 之上以序列到序列的形式构建高效的密集交互模型。这种任务通用设计为构建能够同时处理各种双源信息的统一模型提供了顺畅基础。在若干代表性视觉任务(即遥感变化检测、RGB-T 人群计数、RGB-D/T 显著目标检测和 RGB-D 语义分割)的大量实验中,所提方法持续取得良好性能。代码将在 https://github.com/lartpang/ComPtr 提供。

英文摘要:

Deep learning (DL) has advanced the field of dense prediction, while gradually dissolving the inherent barriers between different tasks. However, most existing works focus on designing architectures and constructing visual cues only for the specific task, which ignores the potential uniformity introduced by the DL paradigm. In this paper, we attempt to construct a novel $\underline{ComP}$lementary $\underline{tr}$ansformer, $\textbf{ComPtr}$, for diverse bi-source dense prediction tasks. Specifically, unlike existing methods that over-specialize in a single task or a subset of tasks, ComPtr starts from the more general concept of bi-source dense prediction. Based on the basic dependence on information complementarity, we propose consistency enhancement and difference awareness components with which ComPtr can evacuate and collect important visual semantic cues from different image sources for diverse tasks, respectively. ComPtr treats different inputs equally and builds an efficient dense interaction model in the form of sequence-to-sequence on top of the transformer. This task-generic design provides a smooth foundation for constructing the unified model that can simultaneously deal with various bi-source information. In extensive experiments across several representative vision tasks, i.e. remote sensing change detection, RGB-T crowd counting, RGB-D/T salient object detection, and RGB-D semantic segmentation, the proposed method consistently obtains favorable performance. The code will be available at https://github.com/lartpang/ComPtr.

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

↑