English

Tracking Drift: Variation-Aware Entropy Scheduling for Non-Stationary Reinforcement Learning

Machine Learning 2026-05-19 v2 Artificial Intelligence

Abstract

Real-world reinforcement learning often faces environment drift, but most existing methods rely on static entropy coefficients/target entropy, causing over-exploration during stable periods and under-exploration after drift, and leaving unanswered the principled question of how exploration intensity should scale with drift magnitude. We show that, under standard assumptions, entropy scheduling in non-stationary maximum-entropy RL can be cast as the dynamic-regret trade-off between tracking a drifting comparator and stabilizing updates, yielding a square-root scaling rule for the entropy weight in terms of a online non-stationarity proxy. Building on this, we propose AES--Adaptive Entropy Scheduling--which adaptively adjusts the entropy coefficient/temperature online using observable drift proxies during training, requiring almost no structural changes and incurring minimal overhead. Across 4 algorithm variants, 12 tasks, and 4 drift modes, AES significantly reduces the fraction of performance degradation caused by drift and accelerates recovery after abrupt changes.

Keywords

Cite

@article{arxiv.2601.19624,
  title  = {Tracking Drift: Variation-Aware Entropy Scheduling for Non-Stationary Reinforcement Learning},
  author = {Tongxi Wang and Zhuoyang Xia and Xinran Chen and Shan Liu},
  journal= {arXiv preprint arXiv:2601.19624},
  year   = {2026}
}

Comments

Accepted by ICML 2026

R2 v1 2026-07-01T09:22:19.128Z