中国科学院机构知识库网格
Chinese Academy of Sciences Institutional Repositories Grid
Bridging the Gap between Different Vocabularies for LLM Ensemble

文献类型:会议论文

作者徐杨一帆1,2; Lu JL(陆金梁)1,2; Zhang JJ(张家俊)1,2,3,4
出版日期2024-06
会议日期June 16–21, 2024
会议地点Mexico City, Mexico
英文摘要

Ensembling different large language models (LLMs) to unleash their complementary potential and harness their individual strengths is highly valuable. Nevertheless, vocabulary discrepancies among various LLMs have constrained previous studies to either selecting or blending completely generated outputs. This limitation hinders the dynamic correction and enhancement of outputs during the generation process, resulting in a limited capacity for effective ensemble. To address this issue, we propose a novel method to Ensemble LLMs via Vocabulary Alignment (EVA). EVA bridges the lexical gap among various LLMs, enabling meticulous ensemble at each generation step. Specifically, we first learn mappings between the vocabularies of different LLMs with the assistance of overlapping tokens. Subsequently, these mappings are employed to project output distributions of LLMs into a unified space, facilitating a fine-grained ensemble. Finally, we design a filtering strategy to exclude models that generate unfaithful tokens. Experimental results on commonsense reasoning, arithmetic reasoning, machine translation, and data-to-text generation tasks demonstrate the superiority of our approach compared with individual LLMs and previous ensemble methods conducted on complete outputs. Further analyses confirm that our approach can leverage knowledge from different language models and yield consistent improvement.

会议录出版者Association for Computational Linguistics
语种英语
源URL[http://ir.ia.ac.cn/handle/173211/57391]  
专题紫东太初大模型研究中心
通讯作者Zhang JJ(张家俊)
作者单位1.School of Artificial Intelligence, University of Chinese Academy of Sciences
2.Institute of Automation, Chinese Academy of Sciences
3.Wuhan AI Research
4.Shanghai Artificial Intelligence Laboratory
推荐引用方式
GB/T 7714
徐杨一帆,Lu JL,Zhang JJ. Bridging the Gap between Different Vocabularies for LLM Ensemble[C]. 见:. Mexico City, Mexico. June 16–21, 2024.

入库方式: OAI收割

来源:自动化研究所

浏览0
下载0
收藏0
其他版本

除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。