Tracking Drift: Variation-Aware Entropy Scheduling for Non-Stationary Reinforcement Learning
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.
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