通过最大化模型多样性,用更少模型实现更高性能的LLM集成。
LLM-TOPLA: Efficient LLM Ensemble by Maximising Diversity
- 引入焦点多样性度量,捕捉模型间差异与性能的关系。
- 剪枝算法选出小规模高绩效子集,如从N个模型中选S个。
- 学习融合不一致输出,适合追求高效高精度的模型集成场景。
将大语言模型在训练或推理阶段组合,可显著提升性能。本文提出LLM-TOPLA,一种基于多样性的高效模型集成方法:(i) 提出焦点多样性度量,捕捉集成中各模型间的多样性与性能关联;(ii) 设计多样性优化剪枝算法,从N个基础模型中筛选出性能最优的top-k子集成,其规模S远小于N;(iii) 采用学习型集成策略,自动检测并修正组件模型输出不一致问题。在四个基准测试上评估显示:(i) 在受限解空间任务中,相比最佳现有集成(Mixtral),LLM-TOPLA在MMLU上准确率提升2.2%,在GSM8k上比最佳集成MoreAgent提升2.1%;(ii) 在生成任务中,相较前两名(Llama70b/Mixtral),在SearchQA上F1提升3.9倍,在XSum上ROUGE-1提升超38点。代码与包含8个现代LLM在4个基准上的输出数据集已开源。
原文摘要 · Abstract (English)
Combining large language models during training or at inference time has shown substantial performance gain over component LLMs. This paper presents LLM-TOPLA, a diversity-optimized LLM ensemble method with three unique properties: (i) We introduce the focal diversity metric to capture the diversity-performance correlation among component LLMs of an ensemble. (ii) We develop a diversity-optimized ensemble pruning algorithm to select the top-k sub-ensembles from a pool of $N$ base LLMs. Our pruning method recommends top-performing LLM subensembles of size $S$, often much smaller than $N$. (iii) We generate new output for each prompt query by utilizing a learn-to-ensemble approach, which learns to detect and resolve the output inconsistency among all component LLMs of an ensemble. Extensive evaluation on four different benchmarks shows good performance gain over the best LLM ensemble methods: (i) In constrained solution set problems, LLM-TOPLA outperforms the best-performing ensemble (Mixtral) by 2.2\% in accuracy on MMLU and the best-performing LLM ensemble (MoreAgent) on GSM8k by 2.1\%. (ii) In generative tasks, LLM-TOPLA outperforms the top-2 performers (Llama70b/Mixtral) on SearchQA by $3.9\mathrm{x}$ in F1, and on XSum by more than $38$ in ROUGE-1. Our code and dataset, which contains outputs of 8 modern LLMs on 4 benchmarks is available at https://github.com/git-disl/llm-topla
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。