ROM: Real-time Overthinking Mitigation via Streaming Detection and Intervention

Xinyan Wang, Xiaogeng Liu, Chaowei Xiao
3/23/2026
cs.LGcs.AIcs.CL

No official code found. Best implementation: SaFo-Lab/ROM (3 stars, MIT, last commit 2026-04-11).

Abstract

Large Reasoning Models (LRMs) achieve strong accuracy on challenging tasks by generating long Chain-of-Thought traces, but suffer from overthinking. Even after reaching the correct answer, they continue generating redundant reasoning steps. This behavior increases latency and compute cost and can also lead to answer drift. Existing mitigation methods either require training-heavy backbone modification or rely on hand-crafted heuristics that do not truly capture overthinking patterns. We propose ROM, the first method that formulates overthinking mitigation as a streaming prediction-and-control problem. ROM attaches a lightweight detection head to the late-layer hidden states of a frozen large language model backbone. It monitors tokens in real time and triggers an early transition to the final answer once overthinking is detected. We also introduce token-level supervision based on solution correctness boundaries and a data augmentation strategy that reduces distilled-data bias. Across seven benchmarks, ROM achieves the highest accuracy (93.51%), the shortest responses (1,159 tokens), and the best response efficiency. Compared with the vanilla baseline, it reduces response length by 47.2% and improves efficiency by 121%. These results show that streaming detection is a promising approach to real-time overthinking mitigation.

View on arXivView PDF

Code Implementations(1)

SaFo-Lab/ROMCommunity70%
MIT

The official implementation of our paper "ROM: Real-time Overthinking Mitigation via Streaming Detection and Intervention"

31PythonMar 23, 20266 months agoMIT

Similar Papers

Cite this paper

@article{wang2026rom,
  title  = {ROM: Real-time Overthinking Mitigation via Streaming Detection and Intervention},
  author = {Xinyan Wang and Xiaogeng Liu and Chaowei Xiao},
  year   = {2026},
  eprint = {2603.22016},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2603.22016v1}
}

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/2603.22016.svg)](https://codeofpaper.com/paper/2603.22016)

Discussion