arXiv:2410.05357cs.LGcs.AI2024-10NeurIPS被引 10

整合多种模型融合技术,提升异构大模型组合性能。

Model-GLUE: Democratized LLM Scaling for A Large Model Zoo in the Wild

  • 通过聚类与最优合并策略选择可兼容模型
  • 在不额外训练下实现平均5.61%性能提升
  • 适合需要高效组合大模型的实践者

随着大语言模型在各类任务和专业领域表现优异,基于现有模型的规模化扩展受到广泛关注,但不同模型组合常导致性能下降。现有模型聚合方法包括模型合并、专家混合(Mixture-of-Experts)和堆叠等。尽管各有优势,但对它们在多样化模型库中的综合比较与协同应用仍缺乏系统研究。为此,本文提出Model-GLUE,一个全面的大模型扩展指导框架。首先对现有模型扩展技术(特别是选择性合并及其变体)进行基准测试;基于测试结果,制定针对异构模型库(包含不同架构与初始化方式)的最优选型与聚合策略。方法包括可合并模型的聚类、最优合并策略选择,以及通过模型混合整合聚类结果。实验基于多样化的基于Llama-2的模型库验证,证明Model-GLUE在无需额外训练的情况下实现了平均5.61%的性能提升。代码已开源:https://github.com/Model-GLUE/Model-GLUE。

原文摘要 · Abstract (English)

As Large Language Models (LLMs) excel across tasks and specialized domains, scaling LLMs based on existing models has garnered significant attention, which faces the challenge of decreasing performance when combining disparate models. Various techniques have been proposed for the aggregation of pre-trained LLMs, including model merging, Mixture-of-Experts, and stacking. Despite their merits, a comprehensive comparison and synergistic application of them to a diverse model zoo is yet to be adequately addressed. In light of this research gap, this paper introduces Model-GLUE, a holistic LLM scaling guideline. First, our work starts with a benchmarking of existing LLM scaling techniques, especially selective merging, and variants of mixture. Utilizing the insights from the benchmark results, we formulate an optimal strategy for the selection and aggregation of a heterogeneous model zoo characterizing different architectures and initialization.Our methodology involves the clustering of mergeable models and optimal merging strategy selection, and the integration of clusters through a model mixture. Finally, evidenced by our experiments on a diverse Llama-2-based model zoo, Model-GLUE shows an average performance enhancement of 5.61%, achieved without additional training. Codes are available at: https://github.com/Model-GLUE/Model-GLUE.

大模型融合模型扩展异构模型LLM优化

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