Bridging the Gap between Different Vocabularies for LLM Ensemble
文献类型:会议论文
作者 | 徐杨一帆1,2![]() ![]() ![]() |
出版日期 | 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收割
来源:自动化研究所
其他版本
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。