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

LongLLaVA:通过混合架构高效地将多模态大语言模型扩展至1000张图像

LongLLaVA: Scaling Multi-modal LLMs to 1000 Images Efficiently via a Hybrid Architecture

The Chinese University of Hong Kong, Shenzhen(香港中文大学(深圳));Lehigh University(莱斯利大学);Meituan(美团)

Xidong Wang, Dingjie Song, Shunian Chen, Junyin Chen, Zhenyang Cai, Chen Zhang, Lichao Sun, Benyou Wang

AI总结:

LongLLaVA提出融合Mamba与Transformer的混合架构,结合时空依赖数据构建和渐进式训练策略,在保持高吞吐和低内存的同时将多模态大语言模型高效扩展至单卡处理近千张图像。

摘要:

扩展多模态大语言模型(MLLMs)的长上下文能力对于推进视频理解和高分辨率图像分析至关重要。实现这一目标需要在模型架构、数据构建和训练策略上进行系统性改进,特别是要应对图像数量增加时性能下降和高计算成本等挑战。在本文中,我们提出了一种融合Mamba与Transformer模块的混合架构,引入了能够同时捕捉时间依赖和空间依赖的数据构建方法,并采用了渐进式训练策略。我们发布的模型LongLLaVA(Long-Context Large Language and Vision Assistant,长上下文大语言与视觉助手)展示了效率与性能之间的有效平衡。LongLLaVA在多个基准测试上取得了具有竞争力的结果,同时保持了高吞吐量和低内存消耗。值得注意的是,它可以在单块A100 80GB GPU上处理近一千张图像,凸显了其在广泛多模态应用中的潜力。

英文摘要:

Expanding the long-context capabilities of Multi-modal Large Language Models~(MLLMs) is critical for advancing video understanding and high-resolution image analysis. Achieving this requires systematic improvements in model architecture, data construction, and training strategies, particularly to address challenges such as performance degradation with increasing image counts and high computational costs. In this paper, we propose a hybrid architecture that integrates Mamba and Transformer blocks, introduce data construction methods that capture both temporal and spatial dependencies, and employ a progressive training strategy. Our released model, LongLLaVA (\textbf{Long}-Context \textbf{L}arge \textbf{L}anguage \textbf{a}nd \textbf{V}ision \textbf{A}ssistant), demonstrates an effective balance between efficiency and performance. LongLLaVA achieves competitive results across various benchmarks while maintaining high throughput and low memory consumption. Notably, it can process nearly one thousand images on a single A100 80GB GPU, underscoring its potential for a wide range of multi-modal applications.

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

↑