Breaking the Ceiling of the LLM Community by Treating Token Generation as a Classification for Ensembling

Yao-Ching Yu, Chun Chih Kuo, Ye Ziqi, Chang Yucheng, Yueh-Se Li


Abstract
Ensembling multiple models has always been an effective approach to push the limits of existing performance and is widely used in classification tasks by simply averaging the classification probability vectors from multiple classifiers to achieve better accuracy. However, in the thriving open-source Large Language Model (LLM) community, ensembling methods are rare and typically limited to ensembling the full-text outputs of LLMs, such as selecting the best output using a ranker, which leads to underutilization of token-level probability information. In this paper, we treat the **G**eneration of each token by LLMs **a**s a **C**lassification (**GaC**) for ensembling. This approach fully exploits the probability information at each generation step and better prevents LLMs from producing early incorrect tokens that lead to snowballing errors. In experiments, we ensemble state-of-the-art LLMs on several benchmarks, including exams, mathematics and reasoning, and observe that our method breaks the existing community performance ceiling. Furthermore, we observed that most of the tokens in the answer are simple and do not affect the correctness of the final answer. Therefore, we also experimented with ensembling only key tokens, and the results showed better performance with lower latency across benchmarks.
Anthology ID:
2024.findings-emnlp.99
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2024
Month:
November
Year:
2024
Address:
Miami, Florida, USA
Editors:
Yaser Al-Onaizan, Mohit Bansal, Yun-Nung Chen
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
1826–1839
Language:
URL:
https://preview.aclanthology.org/Add-Cong-Liu-Florida-Atlantic-University-author-id/2024.findings-emnlp.99/
DOI:
10.18653/v1/2024.findings-emnlp.99
Bibkey:
Cite (ACL):
Yao-Ching Yu, Chun Chih Kuo, Ye Ziqi, Chang Yucheng, and Yueh-Se Li. 2024. Breaking the Ceiling of the LLM Community by Treating Token Generation as a Classification for Ensembling. In Findings of the Association for Computational Linguistics: EMNLP 2024, pages 1826–1839, Miami, Florida, USA. Association for Computational Linguistics.
Cite (Informal):
Breaking the Ceiling of the LLM Community by Treating Token Generation as a Classification for Ensembling (Yu et al., Findings 2024)
Copy Citation:
PDF:
https://preview.aclanthology.org/Add-Cong-Liu-Florida-Atlantic-University-author-id/2024.findings-emnlp.99.pdf
Software:
 2024.findings-emnlp.99.software.zip