Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Reasoning Up the Instruction Ladder for Controllable Language Models

About

As large language model (LLM) based systems take on high-stakes roles in real-world decision-making, they must reconcile competing instructions from multiple sources within a single prompt context. Enforcing an instruction hierarchy, where higher-level directives override lower-priority requests, is critical to the reliability and control of LLMs. In this work, we reframe instruction hierarchy resolution as a reasoning task. The model must first "think" about the relationship between a given user prompt and higher-priority instructions before generating a response. To enable this capability, we construct VerIH, a training dataset of constraint-following tasks with verifiable answers, comprising aligned and conflicting system-user instructions. We show that lightweight reinforcement learning with VerIH effectively transfers general reasoning capabilities of models to instruction prioritization. Our method leads to consistent improvements across multiple model families on both instruction following and instruction hierarchy benchmarks, achieving ~20% absolute improvement in conflict setups. Our method also leads to improved alignment to safety-critical scenarios beyond the training distribution, exhibiting increased robustness against jailbreak and prompt injection, reducing absolute attack success rates by up to 20%. Our results establish reasoning over instruction hierarchies as a practical mechanism for improving AI reliability, where targeted updates to system prompts produce predictable, controllable, and robust changes in model behavior.

Zishuo Zheng, Vidhisha Balachandran, Chan Young Park, Faeze Brahman, Sachin Kumar• 2025

Related benchmarks

TaskDatasetResultRank
Instruction Hierarchy AdherenceIHEval Conflict Multi-turn
Accuracy51.74
16
Instruction Hierarchy AdherenceIHEval Aligned Single-turn
Accuracy84.59
16
Instruction Hierarchy AdherenceIHEval Aligned Multi-turn
Accuracy86.29
16
Instruction Hierarchy AdherenceIHEval Conflict Single-turn
Accuracy47.56
16
Showing 4 of 4 rows

Other info

Follow for update