Re-Tuning: Overcoming the Compositionality Limits of Large Language Models with Recursive Tuning

Eric Pasewark, Kyle Montgomery, Kefei Duan, Dawn Song, Chenguang Wang


Abstract
We present a new method for large language models to solve compositional tasks. Although they have shown strong performance on traditional language understanding tasks, large language models struggle to solve compositional tasks, where the solution depends on solving smaller instances of the same problem. We propose a natural approach to solve compositional tasks recursively. Our method, Re-Tuning, tunes models to break down a problem into subproblems, solve those subproblems, and combine the results. We show that our method significantly improves model performance on three representative compositional tasks: integer addition, dynamic programming, and parity. Compared to state-of-the-art methods that keep intermediate steps towards solving the problems, Re-Tuning achieves significantly higher accuracy and is more GPU memory efficient.
Anthology ID:
2024.acl-long.561
Volume:
Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Month:
August
Year:
2024
Address:
Bangkok, Thailand
Editors:
Lun-Wei Ku, Andre Martins, Vivek Srikumar
Venue:
ACL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
10422–10437
Language:
URL:
https://aclanthology.org/2024.acl-long.561
DOI:
Bibkey:
Cite (ACL):
Eric Pasewark, Kyle Montgomery, Kefei Duan, Dawn Song, and Chenguang Wang. 2024. Re-Tuning: Overcoming the Compositionality Limits of Large Language Models with Recursive Tuning. In Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), pages 10422–10437, Bangkok, Thailand. Association for Computational Linguistics.
Cite (Informal):
Re-Tuning: Overcoming the Compositionality Limits of Large Language Models with Recursive Tuning (Pasewark et al., ACL 2024)
Copy Citation:
PDF:
https://aclanthology.org/2024.acl-long.561.pdf