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

SoupLM:大语言模型与多模态模型中的模型集成

SoupLM: Model Integration in Large Language and Multi-Modal Models

Northeastern University(东北大学);Allen Institute for AI(艾伦人工智能研究所)

Yue Bai, Zichen Zhang, Jiasen Lu, Yun Fu

AI总结:

针对大语言及多模态模型训练成本高的问题,提出SoupLM模型集成方法,通过模型汤策略整合不同LLM变体的跨领域跨模态知识,降低重复训练成本,并系统测试了不同配置下的性能增益。

摘要:

训练大语言模型(LLM)和多模态大语言模型需要大量计算资源,而现有的公开大语言模型通常是在涵盖各类任务的多样化私有策划数据集上预训练的。例如,LLaMA、Vicuna和LLaVA是三种基于LLaMA基座模型、采用截然不同的训练方案、任务和数据模态训练得到的LLM变体,这类LLM变体的训练成本和复杂度增长迅速。本研究中,我们提出采用soup(模型汤)策略,以高成本效益的方式将这些LLM变体组装为单个泛化性良好的多模态大语言模型SoupLM。高效组装这些LLM变体可将不同领域和数据模态下训练得到的知识与专长整合到单一模型中(例如Vicuna从用户共享对话中习得的聊天机器人专长,以及LLaVA从视觉-语言数据中获得的视觉能力),从而避免在多个不同领域重复训练的计算成本。我们提出了一系列soup策略,系统地基准测试了不同配置下的性能提升,并探究了插值空间中不同基座模型的soup表现。

英文摘要:

Training large language models (LLMs) and multimodal LLMs necessitates significant computing resources, and existing publicly available LLMs are typically pre-trained on diverse, privately curated datasets spanning various tasks. For instance, LLaMA, Vicuna, and LLaVA are three LLM variants trained with LLaMA base models using very different training recipes, tasks, and data modalities. The training cost and complexity for such LLM variants grow rapidly. In this study, we propose to use a soup strategy to assemble these LLM variants into a single well-generalized multimodal LLM (SoupLM) in a cost-efficient manner. Assembling these LLM variants efficiently brings knowledge and specialities trained from different domains and data modalities into an integrated one (e.g., chatbot speciality from user-shared conversations for Vicuna, and visual capacity from vision-language data for LLaVA), therefore, to avoid computing costs of repetitive training on several different domains. We propose series of soup strategies to systematically benchmark performance gains across various configurations, and probe the soup behavior across base models in the interpolation space.

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

↑