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

面向跨模态目标检测的 Fusion-Mamba

Fusion-Mamba for Cross-modality Object Detection

Beihang University(北航大学);East China Normal University(东华师范大学);Tencent Youtu Lab(腾讯优图实验室);Eastern Institute of Technology(东部技术研究所)

Wenhao Dong, Haodong Zhu, Shaohui Lin, Xiaoyan Luo, Yunhang Shen, Xuhui Liu, Juan Zhang, Guodong Guo, Baochang Zhang

AI总结:

本文提出 Fusion-Mamba 块,利用改进的 Mamba 在隐状态空间中关联跨模态特征以减少模态差异,并在多个数据集上取得领先的目标检测性能。

摘要:

跨模态融合来自不同模态的互补信息能够有效提升目标检测性能,使其在更广泛的应用中更加实用和鲁棒。现有的融合策略通过精心设计的神经网络模块来组合不同类型的图像或合并不同的骨干特征。然而,这些方法忽略了模态差异会影响跨模态融合性能,因为具有不同相机焦距、安装位置和角度的不同模态很难被融合。在本文中,我们通过基于带有门控机制的改进 Mamba,在隐状态空间中关联跨模态特征来研究跨模态融合。我们设计了一个 Fusion-Mamba 块(FMB),将跨模态特征映射到隐状态空间中进行交互,从而减少跨模态特征之间的差异并增强融合特征表示的一致性。FMB 包含两个模块:状态空间通道交换(SSCS)模块促进浅层特征融合,双状态空间融合(DSSF)在隐状态空间中实现深度融合。通过在公开数据集上的大量实验,我们提出的方法在 $m$AP 指标上优于最先进的方法,在 $M^3FD$ 上提升了 5.9%,在 FLIR-Aligned 数据集上提升了 4.9%,展示了优越的目标检测性能。据我们所知,这是首个探索 Mamba 在跨模态融合中潜力的工作,并为跨模态目标检测建立了新的基线。

英文摘要:

Cross-modality fusing complementary information from different modalities effectively improves object detection performance, making it more useful and robust for a wider range of applications. Existing fusion strategies combine different types of images or merge different backbone features through elaborated neural network modules. However, these methods neglect that modality disparities affect cross-modality fusion performance, as different modalities with different camera focal lengths, placements, and angles are hardly fused. In this paper, we investigate cross-modality fusion by associating cross-modal features in a hidden state space based on an improved Mamba with a gating mechanism. We design a Fusion-Mamba block (FMB) to map cross-modal features into a hidden state space for interaction, thereby reducing disparities between cross-modal features and enhancing the representation consistency of fused features. FMB contains two modules: the State Space Channel Swapping (SSCS) module facilitates shallow feature fusion, and the Dual State Space Fusion (DSSF) enables deep fusion in a hidden state space. Through extensive experiments on public datasets, our proposed approach outperforms the state-of-the-art methods on $m$AP with 5.9% on $M^3FD$ and 4.9% on FLIR-Aligned datasets, demonstrating superior object detection performance. To the best of our knowledge, this is the first work to explore the potential of Mamba for cross-modal fusion and establish a new baseline for cross-modality object detection.

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

↑