acceptodds
Published as a conference paper at NeurIPS 2026

Anchoring Reasoning Distillation via Syntactic Constraints

Zehua Cheng, Wei Dai, Jiahao Sun

Abstract

Distilling "System 2" reasoning into compact student models is bottlenecked by the scarcity of fully-correct teacher traces: as task difficulty grows, rejection sampling discards an increasingly large fraction of teacher generations. "Wrong" (W) traces are abundant, but training on them indiscriminately causes *policy poisoning* the student internalises the teacher's hallucinated arithmetic alongside any useful reasoning structure. We propose **Atomic Reasoning Units (ARU)**, a framework that resolves this trade-off via strict syntactic constraints. ARU enforces a Backus–Naur-Form (BNF) grammar that decomposes each reasoning step into a Premise, an Operation, and a Result, algorithmically disentangling reasoning planning from arithmetic execution. *Counterfactual Logic Verification* (CLV) symbolically re-executes W-traces to recover the that have valid plans but faulty arithmetic, and *Syntactic Loss Masking* (SLM) trains the student on this verified structure while suppressing the gradient on hallucinated result tokens. Across the grid we evaluate ( B Qwen3 students eight benchmarks six baselines), ARU is the strongest method in every cell, improving GSM8K by and MATH-500 by over the closest baseline at  B; on the held-out AIME competition set ARU's Maj@ is versus for PoT (the closest baseline). We further document a small-scale capability gap in which ARU at  B matches Gold-SFT at  B on GSM8K (paired-bootstrap after Bonferroni correction), with a pre-registered ProofWriter control showing the gap collapses on pure-logic reasoning—consistent with arithmetic offloading as the dominant mechanism, though we do not claim the control uniquely isolates it. The code is available at https://anonymous.4open.science/r/ARU-286D.

Related papers

See more related submissions when the market opens.