Low-rank adaptation (LoRA) is commonly used in both supervised fine-tuning (SFT) and reinforcement learning (RL), often under similar optimizer settings. Yet, how the different dynamics of the two stages manifest in the low-rank increment that LoRA optimizes remains unclear. Tracking the singular spectrum and subspaces of LoRA updates on mathematical reasoning tasks, we find that RL concentrates the spectrum at every learning rate, while SFT expands it at moderate learning rates; moreover, RL rotates the singular subspaces much faster than SFT at small learning rates. Interpolating between SFT and RL one ingredient at a time, we pinpoint group-mean baseline subtraction as the source of both contrasts. A quadratic multi-index model updated via LoRA unveils the underlying mechanism: group-mean baseline subtraction tilts the balance between two competing forces, coherent fitting and gradient noise, toward spectral concentration and faster subspace rotation, while forcing the risk-favorable learning rate to be low. Guided by these dynamics, we tailor LoRA updates to each stage: in SFT, we flatten the spectrum of the full update to spread adaptation across singular directions, whereas in RL, we flatten the part of the update outside the current LoRA increment span to promote subspace rotation. Experiments on mathematical reasoning tasks across multiple models show that the respective configurations for SFT and RL yield the best performance among the compared methods in most evaluated settings.
Figures & tables
Figure 1: On GSM8K under B0=0 initialization, GRPO and RFT-B exhibit decreasing effective rank across all learning rates shown. In contrast, SFT and RFT exhibit an increasing trend from approximately step 25 at intermediate learning rates. Columns correspond to different learning rates; the top and bottom rows compare SFT/RFT and GRPO/RFT-B, respectively. Thin curves show individual models, with effective rank averaged over LoRA modules; thick curves show the mean across all models.
Figure 2: On GSM8K under B0=0 initialization, GRPO and RFT-B exhibit a faster decrease in left-subspace overlap than SFT and RFT at small learning rates, indicating faster subspace rotation. This difference narrows as the learning rate increases. The top and bottom rows show the right-subspace overlap ovlpR(5,t) and left-subspace overlap ovlpL(5,t) , respectively, between the current checkpoint at step t and the reference checkpoint at step 5 . Columns correspond to different learning rates; each panel compares SFT, RFT, RFT-B, and GRPO. Thin curves show individual models, with overlap averaged over LoRA modules; thick curves show the mean across all models.
SFT
RL
A direction
B direction
Llama-3.2-1B
Llama-3.2-3B
A direction
B direction
Llama-3.2-1B
Llama-3.2-3B
Adam
Adam
22.24±0.33
42.99±0.38
Adam
Adam
7.61±1.24
42.36±3.25
Adam
M(Adam)
20.90±0.76
44.40±0.93
Adam
M(Adam)
9.67±1.22
46.00±0.77
Adam
M(SGD)
22.82±0.12
45.72±0.61
Adam
M⊥(Adam)
10.00±1.36
47.66±1.85
Table 1: LoRA optimization on GSM8K with Llama-3.2-1B and Llama-3.2-3B. The dynamics-informed configurations in the last row achieve the highest mean accuracies for both SFT (left) and RL (right). Methods are specified by the update directions for the two LoRA factors: Adam and SGD denote the corresponding base optimizer directions ( DtA or DtB ), and M denotes the norm-preserving spectral flattening applied to the full direction. For RL, M⊥(Adam) denotes PtDtB+M((I−Pt)DtB) with an Adam base direction DtB , which preserves the within-subspace component while transforming only the orthogonal component. Values show mean test accuracy (%) across three seeds, with standard errors in subscripts. For each configuration, results are reported at the learning rate with the highest mean accuracy in a completed sweep. Bold indicates the highest reported mean in each column.
Appendix figures & tables13 assets
Supplementary material from the paper’s appendix.
Appendix
Figure 3: On MATH under B0=0 initialization, GRPO and RFT-B exhibit decreasing effective rank across all learning rates shown. In contrast, SFT and RFT exhibit an increasing trend from approximately step 25 at intermediate learning rates. Columns correspond to different learning rates; the top and bottom rows compare SFT/RFT and GRPO/RFT-B, respectively. Thin curves show individual models, with effective rank averaged over LoRA modules; thick curves show the mean across all models.
Figure 4: On MATH under B0=0 initialization, GRPO and RFT-B exhibit a faster decrease in left-subspace overlap than SFT and RFT at small learning rates, indicating faster subspace rotation. This difference narrows as the learning rate increases. The top and bottom rows show the right-subspace overlap ovlpR(5,t) and left-subspace overlap ovlpL(5,t) , respectively, between the current checkpoint at step t and the reference checkpoint at step 5 . Columns correspond to different learning rates; each panel compares SFT, RFT, RFT-B, and GRPO. Thin curves show individual models, with overlap averaged over LoRA modules; thick curves show the mean across all models.
Figure 5: On GSM8K under A0=0 initialization, GRPO and RFT-B exhibit decreasing effective rank across all learning rates shown. In contrast, SFT and RFT exhibit an increasing trend from approximately step 25 at intermediate learning rates. Columns correspond to different learning rates; the top and bottom rows compare SFT/RFT and GRPO/RFT-B, respectively. Thin curves show individual models, with effective rank averaged over LoRA modules; thick curves show the mean across all models.
Figure 6: On MATH under A0=0 initialization, GRPO and RFT-B exhibit decreasing effective rank across all learning rates shown. In contrast, SFT and RFT exhibit an increasing trend from approximately step 25 at intermediate learning rates. Columns correspond to different learning rates; the top and bottom rows compare SFT/RFT and GRPO/RFT-B, respectively. Thin curves show individual models, with effective rank averaged over LoRA modules; thick curves show the mean across all models.
Figure 7: On GSM8K under A0=0 initialization, GRPO and RFT-B exhibit a faster decrease in right-subspace overlap than SFT and RFT at small learning rates, indicating faster subspace rotation. This difference narrows as the learning rate increases. The top and bottom rows show the right-subspace overlap ovlpR(5,t) and left-subspace overlap ovlpL(5,t) , respectively, between the current checkpoint at step t and the reference checkpoint at step 5 . Columns correspond to different learning rates; each panel compares SFT, RFT, RFT-B, and GRPO. Thin curves show individual models, with overlap averaged over LoRA modules; thick curves show the mean across all models.
Figure 8: On MATH under A0=0 initialization, GRPO and RFT-B exhibit a faster decrease in right-subspace overlap than SFT and RFT at small learning rates, indicating faster subspace rotation. This difference narrows as the learning rate increases. The top and bottom rows show the right-subspace overlap ovlpR(5,t) and left-subspace overlap ovlpL(5,t) , respectively, between the current checkpoint at step t and the reference checkpoint at step 5 . Columns correspond to different learning rates; each panel compares SFT, RFT, RFT-B, and GRPO. Thin curves show individual models, with overlap averaged over LoRA modules; thick curves show the mean across all models.
A direction
B direction
Llama-3.2-1B
Llama-3.2-3B
Adam
Adam
5.66±0.09
10.83±0.13
Adam
M(Adam)
5.93±0.14
10.53±0.14
Adam
M(SGD)
6.01±0.25
11.32±0.22
Appendix
Table 2: SFT optimization on MATH with Llama-3.2-1B and Llama-3.2-3B. The dynamics-informed configuration in the last row achieves the highest mean accuracies. Methods are specified by the update directions for the two LoRA factors: Adam and SGD denote the corresponding base optimizer directions ( DtA or DtB ), and M denotes the norm-preserving spectral flattening applied to the full direction. Values show mean test accuracy (%) across three seeds, with standard errors in subscripts. For each configuration, results are reported at the learning rate with the highest mean accuracy in a completed sweep. Bold indicates the highest reported mean in each model column.
A direction
B direction
Llama-3.2-1B
Llama-3.2-1B-Instruct
Llama-3.2-3B
Qwen2.5-3B
Adam
Adam
7.61±1.24
57.85±0.39
42.36±3.25
83.93±0.34
Adam
M(Adam)
9.67±1.22
57.97±0.18
46.00±0.77
84.46±0.38
Adam
M⊥(Adam)
10.00±1.36
58.23±0.72
47.66±1.85
84.51±0.62
Appendix
Table 3: RL optimization on GSM8K with Llama-3.2-1B, Llama-3.2-1B-Instruct, Llama-3.2-3B, and Qwen2.5-3B. The dynamics-informed configuration in the last row achieves the highest reported mean accuracy in each model column. Methods are specified by the update directions for the two LoRA factors: Adam denotes the corresponding base optimizer direction ( DtA or DtB ), and M denotes the norm-preserving spectral flattening applied to the full direction. M⊥(Adam) denotes PtDtB+M((I−Pt)DtB) with an Adam base direction DtB , which preserves the within-subspace component while transforming only the orthogonal component. Values show mean test accuracy (%) across three seeds, with standard errors in subscripts. For each configuration, results are reported at the learning rate with the highest mean accuracy in a completed sweep. Bold indicates the highest reported mean in each model column.
A direction
B direction
Llama-3.2-1B
Llama-3.2-1B-Instruct
Llama-3.2-3B
Qwen2.5-3B
Adam
Adam
5.04±0.28
33.44±0.52
9.41±0.87
63.03±0.23
Adam
M(Adam)
5.43±0.28
34.16±0.61
9.21±0.44
63.37±0.21
Adam
M⊥(Adam)
5.65±0.27
33.71±0.17
8.21±0.42
63.54±0.21
Appendix
Table 4: RL optimization on MATH with Llama-3.2-1B, Llama-3.2-1B-Instruct, Llama-3.2-3B, and Qwen2.5-3B. The dynamics-informed configuration in the last row achieves the highest reported mean accuracy on two of the four models. Methods are specified by the update directions for the two LoRA factors: Adam denotes the corresponding base optimizer direction ( DtA or DtB ), and M denotes the norm-preserving spectral flattening applied to the full direction. M⊥(Adam) denotes PtDtB+M((I−Pt)DtB) with an Adam base direction DtB , which preserves the within-subspace component while transforming only the orthogonal component. Values show mean test accuracy (%) across three seeds, with standard errors in subscripts. For each configuration, results are reported at the learning rate with the highest mean accuracy in a completed sweep. Bold indicates the highest reported mean in each model column.
GSM8K
MATH
A direction
B direction
LR
Llama-3.2-1B
Llama-3.2-3B
Llama-3.2-1B
Llama-3.2-3B
Adam
Adam
5×10−6
10.87±1.20
32.35±0.37
5.06±0.02
10.19±0.21
5×10−5
20.60±0.61
42.99±0.38
5.46±0.12
10.57±0.21
5×10−4
22.24±0.33
42.68±0.92
5.66±0.09
10.83±0.13
5×10−3
2.65±0.08
0.58±0.58
3.43±0.52
2.83±1.41
Adam
M(Adam)
5×10−6
11.22±0.20
30.65±0.62
4.96±0.14
10.44±0.16
Appendix
Table 5: Detailed learning-rate sweeps for SFT on GSM8K and MATH. Methods are specified by the update directions for the two LoRA factors: Adam and SGD denote the corresponding base optimizer directions ( DtA or DtB ), and M denotes the norm-preserving spectral flattening applied to the full direction. Values show mean test accuracy (%) across three seeds, with standard errors in subscripts. Bold indicates the highest mean accuracy across learning rates for each update configuration and model–dataset pair.
A direction
B direction
LR
Llama-3.2-1B
Llama-3.2-1B-Instruct
Llama-3.2-3B
Qwen2.5-3B
Adam
Adam
5×10−6
2.65±0.15
57.85±0.39
27.02±3.31
82.82±1.11
5×10−5
7.61±1.24
57.24±0.29
42.36±3.25
83.93±0.34
5×10−4
1.34±0.77
2.75±2.21
0.15±0.15
0.96±0.68
5×10−3
0.00±0.00
0.00±0.00
0.00±0.00
0.00±0.00
Adam
M(Adam)
5×10−6
2.91±0.21
57.97±0.18
14.30±1.12
82.61±0.48
5×10−5
9.67±1.22
57.24±0.59
46.00±0.77
84.46±0.38
Appendix
Table 6: Detailed learning-rate sweeps for RL on GSM8K. Methods are specified by the update directions for the two LoRA factors: Adam and SGD denote the corresponding base optimizer directions ( DtA or DtB ), and M denotes the norm-preserving spectral flattening applied to the full direction. For RL, M⊥(Adam) denotes PtDtB+M((I−Pt)DtB) with an Adam base direction DtB , which preserves the within-subspace component while transforming only the orthogonal component. Values show mean test accuracy (%) across three seeds, with standard errors in subscripts. Bold indicates the highest mean accuracy across learning rates for each update configuration and model.
A direction
B direction
LR
Llama-3.2-1B
Llama-3.2-1B-Instruct
Llama-3.2-3B
Qwen2.5-3B
Adam
Adam
5×10−6
4.79±0.02
33.44±0.52
7.28±0.19
62.78±0.40
5×10−5
5.04±0.28
32.55±0.27
9.41±0.87
63.03±0.23
5×10−4
0.00±0.00
1.17±1.17
0.01±0.01
0.59±0.59
5×10−3
0.00±0.00
0.00±0.00
0.00±0.00
0.00±0.00
Adam
M(Adam)
5×10−6
4.89±0.01
32.91±0.21
7.01±0.07
61.80±0.64
5×10−5
5.43±0.28
34.16±0.61
9.21±0.44
63.37±0.21
Appendix
Table 7: Detailed learning-rate sweeps for RL on MATH. Methods are specified by the update directions for the two LoRA factors: Adam and SGD denote the corresponding base optimizer directions ( DtA or DtB ), and M denotes the norm-preserving spectral flattening applied to the full direction. For RL, M⊥(Adam) denotes PtDtB+M((I−Pt)DtB) with an Adam base direction DtB , which preserves the within-subspace component while transforming only the orthogonal component. Values show mean test accuracy (%) across three seeds, with standard errors in subscripts. Bold indicates the highest mean accuracy across learning rates for each update configuration and model.
Figure 9: Proof roadmap. The top row supplies the common response, moment, spectral, and static-event estimates. Sections C.6 – C.8 prove Propositions A–C; the rotation proof also reuses the trajectory estimates of Section C.6 . Combining each first-order comparison with Proposition T proves the corresponding theorem in this subsection.
School of Artificial Intelligence, Jilin University · Engineering Research Center of Knowledge-Driven Human-Machine Intelligence, MOE, China · International Center of Future Science, Jilin University
School of Computing and Information Systems, Singapore Management University · School of Computing, National University of Singapore · State Key Laboratory for Multimedia Information Processing, Peking University