用最优传输与最大均值差异提升Mamba的跨模态对齐能力
AlignMamba: Enhancing Multimodal Mamba with Local and Global Cross-modal Alignment
- 基于最优传输设计局部对齐模块,学习模态间词元对应关系
- 引入最大均值差异损失实现全局分布一致性约束
- 在长序列数据上兼具高效性与强融合效果,适合多模态任务
跨模态对齐对于克服模态间固有异质性、实现多模态表示融合至关重要。尽管基于Transformer的方法在建模跨模态关系方面表现良好,但其二次计算复杂度限制了其在长序列或大规模数据上的应用。尽管最近的Mamba方法实现了线性复杂度,但其顺序扫描机制在全面建模跨模态关系方面存在根本挑战。为此,我们提出AlignMamba,一种高效且有效的多模态融合方法。具体而言,基于最优传输,我们引入一个局部跨模态对齐模块,显式学习不同模态间的词元级对应关系。此外,我们提出了基于最大均值差异(Maximum Mean Discrepancy, MMD)的全局跨模态对齐损失,隐式强制不同模态分布之间的一致性。最后,经过局部和全局对齐后的单模态表示被输入Mamba主干网络,以进行进一步的跨模态交互与融合。在完整与不完整多模态融合任务上的大量实验表明,所提方法在有效性和效率上均表现优异。
原文摘要 · Abstract (English)
Cross-modal alignment is crucial for multimodal representation fusion due to the inherent heterogeneity between modalities. While Transformer-based methods have shown promising results in modeling inter-modal relationships, their quadratic computational complexity limits their applicability to long-sequence or large-scale data. Although recent Mamba-based approaches achieve linear complexity, their sequential scanning mechanism poses fundamental challenges in comprehensively modeling cross-modal relationships. To address this limitation, we propose AlignMamba, an efficient and effective method for multimodal fusion. Specifically, grounded in Optimal Transport, we introduce a local cross-modal alignment module that explicitly learns token-level correspondences between different modalities. Moreover, we propose a global cross-modal alignment loss based on Maximum Mean Discrepancy to implicitly enforce the consistency between different modal distributions. Finally, the unimodal representations after local and global alignment are passed to the Mamba backbone for further cross-modal interaction and multimodal fusion. Extensive experiments on complete and incomplete multimodal fusion tasks demonstrate the effectiveness and efficiency of the proposed method.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。