arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2406.03496cs.CLcs.AIcs.LG
收藏

Wings:学习无纯文本遗忘的多模态LLMs

Wings: Learning Multimodal LLMs without Text-only Forgetting

School of Artificial Intelligence, Nanjing University(南京大学人工智能学院);AI Business, Alibaba Group(阿里巴巴集团AI业务部)

Yi-Kai Zhang, Shiyin Lu, Yang Li, Yanqing Ma, Qing-Guo Chen, Zhao Xu, Weihua Luo, Kaifu Zhang, De-Chuan Zhan, Han-Jia Ye

AI总结:

针对多模态大语言模型灾难性遗忘纯文本指令的问题,提出Wings模型,通过构建并行的视觉与文本学习者模块及低秩残差注意力,有效补偿注意力转移,在纯文本与多模态问答任务上均取得优异表现。

摘要:

多模态大语言模型(MLLMs)以训练好的LLM为起点,首先将图像与文本对齐,然后在多模态混合输入上进行微调。然而,MLLM会灾难性地遗忘纯文本指令,这些指令不包含图像且可以在初始LLM内处理。在本文中,我们提出了Wings,一种在纯文本对话和多模态理解方面都表现出色的新型MLLM。分析多模态指令中的MLLM注意力表明,纯文本遗忘与注意力从图像前文本到图像后文本的转移有关。由此,我们构建了额外模块,作为提升学习者来补偿注意力转移。互补的视觉和文本学习者,像两侧的“翅膀”,在每个层的注意力块内并行连接。最初,图像和文本输入与视觉学习者对齐,与主注意力并行运行,平衡对视觉元素的关注。文本学习者随后通过基于注意力的路由协作整合,以融合视觉和文本学习者的输出。我们设计了低秩残差注意力(LoRRA)以保证学习者的高效性。我们的实验结果表明,Wings在纯文本和视觉问答任务中均优于同等规模的MLLMs。在新建的交错图文(IIT)基准上,Wings在从纯文本丰富到多模态丰富的问答任务中表现出色。

英文摘要:

Multimodal large language models (MLLMs), initiated with a trained LLM, first align images with text and then fine-tune on multimodal mixed inputs. However, the MLLM catastrophically forgets the text-only instructions, which do not include images and can be addressed within the initial LLM. In this paper, we present Wings, a novel MLLM that excels in both text-only dialogues and multimodal comprehension. Analyzing MLLM attention in multimodal instructions reveals that text-only forgetting is related to the attention shifts from pre-image to post-image text. From that, we construct extra modules that act as the boosted learner to compensate for the attention shift. The complementary visual and textual learners, like "wings" on either side, are connected in parallel within each layer's attention block. Initially, image and text inputs are aligned with visual learners operating alongside the main attention, balancing focus on visual elements. Textual learners are later collaboratively integrated with attention-based routing to blend the outputs of the visual and textual learners. We design the Low-Rank Residual Attention (LoRRA) to guarantee high efficiency for learners. Our experimental results demonstrate that Wings outperforms equally-scaled MLLMs in both text-only and visual question-answering tasks. On a newly constructed Interleaved Image-Text (IIT) benchmark, Wings exhibits superior performance from text-only-rich to multimodal-rich question-answering tasks.

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

↑