Figure 1: Schematic overview of CyclicReflex. The rightmost subfigure presents a comparison of final answer accuracy between CyclicReflex, the original LRM, and decoding variants using TIP and S1. |
This is the official code repository for the ICLR 2026 paper CyclicReflex: Improving Reasoning Models via Cyclical Reflection Token Scheduling.
Large reasoning models (LRMs), such as OpenAI’s o1 and DeepSeek-R1, harness test-time scaling to perform multi-step reasoning for complex problem-solving. This reasoning process, executed before producing final answers, is often guided by special juncture tokens that prompt self-evaluative reflection. These transition markers and reflective cues are referred to as “reflection tokens” (e.g., “wait”, “but”, “alternatively”). In this work, we treat reflection tokens as a “resource” and introduce the problem of resource allocation, aimed at improving the test-time compute performance of LRMs by adaptively regulating the frequency and placement of reflection tokens. Through empirical analysis, we show that both excessive and insufficient use of reflection tokens, referred to as over-reflection and under-reflection, can degrade model performance. To better understand this trade-off, we draw an analogy between reflection token usage and learning rate scheduling in optimization. Building on this insight, We propose cyclical reflection token scheduling (termed CyclicReflex), a training-free decoding strategy that dynamically modulates reflection token logits with a bidirectional, position-dependent triangular waveform, incurring no additional computation cost. Experiments on MATH500, AIME2024/2025, AMC2023, GPQA Diamond and LiveCodeBench demonstrate that CyclicReflex consistently improves performance across model sizes (1.5B–14B), outperforming standard decoding and recent approaches such as TIP (thought switching penalty) and S1.
@inproceedings{fancyclicreflex,
title={CyclicReflex: Improving Reasoning Models via Cyclical Reflection Token Scheduling},
author={Fan, Chongyu and Zhang, Yihua and Jia, Jinghan and Hero, Alfred O and Liu, Sijia},
booktitle={The Fourteenth International Conference on Learning Representations}
}