NeFut Logo NeFut
Admin Login

[CS.AI] Woodpecker Distillation: Weak Models Diagnose Reasoning Bugs in Strong Models

Published at: 2026-08-07 22:00 Last updated: 2026-08-08 01:08
#Machine Learning #Math #LLM

Large language models often fail on reasoning tasks despite possessing the capability to solve them. We argue that many such failures arise from localized reasoning bugs in intermediate steps rather than from global incompetence. We show that these bugs are frequently repairable: inserting a short patch generated by a weak probe model after the same strong-model reasoning prefix can redirect the trajectory toward a correct solution. However, this corrective effect is not reliably internalized by directly fine-tuning on weak patches or repaired trajectories, suggesting that the useful signal lies not in the intervention text itself, but in how it reshapes the model's future reasoning distribution. We therefore propose Woodpecker Distillation, a weak-to-strong training framework that learns from contrastive local interventions. Our method contrasts successful and unsuccessful weak-model patches at the same prefix, constructs a corrective teacher distribution from their induced future token predictions, and distills this signal into the strong model. Experiments on mathematical reasoning benchmarks show that Woodpecker Distillation consistently improves strong-model performance and outperforms direct imitation baselines.

Blogger's Review: Woodpecker Distillation provides a novel perspective on improving the reasoning capabilities of large language models by diagnosing local reasoning bugs in strong models through weak models and leveraging contrastive learning to enhance strong model performance, which is of significant reference value for the development of language models.

Original Source: https://arxiv.org/abs/2608.05168

[h] Back to Home