CoTranslator: Reverse to Forward Chain-of-Thought Translator for Reasoning Distillation
CoTranslator turns reverse chain-of-thought into verified forward traces, cutting distillation compute while keeping most math accuracy.
CoTranslator learns to map reverse chain-of-thought responses, which condition on a known answer, toward the distribution of verified forward responses used for reasoning distillation. Trained on a small paired set, it then supports large-scale generation of student training data. Across six student models and six mathematics benchmarks, it improves average accuracy over semantic suppression by up to 5.82 percentage points. For two 4B students it retains 95.0% of rejection sampling's average accuracy while using 35.9% of the estimated compute, and it transfers to new math data, another teacher, and STEM problems.
- Translates cheaper reverse chain-of-thought into verified forward-style reasoning traces.
- Trained on a small set of paired reverse and rejection-sampled forward responses.
- Gains up to 5.82 points over semantic suppression across six math benchmarks.
- Two 4B students keep 95.0% of rejection-sampling accuracy at 35.9% of estimated compute.
- Transfers without retraining to a new dataset, teacher, and STEM tasks.
From the original article
1 min read at huggingface.coReasoning distillation requires chain-of-thought (CoT) data with correct answers, but obtaining such data efficiently remains challenging. Forward CoT generation (FCG) followed by rejection sampling (RS) produces verified, question-only responses, yet may require many teacher calls for difficult problems.
By Zinan Tang, Zhuoshi Pan, Kairui Chen · huggingface.co
A short excerpt; the article and its images belong to the publisher. Summary, key facts and assessment above are AI-written from the article and can be wrong.