Beyond Templates: Dynamic Adaptation of Reasoning Demonstrations via Feasibility-Aware Exploration
By: Yong Wu , Weihang Pan , Ke Li and more
Potential Business Impact:
Teaches small computers to think like big ones.
Large language models (LLMs) have shown remarkable reasoning capabilities, yet aligning such abilities to small language models (SLMs) remains a challenge due to distributional mismatches and limited model capacity. Existing reasoning datasets, typically designed for powerful LLMs, often lead to degraded performance when directly applied to weaker models. In this work, we introduce Dynamic Adaptation of Reasoning Trajectories (DART), a novel data adaptation framework that bridges the capability gap between expert reasoning trajectories and diverse SLMs. Instead of uniformly imitating expert steps, DART employs a selective imitation strategy guided by step-wise adaptability estimation via solution simulation. When expert steps surpass the student's capacity -- signaled by an Imitation Gap -- the student autonomously explores alternative reasoning paths, constrained by outcome consistency. We validate DART across multiple reasoning benchmarks and model scales, demonstrating that it significantly improves generalization and data efficiency over static fine-tuning. Our method enhances supervision quality by aligning training signals with the student's reasoning capabilities, offering a scalable solution for reasoning alignment in resource-constrained models.
Similar Papers
DART: Distilling Autoregressive Reasoning to Silent Thought
Computation and Language
Makes AI think faster without losing answers.
Discovery and Reinforcement of Tool-Integrated Reasoning Chains via Rollout Trees
Computation and Language
Teaches computers to use tools for harder problems.
Teaching LLMs According to Their Aptitude: Adaptive Reasoning for Mathematical Problem Solving
Computation and Language
Teaches computers to solve math problems better.