StepOPSD: Step-Aware Online Preference Self-Distillation for Agent Reinforcement Learning

Yanfei Zhang, Xu Lin, Chenglin Wu


Abstract
Reinforcement learning for multi-turn agents suffers from a credit-assignment mismatch: rewards are sparse and trajectory-level, while success often hinges on a few local decisions. Existing online policy distillation (OPD) provides denser token-level supervision, but typically treats heterogeneous agent trajectories as monolithic strings rather than causal interaction units. We present StepOPSD, a post-rollout preference self-distillation framework that takes the agent step as the unit of credit redistribution. StepOPSD decomposes trajectories into action-centered step segments, rescoring them under hindsight-enriched teacher contexts and converting token-level log-probability gaps into sign-preserving advantage shaping with a normalized per-step credit budget before the GRPO update. Across ALFWorld and Search-QA with Qwen3-1.7B and Qwen2.5-3B-Instruct, StepOPSD attains best or second-best results on subsets most sensitive to local causal errors, including first-place performance on ALFWorld Heat (79.1%), PickTwo (95.0%), Search-QA TriviaQA (61.6%), and tied-best performance on HotpotQA (40.4%). The results further reveal a consistent two-knob law: smaller α_clip acts as a broadly stabilizing local trust region, whereas the optimal global mixing strength λ_mix remains task-dependent. These findings suggest that step-aware distillation is most useful when trajectory-level rewards are weakly aligned with the local action that determines downstream success.
Anthology ID:
2026.findings-emnlp.209
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:
4380–4390
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.209/
DOI:
Bibkey:
Cite (ACL):
Yanfei Zhang, Xu Lin, and Chenglin Wu. 2026. StepOPSD: Step-Aware Online Preference Self-Distillation for Agent Reinforcement Learning. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 4380–4390, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
StepOPSD: Step-Aware Online Preference Self-Distillation for Agent Reinforcement Learning (Zhang et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.209.pdf