AWS: Self-Distilled Reasoning restores Amazon Nova 2 math accuracy from 6% to about 68% after fine-tuning
AWS has introduced Self-Distilled Reasoning (SDR), a supervised fine-tuning technique for Amazon Nova 2 models when training data lacks a chain of thought. The method uses the base Nova 2 Lite model to generate chain-of-thought traces, achieves over a 6.5% relative improvement in target performance, and restores overall math accuracy from 6% to about 68%, tested on three benchmarks without human annotation.
This article was generated using artificial intelligence from primary sources.
What is Self-Distilled Reasoning and what problem does it solve?
AWS has introduced Self-Distilled Reasoning (SDR), a technique for supervised fine-tuning (further training a model on labeled examples) of Amazon Nova 2 models in situations where the training data lacks a chain of thought — the reasoning steps a model goes through before its final answer. Such data is common in practice: companies have question-answer pairs without a recorded reasoning path, which makes it harder to train a model to reason on its own.
Why does vanilla fine-tuning wreck overall performance?
When a model is fine-tuned directly on narrow data without reasoning traces (vanilla SFT), the model’s overall math performance drops from the base model’s 70% to just 6% — a phenomenon known as catastrophic forgetting, the loss of previously learned abilities due to overly narrow subsequent training. The model becomes good at the narrow task but loses the math problem-solving ability it previously had.
SDR uses the model to generate its own training data
The solution AWS proposes uses the base Nova 2 Lite model to autonomously generate chain-of-thought traces for existing training data that lacks them, without needing human annotation. The enriched data is then used to fine-tune the target model. Result: SDR achieves over a 6.5% relative improvement in performance on the target task while simultaneously restoring overall math accuracy from 6% to about 68% — a difference that shows how much preserving reasoning ability during fine-tuning reduces collateral damage to the model’s other skills.
Tested on three different benchmarks
AWS tested the method on three benchmarks from different domains: MedMCQA (medical questions), CoCoHD, and Invoice-OCR (invoice processing). The consistency of results across three unrelated tasks suggests SDR is not a narrow solution for one type of data, but a general approach to fine-tuning when reasoning traces are simply missing from the original data.
Frequently Asked Questions
- What is supervised fine-tuning (SFT)?
- Supervised fine-tuning is additional supervised training of an already-trained model on a specific set of labeled examples so it performs a given task better.
- What is catastrophic forgetting and how does SDR address it?
- Catastrophic forgetting is the loss of a model's general ability after narrow fine-tuning; with vanilla SFT, Amazon Nova 2's overall math performance dropped from 70% to 6%, while the SDR technique restores it to about 68%.
- How does Self-Distilled Reasoning generate training data?
- SDR uses the base Amazon Nova 2 Lite model to autonomously generate chain-of-thought traces for existing training data that lacks them, without needing human annotation.
📬 AI news in your inbox
A daily digest built your way — pick topics, sources and cadence. One-click unsubscribe.
Related news
Anthropic: Claude Code v2.1.217 fixes an MCP memory leak and introduces a 20-concurrent-subagent limit
arXiv:2607.16122: CRAFT Method Clusters Grading Rubrics to Reveal Weaknesses in Language Models
Anthropic: Claude Code v2.1.215 and v2.1.216 Fix Performance, an OAuth Error, and Agent Worktree Isolation