STAIF: A Stage-wise Optimization for Complex Instruction Following
Not on the map. Why?
Abstract (arXiv)
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 underemphasize 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 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.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 10 of its 18 references that are in the map. First posted on arXiv 26 Jun 2026.
Builds on
10 of 18 references.
References · 10
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
- 2017Deep reinforcement learning from human preferencescited by 3,846
- 2021Finetuned Language Models Are Zero-Shot Learnerscited by 3,747
- 2023CAMEL: Communicative Agents for "Mind" Exploration of Large Language Model Societycited by 1,348
- 2025VerIF: Verification Engineering for Reinforcement Learning in Instruction Followingcited by 43
- 2024IOPO: Empowering LLMs with Complex Instruction Following via Input-Output Preference Optimizationcited by 30
- 2024The Impossibility of Fair LLMscited by 22
- 2025MuSC: Improving Complex Instruction Following with Multi-granularity Self-Contrastive Trainingcited by 12
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026TinyJudge: Unverifiable Constraint Alignment via Lightweight Specialist Ensembles
- 2025BLEUBERI: BLEU is a surprisingly effective reward for instruction following
- 2025Reverse Preference Optimization for Complex Instruction Following
- 2025Incentivizing Reasoning for Advanced Instruction-Following of Large Language Models
- 2026Cross-Relational Preference Learning for Better LLM Instruction Following
- 2026HIPO: Instruction Hierarchy via Constrained Reinforcement Learning
- 2024Align2LLaVA: Cascaded Human and Large Language Model Preference Alignment for Multi-modal Instruction Curation
- 2026IF-RewardBench: Benchmarking Judge Models for Instruction-Following Evaluation