Compositional Generalization with Outcome-Level Optimization

Xiyan Fu, Wei Liu


Abstract
Compositional generalization refers to correctly interpret novel combinations of known primitives, which remains a major challenge. Existing approaches often rely on supervised fine-tuning, which encourages models to imitate target outputs. This token-level training paradigm fails to capture the global compositional structure required for generalizing to unseen combinations. In this work, we investigate whether compositional generalization can instead be improved through outcome-level reinforcement learning. We adopt Group Relative Policy Optimization to optimize models based on feedback on their final outputs. Within this framework, we explore both a simple binary outcome reward and a composite reward that provides additional composition feedback. Experiments on multiple compositional benchmarks show that reinforcement learning improves compositional generalization compared to supervised fine-tuning. Further analysis reveals that supervised models tend to overfit frequent training compositions, whereas reinforcement learning improves compositional generalization by reshaping the output distribution, particularly for more complex composition types.
Anthology ID:
2026.findings-emnlp.1496
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
31845–31857
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.1496/
DOI:
10.18653/2026.findings-emnlp.1496
Bibkey:
Cite (ACL):
Xiyan Fu and Wei Liu. 2026. Compositional Generalization with Outcome-Level Optimization. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 31845–31857, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Compositional Generalization with Outcome-Level Optimization (Fu & Liu, Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.1496.pdf