RSI-Opt: Recursive Self-Improvement Optimizer Agents For Model Training
Organizations: Purdue University · Independent Researcher
Abstract
Conventional neural-network training typically follows a predefined optimization recipe, motivating a LLM agent that monitors training dynamics and adjusts optimizer settings as training progresses, but a fixed agent policy does not accumulate experience across runs. We introduce RSI-Opt, a plug-and-play Recursive Self-Improvement Optimizer Agents which evolves the knowledge that guides this process: a prompt, a playbook of procedures, a library of tested optimizer code, and a decision cadence. A meta-agent rewrites this policy using evidence from earlier runs, and an archive retains evaluated policies for further evolution. Across 5 generations on three training tasks, the meta-agents develop reusable procedures and code and conduct their own CPU proxy experiments. RSI-Opt substantially improves optimizer-agent performance, with representative raw-space summary -scores increasing from to on the evolution suite with Opus 5.5 and from to on held-out tasks with GPT-5.5 through cross-model policy transfer. On the Burgers PDE, it reduces mean relative error by approximately 2.7-fold with Opus 5.5 and 8.8-fold with GPT-6.1 over the fixed-prompt agent baseline, although PDE transfer varies across LLMs and tasks. Controlled comparisons show that the evolved policy combines reusable optimization knowledge with adaptation to each task: initial decisions capture most observed gains on seen tasks, while continued adaptation improves outcomes on unseen tasks. These findings demonstrate how procedures and executable tools accumulated across training runs can improve optimizer-agent performance and support transfer to scientific machine learning.
Figures & tables
| Group | Task | Static recipe | Metric | Static result |
|---|---|---|---|---|
| Training suite | ResNet-18 / CIFAR-10 | SGD-Nesterov, one-cycle | test acc. | 94.60 0.16 |
| GPT-34M / FineWeb | AdamW, warmup-cosine | val. loss | 4.480 0.008 | |
| Flow UNet / MNIST | AdamW, cosine, EMA | EMA val. MSE | 0.1700 0.0004 | |
| Held-out ML | ViT / CIFAR-100 | AdamW, warmup-cosine | test acc. | 55.84 0.34 |
| LSTM / WikiText-2 | Adam, step decay | test loss | 5.386 0.064 | |
| ResNet-18 / noisy CIFAR-10 | AdamW, warmup-cosine | test acc. | 73.76 1.71 |
| Model | Method | Train | Held-out ML | Held-out PDE | All held-out |
|---|---|---|---|---|---|
| – | Static, tuned LR (oracle) | +3.46 | +0.49 | +1.11 | +0.80 |
| Opus 5.5 | Single-agent | +10.48 | +8.71 | +1.72 | +5.22 |
| RSI-Opt | +14.33 | +8.23 | +1.51 | +4.87 | |
| wins | 3/3 | 2/3 | 2/3 | 4/6 | |
| GPT-5.5 | Single-agent | +4.33 | +1.18 | -1.90 | -0.36 |
| RSI-Opt | +14.10 | +6.54 | -0.32 | +3.11 |
| Task | Opus 5.5 | GPT-5.5 | GPT-6.1 |
|---|---|---|---|
| ResNet-18 / CIFAR-10 | +0.23 | +0.19 | +0.45 |
| GPT-34M / FineWeb | +0.025 | +0.099 | +0.005 |
| Flow UNet / MNIST | +0.0028 | +0.0063 | +0.0038 |
| Training suite: mean | +3.85 | +9.77 | +4.31 |
| ViT / CIFAR-100 | -1.10 | +4.42 | +6.38 |
| LSTM / WikiText-2 | +0.001 | -0.030 | -0.124 |
| Condition | Train | Held-out ML |
|---|---|---|
| Static recipe | +0.00 | +0.00 |
| Static, tuned LR (oracle) | +3.46 | +0.49 |
| Frozen library recipe (no agent) | +9.71 | -21.80 |
| Single-agent: first change, then frozen | +10.72 | +6.35 |
| Single-agent: one decision | +9.43 | +6.12 |
| Single-agent: full agent (20 decisions) | +10.48 | +8.71 |
| Policy | Train | Held-out ML |
|---|---|---|
| Single-agent (fixed prompt) | +10.48 | +8.71 |
| RSI-Opt (full policy) | +14.33 | +8.23 |
| library | +14.45 | +8.00 |
| playbook | +13.60 | +8.06 |
| evolved prompt | – | – |
| Component | Single-agent | RSI-Opt |
| Muon / orthogonalised update | 35% | 79% |
| Kronecker-factored (Shampoo / SOAP) | 15% | 10% |
| Second-order / L-BFGS-like | 11% | 10% |
| Clipping or update-size control | 60% | 79% |
| Per-update loss feedback (ctx.loss) | 20% | 10% |
| Uses a module of the evolved library | 0% | 82% |
| Model | Method | Input tok. (M) | Cached (M) | Output tok. (k) | Changing decisions |
|---|---|---|---|---|---|
| Opus 5.5 | Single-agent | 5.63 | 5.63 | 68 | 9.1 |
| RSI-Opt | 4.02 | 4.02 | 40 | 3.5 | |
| GPT-5.5 | Single-agent | 8.90 | 8.49 | 79 | 11.6 |
| RSI-Opt | 4.61 | 4.24 | 42 | 5.0 | |
| GPT-6.1 | Single-agent | 6.99 | 6.99 | 58 | 13.9 |
| RSI-Opt | 3.65 | 3.65 | 24 | 4.2 |
Large Language Model-Guided Optimization
Large Language Model-Guided Optimization uses language models to propose, steer, and refine solutions to hard search and design problems.
See all 608 papers