Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

LLMZero: Discovering Adaptive Training Strategies for RL Post-Training via LLM Agents

About

RL post-training strategies are dataset-dependent and reveal a recurring empirical pattern: capacity parameters accumulate monotonically across stages, while regularization parameters predominantly oscillate in response to shifting training dynamics. This distinction matters because fixed schedules commit all parameters to fixed trajectories and therefore cannot express the non-stationary exploration-exploitation tradeoffs that regularization must track; the principle provides actionable design rules for multi-stage training. We discover this through LLMZero, a system where LLM agents search over training trajectories via tree search, diagnosing pathologies at each checkpoint and proposing coordinated multi-parameter transitions. Across 4 diverse GRPO tasks, LLMZero discovers strategies that improve over the base model by 9% to 140% relative and over grid search by 6% to 15% relative, consistently outperforming random search and the skill-based agent. The structural principle transfers across tasks, providing an explanation for why discovered strategies take qualitatively different forms yet share similar parameter dynamics.

Haoyang Fang, Wei Zhu, Boran Han, Alex Zhang, Zhenyu Pan, Shuo Yang, Shuai Zhang, Jiading Gai, Peng Tang, Cuixiong Hu, Xuan Zhu, Huzefa Rangwala, George Karypis, Bernie Wang• 2026

Related benchmarks

TaskDatasetResultRank
Biomedical Question AnsweringPaperSearchQA (test)
Bio Score42.6
6
chemistry reasoningChemCoT (test)
Und Score69.8
6
Multi-Discipline Scientific ReasoningWildSci (test)
Sci Score58.5
6
Music AnalysisSSMR Music (test)
Scale Score94.4
6
Showing 4 of 4 rows

Other info

Follow for update