English

The surprising efficiency of temporal difference learning for rare event prediction

Machine Learning 2025-01-17 v3 Artificial Intelligence

Abstract

We quantify the efficiency of temporal difference (TD) learning over the direct, or Monte Carlo (MC), estimator for policy evaluation in reinforcement learning, with an emphasis on estimation of quantities related to rare events. Policy evaluation is complicated in the rare event setting by the long timescale of the event and by the need for \emph{relative accuracy} in estimates of very small values. Specifically, we focus on least-squares TD (LSTD) prediction for finite state Markov chains, and show that LSTD can achieve relative accuracy far more efficiently than MC. We prove a central limit theorem for the LSTD estimator and upper bound the \emph{relative asymptotic variance} by simple quantities characterizing the connectivity of states relative to the transition probabilities between them. Using this bound, we show that, even when both the timescale of the rare event and the relative accuracy of the MC estimator are exponentially large in the number of states, LSTD maintains a fixed level of relative accuracy with a total number of observed transitions of the Markov chain that is only \emph{polynomially} large in the number of states.

Keywords

Cite

@article{arxiv.2405.17638,
  title  = {The surprising efficiency of temporal difference learning for rare event prediction},
  author = {Xiaoou Cheng and Jonathan Weare},
  journal= {arXiv preprint arXiv:2405.17638},
  year   = {2025}
}

Comments

Final camera-ready version published at NeurIPS 2024. Correct an assumption statement and typos, and change/add a few sentences from the last version

R2 v1 2026-06-28T16:42:55.081Z