Self-training Language Models for Arithmetic Reasoning

Marek Kadlčík, Michal Štefánik


Abstract
Recent language models achieve impressive results in tasks involving complex multistep reasoning, but scaling these capabilities further traditionally requires expensive collection of more annotated data.In this work, we explore the potential of improving models’ reasoning capabilities without new data, merely using automated feedback to the validity of their predictions in arithmetic reasoning (self-training).In systematic experimentation across six different arithmetic reasoning datasets, we find that models can substantially improve in both single-round (offline) and online self-training, reaching a correct result in +13.9% and +25.9% more cases, respectively, underlining the importance of actuality of self-training feedback. We further find that in the single-round, offline self-training, traditional supervised training can deliver gains comparable to preference optimization, but in online self-training, preference optimization methods largely outperform supervised training thanks to their superior stability and robustness on unseen types of problems.
Anthology ID:
2024.findings-emnlp.721
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:
12378–12386
Language:
URL:
https://aclanthology.org/2024.findings-emnlp.721
DOI:
Bibkey:
Cite (ACL):
Marek Kadlčík and Michal Štefánik. 2024. Self-training Language Models for Arithmetic Reasoning. In Findings of the Association for Computational Linguistics: EMNLP 2024, pages 12378–12386, Miami, Florida, USA. Association for Computational Linguistics.
Cite (Informal):
Self-training Language Models for Arithmetic Reasoning (Kadlčík & Štefánik, Findings 2024)
Copy Citation:
PDF:
https://aclanthology.org/2024.findings-emnlp.721.pdf
Software:
 2024.findings-emnlp.721.software.zip