Kun Zhou
Author directoryOther people with similar names: Kun Zhou, Kun Zhou
Unverified author pages with similar names: Kun Zhou
2026
PA3: Policy-Aware Agent Alignment through Chain-of-Thought
Shubhashis Roy Dipta | Daniel Bis | Kun Zhou | Lichao Wang | Benjamin Z. Yao | Chenlei Guo | Ruhi Sarikaya
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Shubhashis Roy Dipta | Daniel Bis | Kun Zhou | Lichao Wang | Benjamin Z. Yao | Chenlei Guo | Ruhi Sarikaya
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Conversational assistants powered by large language models (LLMs) excel at tool-use tasks but struggle with adhering to complex, business-specific rules. While models can reason over business rules provided in context, including all policies for every query introduces high latency and wastes compute. Furthermore, these lengthy prompts lead to long contexts, harming overall performance due to the ’needle-in-a-haystack’ problem. To address these challenges, we propose a multi-stage alignment method that teaches models to recall and apply relevant business policies during chain-of-thought reasoning at inference time, without including the full business policy in-context. Furthermore, we introduce a novel Policy Recall reward based on the Jaccard score and a Hallucination Penalty for GRPO training. Altogether, our best model outperforms the baseline by **16 points** and surpasses comparable *in-context* baselines of similar model size by **3 points**, while using **40% fewer words**.