Efficient Reinforcement Finetuning via Adaptive Curriculum Learning
By: Taiwei Shi , Yiyang Wu , Linxin Song and more
Potential Business Impact:
Teaches computers math faster and better.
Reinforcement finetuning (RFT) has shown great potential for enhancing the mathematical reasoning capabilities of large language models (LLMs), but it is often sample- and compute-inefficient, requiring extensive training. In this work, we introduce AdaRFT (Adaptive Curriculum Reinforcement Finetuning), a method that significantly improves both the efficiency and final accuracy of RFT through adaptive curriculum learning. AdaRFT dynamically adjusts the difficulty of training problems based on the model's recent reward signals, ensuring that the model consistently trains on tasks that are challenging but solvable. This adaptive sampling strategy accelerates learning by maintaining an optimal difficulty range, avoiding wasted computation on problems that are too easy or too hard. AdaRFT requires only a lightweight extension to standard RFT algorithms like Proximal Policy Optimization (PPO), without modifying the reward function or model architecture. Experiments on competition-level math datasets-including AMC, AIME, and IMO-style problems-demonstrate that AdaRFT significantly improves both training efficiency and reasoning performance. We evaluate AdaRFT across multiple data distributions and model sizes, showing that it reduces training time by up to 2x and improves accuracy by a considerable margin, offering a more scalable and effective RFT framework.
Similar Papers
SPaRFT: Self-Paced Reinforcement Fine-Tuning for Large Language Models
Machine Learning (CS)
Teaches computers to learn smarter, faster, with less data.
On the Suitability of Reinforcement Fine-Tuning to Visual Tasks
CV and Pattern Recognition
Makes AI better at understanding pictures.
Visual-RFT: Visual Reinforcement Fine-Tuning
CV and Pattern Recognition
Teaches computers to understand pictures better.