AceGRPO: Adaptive Curriculum Enhanced Group Relative Policy Optimization for Autonomous Machine Learning Engineering
Abstract
Autonomous Machine Learning Engineering (MLE) requires agents to perform sustained, iterative optimization over long horizons. While recent LLM-based agents show promise, current prompt-based agents for MLE suffer from behavioral stagnation due to frozen parameters. Although Reinforcement Learning (RL) offers a remedy, applying it to MLE is hindered by prohibitive execution latency and inefficient data selection. Recognizing these challenges, we propose AceGRPO with two core components: (1) Evolving Data Buffer that continuously repurposes execution traces into reusable training tasks, and (2) Adaptive Sampling guided by a Learnability Potential function, which dynamically prioritizes tasks at the agent's learning frontier to maximize learning efficiency. Trained with AceGRPO, Ace-35B substantially improves over its base model, achieving a 100% valid submission rate on MLE-Bench-Lite and approaching or surpassing larger frontier models on key metrics. These results demonstrate the effectiveness of adaptive, execution-grounded training for building capable and reliable long-horizon MLE agents.
Then back it, or bet against it.
Related papers
Open the market on this paper to see 7 more related papers.