Less Data Less Tokens: Multilingual Unification Learning for Efficient Test-Time Reasoning in LLMs
This addresses data collection and compute efficiency challenges in LLMs, offering a domain-specific solution for multilingual reasoning tasks.
The paper tackles the problem of test-time scaling inefficiency in large language models by proposing a multilingual unification learning approach that reduces both data requirements and inference tokens while maintaining performance, showing that small amounts of multilingual data can significantly improve reasoning capabilities.
This paper explores the challenges of test-time scaling of large language models (LLMs), regarding both the data and inference efficiency. We highlight the diversity of multi-lingual reasoning based on our pilot studies, and then introduce a novel approach, \(L^2\) multi-lingual unification learning with a decoding intervention strategy for further investigation. The basic idea of \(L^2\) is that the reasoning process varies across different languages, which may be mutually beneficial to enhance both model performance and efficiency. In specific, there are two types of multi-lingual data: the entire long chain-of-thought annotations in different languages and the step-wise mixture of languages. By further tuning based on them, we show that even small amounts of data can significantly improve reasoning capabilities. Our findings suggest that multilingual learning reduces both the required data and the number of inference tokens while maintaining a comparable performance. Furthermore, \(L^2\) is orthogonal to other data efficient methods. Thus, we also emphasize the importance of diverse data selection. The \(L^2\) method offers a promising solution to the challenges of data collection and test-time compute efficiency in LLMs.