on-policy self-distillation

Teaching Foundation Models New Skills Without Breaking Them

Teaching foundation models new skills without breaking what they already know is one of the hardest challenges in AI.

3 min readTowards Data Science

Teaching foundation models new skills without breaking them is one of the hardest problems in AI right now, and a new reinforcement learning technique deserves serious attention for how it approaches the challenge. The method, on-policy self-distillation, offers a path to post-training that preserves what a model already knows while teaching it something new, an outcome that has proven elusive with conventional fine-tuning. This matters because the practical ceiling on what these models can do is often set not by their architecture, but by how brittle they become after we try to improve them. When AI Meets Grid Puzzles: Where Spatial Reasoning Still Falls Short showed us that even simple spatial tasks can break models that otherwise perform well on language benchmarks. The pattern is consistent: narrow improvements often come at the cost of broad capability loss.

What makes on-policy self-distillation different is that it lets the model learn from its own best outputs during the training process, rather than forcing external labels or synthetic data that may conflict with its existing knowledge. This is not a theoretical curiosity, it directly addresses a bottleneck that anyone building on top of foundation models has felt. When you try to teach a model a specialized skill, say extracting entities from messy text, the standard approach often degrades its general reasoning. Make LLM Reasoning Your NER Engine in Just a Few Steps demonstrated that with careful pipeline design, you can get strong results without retraining the model at all. The distillation approach takes that logic one step further: it embeds the new skill into the model itself without triggering the forgetting problem.

The most practical implication is this: we may soon be able to treat foundation models as genuinely adaptable tools rather than fixed black boxes that we prompt around. The reinforcement learning loop in this method rewards the model for outputs that improve on its own prior behavior, which means the model is not being forced into a rigid new pattern, it is being guided to extend its own capabilities. That is fundamentally different from supervised fine-tuning, where the model must learn to mimic a dataset that may not align with its internal representations. This AI built a coding tool largely on its own showed that models can already generate new tools autonomously, but those tools still need to be compatible with the model's existing knowledge. On-policy distillation could make that compatibility much easier to achieve.

The specific detail to watch is whether this method scales to large, multi-task models without introducing new failure modes. The research is promising, but the real test will come when someone tries to teach a general-purpose model a dozen new skills simultaneously and checks whether it still solves the grid puzzles it handled before. If it does, post-training stops being a damage-control exercise and becomes an actual upgrade path.

From Towards Data Science

Teaching foundation models new skills without breaking them is hard. A new reinforcement learning-based approach might have the power to change that.

The post Could On-Policy Self-Distillation Revolutionize Post-Training? appeared first on Towards Data Science.

Read the original at Towards Data Science