Clipped Q-Learning: Your Value Clipping Is Secretly A Robust Operator
Abstract
Lay Summary
Reinforcement learning trains an AI agent to make a sequence of decisions, such as controlling a robot or choosing a treatment strategy. However, agents trained in one environment can fail when the real world behaves differently from the training environment. This paper studies a very simple modification to Q-learning, one of the most basic reinforcement learning algorithms: when estimating the future value of an action, we cap that value at a chosen threshold. Although this looks like a small technical trick, we show that it has a deeper meaning. Value clipping makes Q-learning behave as if it is preparing for a worst-case change in the environment, where highly rewarding future outcomes may become less reliable. As a result, the learned policy can favor safer and more stable decisions instead of risky decisions that only work under ideal conditions. We provide mathematical guarantees showing that this clipped version of Q-learning can learn efficiently. We also show in experiments that adding clipping to a deep Q-learning method can improve robustness when the testing environment differs from training.