Aha-Flow Distillation: Flow Markers Matter in LLM Reasoning
Organizations: Peking University · Pengcheng Laboratory
Abstract
We identify the Flow Moment, a reasoning pattern characterized by sustained, process-confirming verbalizations such as I'm doing, in contrast to the revision- and backtracking-oriented Aha Moment. We refer to their corresponding linguistic expressions as Flow Markers and Aha Markers, respectively. Based on this observation, we construct Flow-CoT by rewriting the discourse markers of original reasoning traces while preserving their underlying reasoning content, and use it as auxiliary supervision for on-policy self-distillation (OPSD). We further propose Aha-Flow Distillation (AFD), a dual-mode extension of OPSD that pairs different forms of privileged information with corresponding reasoning instructions. The Aha branch retains concise solution-based supervision, while the Flow branch introduces rewritten Flow-CoT under a direct and confident reasoning instruction. At inference time, the model uses only the standard reflective instruction, so Flow-style reasoning serves purely as a training signal. Experiments on AIME25 and HMMT25 show consistent improvements across Qwen3-8B and Qwen3-4B: AFD improves Avg@12 from 60.8 to 61.3 on Qwen3-8B and from 57.5 to 58.6 on Qwen3-4B over our reproduced OPSD baselines. Controlled ablations further show that, with the same Flow-CoT/Aha-CoT composition, dual-mode training improves Avg@12 from 59.5 to 60.1, indicating that the benefit comes not only from introducing heterogeneous reasoning supervision, but also from how it is organized during self-distillation. The code is available at https://github.com/Wang-Xiaodong1899/Aha-Flow-Distillation.
Figures & tables
| Method | AIME25 | HMMT25 | Average |
|---|---|---|---|
| Qwen3-8B | |||
| Base (Instruct) | 65.6 | 43.9 | 54.8 |
| + SFT | 64.2 | 42.9 | 53.6 |
| + GRPO ( Guo et al., 2025 ) | 68.9 | 46.7 | 57.8 |
| + OPSD ( Zhao et al., 2026 ) | 70.8 | 45.8 | 58.3 |
| + OPSD † | 73.1 | 48.6 | 60.8 |
| Privileged information | Training mode | AIME25 | HMMT25 | Average |
|---|---|---|---|---|
| Qwen3-8B | ||||
| – | – | 65.6 | 43.9 | 54.8 |
| Solution | OPSD | 73.1 | 48.6 | 60.8 |
| Aha-CoT | OPSD | 70.8 | 47.5 | 59.2 |
| Flow-CoT / Aha-CoT | OPSD | 70.8 | 48.1 | 59.5 |
| Flow-CoT / Aha-CoT | AFD | 72.8 | 48.3 | 60.1 |
LLM Reasoning
Large language model reasoning studies how models solve multi-step problems, verify intermediate steps, and use computation or tools reliably.
See all 926 papers