mlsanityData to 7 Oct 2026

STAIF: A Stage-wise Optimization for Complex Instruction Following

arXiv 2607.22649 ↗First posted 26 Jun 2026cs.AI cs.CLcited by 0

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

  1. 2022Training language models to follow instructions with human feedbackcited by 15,712
  2. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
  3. 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
  4. 2017Deep reinforcement learning from human preferencescited by 3,846
  5. 2021Finetuned Language Models Are Zero-Shot Learnerscited by 3,747
  6. 2023CAMEL: Communicative Agents for "Mind" Exploration of Large Language Model Societycited by 1,348
  7. 2025VerIF: Verification Engineering for Reinforcement Learning in Instruction Followingcited by 43
  8. 2024IOPO: Empowering LLMs with Complex Instruction Following via Input-Output Preference Optimizationcited by 30
  9. 2024The Impossibility of Fair LLMscited by 22
  10. 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

  1. 2026TinyJudge: Unverifiable Constraint Alignment via Lightweight Specialist Ensembles
  2. 2025BLEUBERI: BLEU is a surprisingly effective reward for instruction following
  3. 2025Reverse Preference Optimization for Complex Instruction Following
  4. 2025Incentivizing Reasoning for Advanced Instruction-Following of Large Language Models
  5. 2026Cross-Relational Preference Learning for Better LLM Instruction Following
  6. 2026HIPO: Instruction Hierarchy via Constrained Reinforcement Learning
  7. 2024Align2LLaVA: Cascaded Human and Large Language Model Preference Alignment for Multi-modal Instruction Curation
  8. 2026IF-RewardBench: Benchmarking Judge Models for Instruction-Following Evaluation