A Stitch in Time Saves Nine: Proactive Self-Refinement for Language Models
Abstract
Recent advances in self-refinement have demonstrated significant potential for improving the outputs of large language models (LLMs) through iterative refinement. However, most existing self-refinement methods rely on a reactive process with a fixed number of iterations, making it difficult to determine the optimal timing and content of refinement based on the evolving generation context. Inspired by the way humans dynamically refine their thoughts during execution, we propose ProActive Self-Refinement (PASR), a novel method that enables LLMs to refine their outputs during the generation process. Unlike methods that regenerate entire responses, PASR proactively decides whether, when, and how to refine based on the model’s internal state and evolving context. We conduct extensive experiments on a diverse set of 10 tasks to evaluate the effectiveness of PASR. Experimental results show that PASR significantly enhances problem-solving performance. In particular, on Qwen3-8B, PASR reduces average token consumption by 41.6% compared to standard generation, while also achieving an 8.2% improvement in accuracy. Our code and baselines used in the paper are available in the GitHub.
Cite
Text
Han et al. "A Stitch in Time Saves Nine: Proactive Self-Refinement for Language Models." International Conference on Learning Representations, 2026.Markdown
[Han et al. "A Stitch in Time Saves Nine: Proactive Self-Refinement for Language Models." International Conference on Learning Representations, 2026.](https://mlanthology.org/iclr/2026/han2026iclr-stitch/)BibTeX
@inproceedings{han2026iclr-stitch,
title = {{A Stitch in Time Saves Nine: Proactive Self-Refinement for Language Models}},
author = {Han, Jinyi and Wang, Xinyi and Zhao, Haiquan and Li, Tingyun and Jiang, Zishang and Jiang, Sihang and Liang, Jiaqing and Lin, Xin Alex and Zhou, Weikang and Sun, Zeye and Yu, Fei and Xiao, Yanghua},
booktitle = {International Conference on Learning Representations},
year = {2026},
url = {https://mlanthology.org/iclr/2026/han2026iclr-stitch/}
}