mlsanityData to 14 Oct 2026

Memory-Computation Tradeoffs in Semi Amortized Parametric Optimization

arXiv 2607.20769 ↗First posted 22 Jul 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

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 K 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 K. For convex objectives satisfying a β-growth condition (β>2), we obtain near-matching bounds and identify a phase transition in K 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.

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

Builds on

31 of 47 references.

References · 31

  1. 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,897
  2. 2021Convex optimizationcited by 2,515
  3. 2016Learning to learn by gradient descent by gradient descentcited by 620
  4. 2017OptNet: Differentiable Optimization as a Layer in Neural Networkscited by 599
  5. 2019Differentiable Convex Optimization Layerscited by 465
  6. 2018Probabilistic Model-Agnostic Meta-Learningcited by 256
  7. 2021Learning to Optimize: A Primer and A Benchmarkcited by 208
  8. 2018Competitive caching with machine learned advicecited by 195
  9. 2020Algorithms with Predictionscited by 153
  10. 2019Fast and Flexible Multi-Task Classification Using Conditional Neural Adaptive Processescited by 131
  11. 2019Multimodal Model-Agnostic Meta-Learning via Task-Aware Modulationcited by 103
  12. 2018Semi-Amortized Variational Autoencoderscited by 83
19 more
  1. 2022Tutorial on amortized optimizationcited by 83
  2. 2024Improving Online Algorithms via ML Predictionscited by 66
  3. 2018Learning an Approximate Model Predictive Controller with Guaranteescited by 57
  4. 2015From error bounds to the complexity of first-order descent methods for convex functionscited by 43
  5. 2019Learning to Solve Large-Scale Security-Constrained Unit Commitment Problemscited by 40
  6. 2019Online Mixed-Integer Optimization in Millisecondscited by 38
  7. 2016Calculus of the exponent of Kurdyka-Lojasiewicz inequality and its applications to linear convergence of first-order methodscited by 35
  8. 2023Learning to Warm-Start Fixed-Point Optimization Algorithmscited by 32
  9. 2014Splitting methods with variable metric for KL functionscited by 26
  10. 2022End-to-End Learning to Warm-Start for Real-Time Quadratic Optimizationcited by 26
  11. 2022Learning Predictions for Algorithms with Predictionscited by 25
  12. 2019Warm Start of Mixed-Integer Programs for Model Predictive Control of Hybrid Systemscited by 17
  13. 2019Safe and Near-Optimal Policy Learning for Model Predictive Control using Primal-Dual Neural Networkscited by 13
  14. 2023Automatic nonlinear MPC approximation with closed-loop guaranteescited by 12
  15. 2021Neural Fixed-Point Acceleration for Convex Optimizationcited by 10
  16. 2021Use of Machine Learning Models to Warmstart Column Generation for Unit Commitmentcited by 5
  17. 2013Embedded Online Optimization for Model Predictive Control at Megahertz Ratescited by 3
  18. 2025Warm-starting active-set solvers using graph neural networkscited by 2
  19. 2025Data-driven Acceleration of MPC with Guaranteescited by 2

Built on by

Not cited yet.