Spectral-Driven Adaptive-Rank Optimizer for Large Language Model Reinforcement Learning with Verifiable Rewards
Organizations: Gaoling School of Artificial Intelligence, Renmin University of China
Abstract
Reinforcement learning with verifiable rewards (RLVR) is reshaping how large language models (LLMs) acquire reasoning capabilities, yet the optimizers used remain largely inherited from teacher-forcing training. This raises a fundamental question: does RLVR induce a distinct optimization geometry that calls for a different optimizer? We investigate this question through the spectral structure of LLM gradients and find that gradient energy progressively concentrates into a shrinking effective subspace, which is characteristic of RLVR, as it is absent under supervised fine-tuning (SFT) and is largely removed when the verifiable reward is replaced by a random one. Moreover, the rank required to capture a fixed fraction of gradient energy steadily decreases over training, revealing that the effective dimensionality of useful updates itself evolves with training. These findings motivate a new optimization principle: rather than prescribing the update rank externally, an optimizer should adapt its rank to the evolving spectral structure of the gradient. We introduce Ares (Adaptive Rank-Effective Spectral Optimization), which estimates the effective update rank from the gradient's spectral energy distribution, tracks its evolution, and confines the gradient signal driving each update to the resulting low-rank subspace. Across LLMs of different scales, Ares improves accuracy over widely used optimizers and generalizes across RLVR algorithms and tasks. Results suggest that effective optimization may require optimizers to be designed around the geometry and dynamics of the training process. We release the code and data at https://github.com/RUCAIBox/Ares.
Figures & tables
| Optimizer | Update rule | Low-rank | Rank | Spectral-driven |
| SGD | First-order momentum | No | — | No |
| RMSprop | Diagonal adaptive | No | — | No |
| Adam [ 16 ] | Diagonal adaptive | No | — | No |
| AdamW [ 22 ] | Diagonal adaptive | No | — | No |
| Adafactor [ 29 ] | Factored second moment | No | — | No |
| Lion [ 6 ] | Sign momentum | No | — | No |
| Optimizer | AIME24 | AIME25 | AIME26 | AMC23 | MATH500 | Minerva | OlyBench | Avg. |
| Qwen3-1.7B-Base | ||||||||
| AdamW | 7.9 | 4.2 | 5.0 | 37.8 | 59.2 | 15.5 | 25.7 | 22.2 |
| SGD | 0.4 | 0.0 | 0.4 | 2.8 | 6.3 | 2.9 | 1.3 | 2.0 |
| RMSprop | 0.0 | 0.0 | 0.0 | 13.1 | 33.5 | 4.8 | 9.6 | 8.7 |
| Adam | 7.5 | 6.3 | 4.6 | 40.0 | 59.3 | 15.3 | 25.1 | 22.6 |
| GaLore | 8.3 | 6.3 | 6.3 | 37.8 | 58.9 | 16.9 | 26.1 | 22.9 |
| Optimizer | AIME24 | AIME25 | AIME26 | AMC23 | MATH500 | Minerva | OlyBench | Avg. |
| Ares (Ours) | 15.0 | 10.8 | 10.0 | 43.4 | 61.6 | 20.4 | 29.4 | 27.2 |
| w/o Rank Momentum | 10.8 | 7.5 | 5.8 | 41.9 | 57.9 | 18.2 | 26.7 | 24.1 |
| w/o Dynamic Rank | 12.9 | 9.2 | 5.0 | 48.1 | 61.9 | 19.7 | 29.5 | 26.6 |
| w/o Low Rank | 0.0 | 0.0 | 0.4 | 4.4 | 9.1 | 2.9 | 2.0 | 2.7 |
| w/o Spectral Normalize | 10.8 | 6.7 | 5.0 | 48.1 | 61.9 | 20.0 | 28.0 | 25.8 |
| w/o Adaptive LR | 10.0 | 5.0 | 4.2 | 39.4 | 60.5 | 17.4 | 25.9 | 23.2 |
Appendix figures & tables10 assets
Supplementary material from the paper’s appendix.
Appendix
| Optimizer | AIME24 | AIME25 | AIME26 | AMC23 | MATH500 | Minerva | OlyBench | Avg. |
| Qwen3-1.7B-Base | ||||||||
| Ares (Ours) | 15.0 | 10.8 | 10.0 | 43.4 | 61.6 | 20.4 | 29.4 | 27.2 |
| w/o Rank Momentum | 10.8 | 7.5 | 5.8 | 41.9 | 57.9 | 18.2 | 26.7 | 24.1 |
| w/o Dynamic Rank | 12.9 | 9.2 | 5.0 | 48.1 | 61.9 | 19.7 | 29.5 | 26.6 |
| w/o Low Rank | 0.0 | 0.0 | 0.4 | 4.4 | 9.1 | 2.9 | 2.0 | 2.7 |
| w/o Spectral Normalize | 10.8 | 6.7 | 5.0 | 48.1 | 61.9 | 20.0 | 28.0 | 25.8 |
| Optimizer | AIME24 | AIME25 | AIME26 | AMC23 | MATH500 | Minerva | OlyBench | Avg. |
| Qwen3-1.7B-Base | ||||||||
| Exp | 15.0 | 10.8 | 10.0 | 43.4 | 61.6 | 20.4 | 29.4 | 27.2 |
| Linear | 10.8 | 7.5 | 5.8 | 41.9 | 57.9 | 18.2 | 26.6 | 24.1 |
| Threshold | 7.1 | 3.8 | 4.6 | 39.1 | 59.0 | 16.8 | 25.8 | 22.3 |
| Qwen3-4B-Base | ||||||||
| Exp | 25.4 | 17.5 | 20.0 | 65.0 | 73.4 | 28.2 | 40.3 | 38.5 |
| Methods | Enginneering | Philosophy | CS | Physics | Math | Economics | Health | Avg. |
| Qwen3-1.7B-Base | ||||||||
| AdamW | 25.6 | 29.3 | 35.0 | 26.6 | 27.5 | 48.9 | 34.8 | 32.5 |
| GaLore | 27.6 | 27.5 | 36.0 | 25.5 | 26.6 | 47.7 | 35.8 | 32.4 |
| Ares (Ours) | 37.3 | 28.7 | 38.7 | 26.9 | 27.5 | 48.9 | 36.2 | 34.9 |
| Qwen3-4B-Base | ||||||||
| AdamW | 38.7 | 40.3 | 54.9 | 56.7 | 65.4 | 65.6 | 52.6 | 53.5 |
| AIME24 | AIME25 | AIME26 | AMC23 | MATH500 | Minerva | OlyBench | Avg. | |
| Qwen3-1.7B-Base | ||||||||
| RLOO w/ AdamW | 10.4 | 5.0 | 4.6 | 39.4 | 60.1 | 17.6 | 26.5 | 23.4 |
| RLOO w/ Ares | 12.5 | 11.7 | 8.8 | 48.8 | 61.8 | 19.7 | 29.6 | 27.6 |
| +2.1 | +6.7 | +4.2 | +9.4 | +1.7 | +2.1 | +3.1 | +4.2 | |
| REINFORCE++ w/ AdamW | 7.5 | 2.1 | 2.9 | 36.7 | 61.4 | 17.0 | 26.4 | 22.0 |
| REINFORCE++ w/ Ares | 10.0 | 6.3 | 4.6 | 43.4 | 61.5 | 19.3 | 29.0 | 24.9 |
| AIME24 | AIME25 | AIME26 | AMC23 | MATH500 | Minerva | OlyBench | Avg. | |
| AdamW | 17.0 | 21.0 | 17.3 | 58.3 | 69.2 | 26.5 | 34.3 | 34.8 |
| Ares | 19.9 | 20.3 | 18.1 | 58.3 | 69.5 | 26.7 | 34.3 | 35.3 |
RL Fine-Tuning
Reinforcement learning fine-tuning adapts pretrained models and policies through task rewards, preference feedback, or environment interaction.
See all 233 papers