cs.LGJul 22, 2026

Memory-Computation Tradeoffs in Semi Amortized Parametric Optimization

Authors: Shijie PanAgustin CastellanoZeyu ShenEnrique Mallada

Organizations: Department of Electrical and Computer Engineering, Johns Hopkins University · Department of Applied Mathematics and Statistics, Johns Hopkins University Baltimore, MD 21218, USA

Abstract

Learning-enabled decision systems often use offline data or computation to reduce online compute cost. Despite the empirical success of such approaches, there is limited general understanding of how much offline information is needed to achieve a desired accuracy under a fixed online computation budget. We study this question through the lens of amortized parametric optimization: an offline phase stores a finite memory of solved problem instances, and an online phase produces a solution to a new instance by retrieving a warm start and applying KK steps of projected gradient descent. We analyze this setup for smooth convex parametric optimization over a compact domain, using a nonparametric predictor built from the stored offline solutions. For μμ-strongly convex objectives, we establish matching upper and lower bounds on the memory required to guarantee ε\varepsilon-accuracy under a fixed online iteration budget KK. For convex objectives satisfying a ββ-growth condition (β>2β>2), we obtain near-matching bounds and identify a phase transition in KK beyond which additional memory provides no benefit. We further provide a general proof framework that (i) explicitly quantifies the memory cost of acceleration---how much offline memory is required to achieve a prescribed speedup over the unaided online optimizer---and (ii) identifies two key quantities driving this cost: the convergence rate of the online optimizer and the Lipschitz sensitivity of the solution map to the problem parameter. Experiments on parameterized ridge regression confirm the predicted memory--computation--accuracy tradeoffs.

Explore similar work

Jul 21, 2026cs.DS

Stronger Memory-Query Tradeoffs for Convex Optimization: The Limitations of Subquadratic Memory

We prove two lower bounds for the first order oracle complexity of minimizing a dd-dimensional 11-Lipschitz convex function over the unit ball with mm bits of memory. We first show that any such (possibly randomized) algorithm must make Ω~(d2m)\tildeΩ(\frac{d^2}{\sqrt{m}}) oracle queries. For deterministic optimization algorithms, we show that Ω~(min{d1.6,d8/3m2/3})\tildeΩ(\min\{d^{1.6},\frac{d^{8/3}}{m^{2/3}}\}) queries are required. For all memory regimes of interest, these improves upon the previous best known lower bounds of Ω~(max{d8/3m4/3,d4/3m1/6})\tildeΩ(\max\{\frac{d^{8/3}}{m^{4/3}},\frac{d^{4/3}}{m^{1/6}}\}) and Ω~(d5/3m1/3)\tildeΩ(\frac{d^{5/3}}{m^{1/3}}) for randomized and deterministic algorithms respectively. Notably, due to existing upper bounds, our lower bound for deterministic algorithms is the first to show a sharp oracle complexity phase transition around md2m\approx d^2, where a polylogarithmic change in memory leads to a poly(d)\mathsf{poly}(d) change in the number of required oracle calls. Further, when the suboptimality is polynomially small in dd, our lower bound randomized algorithms is the first to show that Ω~(d2)\tildeΩ(d^2) memory is necessary to nearly match the optimal query complexity among algorithms without memory constraints. Previously, such a result was only known for the regime where the suboptimality is quasipolynomially small in dd.
Michael Menart, Aleksandar Nikolov, Ohad Shamir
Apr 29, 2026math.OC

Learning Over-Relaxation Policies for ADMM with Convergence Guarantees

The Alternating Direction Method of Multipliers (ADMM) is a widely used method for structured convex optimization, and its practical performance depends strongly on the choice of penalty and relaxation parameters. Motivated by settings such as Model Predictive Control (MPC), where one repeatedly solves related optimization problems with fixed structure and changing parameter values, we propose learning online updates of the relaxation parameter to improve performance on problem classes of interest. This choice is computationally attractive in OSQP-like architectures, since adapting relaxation does not trigger the matrix refactorizations associated with penalty updates. We establish convergence guarantees for ADMM with time-varying penalty and relaxation parameters under mild assumptions, and show on benchmark quadratic programs that the resulting learned policies improve both iteration count and wall-clock time over baseline OSQP.
Junan Lin, Paul J. Goulart, Luca Furieri
May 28, 2026cs.CL

How LoRA Remembers? A Parametric Memory Law for LLM Finetuning

Large Language Models (LLMs) must continuously learn and update knowledge to remain effective in dynamic real-world environments. While Low-Rank Adaptation (LoRA) is widely used for such memory updates, existing studies mainly rely on qualitative downstream evaluations, leaving the quantitative capacity limits and underlying dynamics of exact parametric memory largely unexplored. To bridge this gap, we employ LoRA as a controlled memory capacity probe within the latent space to systematically quantify exact parametric memory. We introduce the Parametric Memory Law, a robust power law linking loss reduction Delta L to effective parameters and sequence length. At the token level, fine-grained analysis reveals a deterministic phase transition, demonstrating that a prediction probability of p > 0.5 constitutes a sufficient condition for verbatim recall under greedy decoding. Driven by these insights, we introduce MemFT, a threshold-guided optimization strategy that dynamically redistributes the training budget toward sub-threshold tokens. Empirical evaluations demonstrate that MemFT can enhance memory fidelity and efficiency. Code will be released at https://github.com/zjunlp/ParametricMemoryLaw.
Ziwen Xu, Haiwen Hong, Linsong Yu +4