arXiv:2410.05938cs.CVcs.AI2024-10ICLR被引 6

EMMA提升视觉与文本对齐,让Mamba模型更准更快地理解图文。

EMMA: Empowering Multi-modal Mamba with Structural and Hierarchical Alignment

  • 引入像素级对齐模块,实现图像与文本的结构化对齐。
  • 多尺度特征融合模块保留视觉细节,推理速度比同类Transformer快近4倍。
  • 适合追求高效精准多模态理解的研究者与开发者。

基于Mamba的架构因其优异性能和亚二次部署速度成为深度学习新方向。然而,现有Mamba多模态大模型在视觉特征提取上不足,导致视觉与文本潜在表示间跨模态对齐失衡,影响多模态任务表现。本文提出EMMA(Empowering Multi-modal Mamba with Structural and Hierarchical Alignment),通过像素级对齐模块,自回归优化空间图像特征与文本标记的学习与处理,实现图像层面的结构对齐;同时设计多尺度特征融合(MFF)模块,整合中间层多尺度视觉特征,防止跨模态对齐过程中的信息退化,实现特征层面的层次对齐。在多个多模态基准测试中,该模型推理延迟低于其他Mamba基模型,且接近同类规模Transformer模型的四倍加速。由于更优的跨模态对齐,模型幻觉程度更低,对视觉细节更敏感,在多样多模态任务上表现更优。代码将公开。

原文摘要 · Abstract (English)

Mamba-based architectures have shown to be a promising new direction for deep learning models owing to their competitive performance and sub-quadratic deployment speed. However, current Mamba multi-modal large language models (MLLM) are insufficient in extracting visual features, leading to imbalanced cross-modal alignment between visual and textural latents, negatively impacting performance on multi-modal tasks. In this work, we propose Empowering Multi-modal Mamba with Structural and Hierarchical Alignment (EMMA), which enables the MLLM to extract fine-grained visual information. Specifically, we propose a pixel-wise alignment module to autoregressively optimize the learning and processing of spatial image-level features along with textual tokens, enabling structural alignment at the image level. In addition, to prevent the degradation of visual information during the cross-model alignment process, we propose a multi-scale feature fusion (MFF) module to combine multi-scale visual features from intermediate layers, enabling hierarchical alignment at the feature level. Extensive experiments are conducted across a variety of multi-modal benchmarks. Our model shows lower latency than other Mamba-based MLLMs and is nearly four times faster than transformer-based MLLMs of similar scale during inference. Due to better cross-modal alignment, our model exhibits lower degrees of hallucination and enhanced sensitivity to visual details, which manifests in superior performance across diverse multi-modal benchmarks. Code will be provided.

多模态Mamba对齐优化高效推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。