STAIF: A Stage-wise Optimization for Complex Instruction Following

Jian Hong, Chen Cheng, Quan Liu, Yuhao Chen, Enhong Chen


Abstract
Following complex instructions with multiple explicit constraints remains a fundamental challenge for large language models (LLMs). Existing alignment methods, such as DPO, optimize holistic reward signals that often under-emphasize strict satisfaction of individual constraints, particularly under out-of-distribution or multi-constraint settings. In this paper, we propose STAIF, a stage-wise optimization framework that decouples the alignment of subjective (soft) constraints from the optimization of objectively verifiable (hard) constraints. Stage 1 applies preference optimization with multiple negative samples to sharpen the sensitivity to soft constraints, while Stage 2 applies Reinforcement Learning with Verifiable Rewards (RLVR) to enforce strict compliance with hard constraints. To support this method, we construct STAINSTRUCT, a high-quality bilingual (English, Chinese) dataset of approximately 31,000 complex multi-constraint instructions. Extensive analyses validate the design of STAIF and show state-of-the-art performance on representative benchmarks against strong baselines, as well as genuine generalization.
Anthology ID:
2026.findings-emnlp.1281
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:
27136–27153
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.1281/
DOI:
Bibkey:
Cite (ACL):
Jian Hong, Chen Cheng, Quan Liu, Yuhao Chen, and Enhong Chen. 2026. STAIF: A Stage-wise Optimization for Complex Instruction Following. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 27136–27153, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
STAIF: A Stage-wise Optimization for Complex Instruction Following (Hong et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.1281.pdf