Answer-Conditioned Chains of Thought Degrade Verifiable-Reasoning Distillation in Large Language Models

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, quick

Summary

A study reveals that a common technique for distilling large language model (LLM) reasoning ability, which involves sampling chains of thought and fine-tuning on correct ones, is degraded by answer-conditioning. When initial sampling fails, showing the generator the gold answer and requesting a chain to reach it creates training data damage that correctness filters cannot detect. A controlled experiment, fixing the generator, problem set, and filter, found that training a strong instruction-tuned reasoning model on its own answer-conditioned chains sharply lowered its verifiable-reasoning accuracy. This loss reached approximately 27 points on the hardest competition problems. The mechanism involves backward rationalization from the shown answer, evidenced by early final-answer statements. This harm is a data property, not a generator property, and transfers across teacher families, localized to the "rationalize-toward" instruction.

Key takeaway

For machine learning engineers distilling LLM reasoning, you should prioritize generating answer-blind chains of thought. Relying on answer-conditioned generation, where the gold answer is provided, significantly degrades the quality of training data by promoting backward rationalization, which correctness filters cannot identify. This can sharply lower your model's verifiable-reasoning accuracy, with observed drops of up to 27 points on complex problems.

Key insights

Answer-conditioned chain of thought generation degrades LLM reasoning distillation by inducing backward rationalization.

Principles

In practice

Topics

Best for: Research Scientist, AI Engineer, AI Scientist, Machine Learning Engineer, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.