Temporal Sampling for Forgotten Reasoning in LLMs

Yuetai Li, Zhangchen Xu, Fengqing Jiang, Bhaskar Ramasubramanian, Luyao Niu +3 more
5/26/2025
cs.AIcs.LG

No official code found. Best implementation: uw-nsl/Temporal_Forgetting (12 stars, MIT, last commit 2025-07-19).

Abstract

Fine-tuning large language models (LLMs) is intended to improve their reasoning capabilities, yet we uncover a counterintuitive effect: models often forget how to solve problems they previously answered correctly during training. We term this phenomenon temporal forgetting and show that it is widespread across model sizes, fine-tuning methods (both Reinforcement Learning and Supervised Fine-Tuning), and multiple reasoning benchmarks. To address this gap, we introduce Temporal Sampling, a simple decoding strategy that draws outputs from multiple checkpoints along the training trajectory. This approach recovers forgotten solutions without retraining or ensembling, and leads to substantial improvements in reasoning performance, gains from 4 to 19 points in Pass@k and consistent gains in Majority@k across several benchmarks. We further extend our method to LoRA-adapted models, demonstrating that storing only adapter weights across checkpoints achieves similar benefits with minimal storage cost. By leveraging the temporal diversity inherent in training, Temporal Sampling offers a practical, compute-efficient way to surface hidden reasoning ability and rethink how we evaluate LLMs.

View on arXivView PDF

Code Implementations(1)

MIT
121C++, ShellMay 25, 20251 years agoMIT

Similar Papers

Cite this paper

@article{li2025temporal,
  title  = {Temporal Sampling for Forgotten Reasoning in LLMs},
  author = {Yuetai Li and Zhangchen Xu and Fengqing Jiang and Bhaskar Ramasubramanian and Luyao Niu and Bill Yuchen Lin and Xiang Yue and Radha Poovendran},
  year   = {2025},
  eprint = {2505.20196},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2505.20196v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2505.20196.svg)](https://codeofpaper.com/paper/2505.20196)

Discussion