From Monolithic Prioritization to Stratified Replay: Hierarchical Prioritized Experience Replay for Task-Oriented Dialogue Policy Learning

Yixin Zhang, Yangyang Zhao, Kai Xu, Lingyun Xiang, Yun Song


Abstract
Replay sampling is central to off-policy task-oriented dialogue policy learning, as it determines which past interactions are revisited during policy updates. Existing priority-based replay methods often rely on a monolithic priority distribution to form minibatches, which entangles coverage across error magnitudes with emphasis on high-error transitions. Under sparse dialogue success and heavy-tailed temporal-difference (TD) errors, this design overexposes extreme residuals and suppresses informative medium-error transitions, a failure mode we call mid-magnitude starvation. To address this problem, we propose Hierarchical Prioritized Experience Replay (HPER), a two-level replay sampler that separates coverage from prioritization. HPER first samples TD-error quantile strata according to occupancy, ensuring that different error magnitudes enter training, and then applies local prioritized replay within the selected stratum to retain emphasis on large residuals. This coverage-first design restores intermediate learning signals without diluting high-error samples. Experiments on multiple dialogue datasets show that HPER improves replay efficiency and yields faster competence onset and higher terminal success than uniform replay and monolithic priority-based replay under matched budgets.
Anthology ID:
2026.findings-emnlp.584
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:
12336–12350
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.584/
DOI:
Bibkey:
Cite (ACL):
Yixin Zhang, Yangyang Zhao, Kai Xu, Lingyun Xiang, and Yun Song. 2026. From Monolithic Prioritization to Stratified Replay: Hierarchical Prioritized Experience Replay for Task-Oriented Dialogue Policy Learning. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 12336–12350, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
From Monolithic Prioritization to Stratified Replay: Hierarchical Prioritized Experience Replay for Task-Oriented Dialogue Policy Learning (Zhang et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.584.pdf