mlsanityData to 14 Oct 2026

MpSub: A Momentum p-Dimensional Subspace Trust-Region Method for Derivative-Free Fine-Tuning of Large Language Models

arXiv 2609.07666 ↗First posted 7 Sep 2026cs.LG math.OCcited by 0

Not on the map. Why?

Abstract (arXiv)

Full-parameter fine-tuning of large language models has substantial memory costs because backpropagation stores activations and gradients. Zeroth-order optimization avoids this by estimating update directions from loss evaluations, but existing methods require tuning a sensitive learning rate for each model and task. We propose the momentum p-dimensional subspace trust-region method (MpSub). At each iteration, MpSub searches within a p-dimensional subspace: one direction preserves historical momentum from the most recent accepted step, while the remaining directions explore via fresh random sampling. The subspace gradient is estimated by central differences, a trial step is computed from a linear trust-region model, and the trust-region radius adapts according to the agreement between predicted and observed loss reduction, eliminating the learning rate. For LLM fine-tuning, evaluations within an iteration share a minibatch, and directions are regenerated in place from seeds, using forward passes alone. For smooth deterministic objectives under unorthogonalized Gaussian directions, we bound the finite-difference error, quantify gradient energy captured by the subspace, and prove that limk→∞ |nabla f(xk)|2 = 0 almost surely under a safeguarded radius update. Under a matched budget of 8,400 training-objective forward passes, we fine-tune OPT-125M and OPT-350M on CommitmentBank. With the same preset parameters at both model sizes, MpSub attains mean test accuracies of 0.673 and 0.690 over three seeds, matching tuned MeZO (0.685) without any learning-rate search.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 7 of its 14 references that are in the map. First posted on arXiv 7 Sep 2026.

Builds on

7 of 14 references.

References · 7

  1. 2023Fine-Tuning Language Models with Just Forward Passescited by 345
  2. 2024Enhancing Zeroth-order Fine-tuning for Language Models with Low-rank Structurescited by 50
  3. 2024Zeroth-Order Fine-Tuning of LLMs in Random Subspacescited by 26
  4. 2020A stochastic subspace approach to gradient-free optimization in high dimensionscited by 16
  5. 2021Scalable Subspace Methods for Derivative-Free Nonlinear Least-Squares Optimizationcited by 10
  6. 2022Randomised subspace methods for non-convex optimization, with applications to nonlinear least-squarescited by 4
  7. 2023A New Two-dimensional Model-based Subspace Method for Large-scale Unconstrained Derivative-free Optimization: 2D-MoSubcited by 2

Built on by

Not cited yet.