Actor-Curator: Automated RL Curriculum Learning
- Actor-Curator is a fully automated curriculum learning framework that coordinates an actor LLM and a neural curator using a non-stationary bandit formulation.
- It leverages online stochastic mirror descent for curator updates, ensuring unbiased utility estimates and providing theoretical regret guarantees.
- Empirical evaluations on complex reasoning benchmarks show significant gains in stability and efficiency, achieving target accuracy with up to 80% fewer training steps.
Actor-Curator (AC) is a fully automated curriculum learning framework designed for reinforcement learning (RL) post-training of LLMs. The method introduces a co-adaptive interaction between two components: the actor, a policy model (typically a pretrained LLM), and the curator, a neural network that selects training problems from a large, heterogeneous problem bank. By formulating the curriculum selection as a non-stationary stochastic bandit problem, Actor-Curator directly optimizes for expected policy improvement, is robust to large unstructured problem banks, and features theoretical regret guarantees under partial feedback. Actor-Curator demonstrates significant empirical gains in training stability, efficiency, and final performance over baseline sampling or hand-engineered curricula, with particular efficacy on complex reasoning benchmarks (Gu et al., 24 Feb 2026).
1. Problem Formulation and Bandit Perspective
Actor-Curator is situated in the context of RL post-training on vast, heterogeneous task distributions. The setup involves:
- Problem bank : a large, diverse set of problems (e.g., math, algorithmic, symbolic reasoning tasks).
- Actor : the RL policy, such as a partially or fully pretrained LLM; for an input problem , generates a solution .
- Curator : a trainable selector with parameters , assigning scores which induce a sampling distribution .
At each iteration :
- The curator samples a mini-batch ;
- The actor is rolled out on each 0;
- The actor is updated to 1 using a reward model and gathered trajectories.
The overarching objective is to maximize expected reward:
2
The per-iteration gain is 3.
Bandit formulation: Each problem 4 is conceptualized as a non-stationary bandit arm with utility:
5
where 6 is the policy advantage. Importance sampling enables unbiased utility estimation even under partial batchwise feedback.
2. Curator Optimization via Online Stochastic Mirror Descent
The curator’s update is grounded in online stochastic mirror descent (OSMD):
- Tabular OSMD: For the tabular regime, 7 is updated by exponentiated gradient:
8
where 9 is an unbiased utility estimate.
- Neural function approximation: When using a neural curator 0, the objective is to minimize
1
- Two-stage sampling: In practical large-scale settings, a candidate set 2 is first sampled, followed by subsampling according to 3. The two-stage unbiased estimator and loss function account for proposal bias and importance weighting.
- Stabilization via proximal clipping: A Proximal Clipping Objective (PCO), inspired by PPO, stabilizes the curator’s update by clipping the importance-weighted policy ratio, providing robustness to volatility in the curation distribution.
3. Theoretical Guarantees
Actor-Curator provides rigorous guarantees under mild assumptions:
- Unbiasedness: Both single- and two-stage bandit estimators are proven to be unbiased: 4.
- Dynamic regret bound: Under OSMD with negative-entropy mirror map and partial-feedback estimation, the best-in-hindsight dynamic regret is
5
where 6 quantifies non-stationarity of per-problem utilities.
- Assumptions: The guarantees assume bounded utility functions, estimator unbiasedness, a minimum inclusion probability 7, entropy regularization (8), and mild Lipschitzness.
The proof approach decomposes the time horizon into near-stationary blocks, uses mirror descent inequalities, and strategically balances estimation error with non-stationary drift via block and step-size optimization.
4. Algorithmic Structure
The Actor-Curator algorithm (Algorithm 1 in the paper) operates as follows:
- Initialize the problem bank, actor, reward model, neural curator, and proposal distribution.
- For each training step 9:
- Candidate sampling: Draw candidate problems 0 of size 1.
- Selection: Subsample batch 2 of size 3 from 4 using the curator.
- Data collection: Run 5 policy rollouts per selected problem, compute rewards.
- Actor update: Update 6 using collected data (e.g., guided by GSPO/GRPO).
- Utility estimation: Compute per-problem advantage estimates 7, then unbiased utility 8 for each 9.
- Curator update: Update curator via gradient step on 0.
Utility computations require both pre- and post-update policies and leverage importance sampling for unbiasedness.
5. Empirical Evaluation
Experiments address diverse curriculum learning settings for LLM post-training:
- Benchmarks:
- Symbolic: Countdown (and hard), Zebra (and hard)
- Visual: ARC-1D (and ARC-hard)
- Mathematics: MATH500, AIME2024
- Setup:
- Actor: Qwen2.5-3B or Llama3.2-3B-IT, trained with GSPO.
- Curator: Qwen3-0.6B.
- Metrics: Held-out test accuracy (peak within 100 steps).
- Results:
- ARC-1D: Uniform 26.7% 1 Actor-Curator 36.4% (+30.5%)
- AIME24: Uniform 23.3% 2 Actor-Curator 30.0% (+28.6%)
- Largest gains observed on hard subsets (e.g., ARC-hard +59%).
- Speed: Actor-Curator achieves target test accuracy in 380% fewer training steps compared to uniform sampling.
- Ablation studies: Removing the policy-improvement target or OSMD in favor of regression degrades performance; method remains robust to actor update mode and curator/candidate batch sizes.
6. Strengths, Limitations, and Future Directions
Strengths:
- Direct optimization of a policy-improvement-driven curriculum objective
- Explicit handling of bandit feedback, exploration–exploitation, and non-stationarity
- Scalability to large, unstructured task banks via neural curation
- Theoretical guarantees with minimal assumptions
- Empirical gains in convergence and final accuracy across disparate reasoning tasks
Limitations:
- Efficacy relies on the existence of a reliable reward model (critical for verifiable correctness tasks)
- Curator training incurs a modest (9–15%) wall-clock overhead; offset by resultant speedups
- Stability of actor updates is necessary; RL collapse risk persists
- Neural function approximation only approximately captures utility cross-interference between problems
Future Work:
- Extension to continuous/environment-parameterized task spaces
- Joint adaptation of actor update hyperparameters using the bandit signal
- Theoretical quantification of cross-problem interference with deep nets
- Integration with search-based rollout methods (e.g., Tree-of-Thoughts) for further performance improvement
Actor-Curator establishes a practical, theoretically sound methodology for scalable, automated curriculum discovery in RL-finetuning of LLMs, delivering faster convergence and higher performance on reasoning-intensive benchmarks (Gu et al., 24 Feb 2026).