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