On Differentiating Parameterized Argmin and Argmax Problems with Application to Bi-level Optimization
Not on the map. Why?
Abstract (arXiv)
Some recent works in machine learning and computer vision involve the solution of a bi-level optimization problem. Here the solution of a parameterized lower-level problem binds variables that appear in the objective of an upper-level problem. The lower-level problem typically appears as an argmin or argmax optimization problem. Many techniques have been proposed to solve bi-level optimization problems, including gradient descent, which is popular with current end-to-end learning approaches. In this technical report we collect some results on differentiating argmin and argmax optimization problems with and without constraints and provide some insightful motivating examples.
Key facts. Cited by 108 papers in the map (data through 7 Oct 2026). First posted on arXiv 19 Jul 2016.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 24; 2022: 21; 2023: 22; 2024: 21; 2025: 6; 2026: 14.
108 papers.
Sort:
By area:
- 2021Graph Contrastive Learning Automatedcited by 287
- 2023Model Sparsity Can Simplify Machine Unlearningcited by 206
- 2021Efficient and Modular Implicit Differentiationcited by 199
- 2021Investigating Bi-Level Optimization for Learning and Vision from a Unified Perspective: A Survey and Beyondcited by 151
- 2021Hyperbolic Deep Neural Networks: A Surveycited by 138
- 2021Sinkformers: Transformers with Doubly Stochastic Attentioncited by 136
- 2023Decision-Focused Learning: Foundations, State of the Art, Benchmark and Future Opportunitiescited by 129
- 2021Counterfactual Explanations Can Be Manipulatedcited by 113
- 2021Provably Faster Algorithms for Bilevel Optimizationcited by 106
- 2022Equivariance with Learned Canonicalization Functionscited by 106
- 2024SOUL: Unlocking the Power of Second-Order Optimization for LLM Unlearningcited by 104
- 2022Tutorial on amortized optimizationcited by 83
38 more
- 2021Revisiting and Advancing Fast Adversarial Training Through The Lens of Bi-Level Optimizationcited by 75
- 2022Theseus: A Library for Differentiable Nonlinear Optimizationcited by 64
- 2021Amortized Implicit Differentiation for Stochastic Bilevel Optimizationcited by 62
- 2022Advancing Model Pruning via Bi-level Optimizationcited by 55
- 2021CombOptNet: Fit the Right NP-Hard Problem by Learning Integer Programming Constraintscited by 49
- 2023On Finding Small Hyper-Gradients in Bilevel Optimization: Hardness Results and Improved Analysiscited by 39
- 2023An Introduction to Bi-level Optimization: Foundations and Applications in Signal Processing and Machine Learningcited by 39
- 2022DDG-DA: Data Distribution Generation for Predictable Concept Drift Adaptationcited by 36
- 2021On the Convergence Theory for Hessian-Free Bilevel Algorithmscited by 33
- 2022PyEPO: A PyTorch-based End-to-End Predict-then-Optimize Library for Linear and Integer Programmingcited by 32
- 2023Optimal Algorithms for Stochastic Bilevel Optimization under Relaxed Smoothness Conditionscited by 31
- 2021Max-Margin Contrastive Learningcited by 27
- 2024WAGLE: Strategic Weight Attribution for Effective and Modular Unlearning in Large Language Modelscited by 25
- 2021Implicit differentiation for fast hyperparameter selection in non-smooth convex learningcited by 22
- 2022Alternating Differentiation for Optimization Layerscited by 22
- 2022Decentralized Stochastic Bilevel Optimization with Improved per-Iteration Complexitycited by 21
- 2024Functional Bilevel Optimization for Machine Learningcited by 20
- 2022Efficiently Escaping Saddle Points in Bilevel Optimizationcited by 19
- 2021Multiset-Equivariant Set Prediction with Approximate Implicit Differentiationcited by 17
- 2022A Conditional Gradient-based Method for Simple Bilevel Optimization with Convex Lower-level Problemcited by 17
- 2023Landscape Surrogate: Learning Decision Losses for Mathematical Optimization Under Partial Informationcited by 17
- 2023Unlocking Slot Attention by Changing Optimal Transport Costscited by 16
- 2023Behavior Alignment via Reward Function Optimizationcited by 16
- 2024Adversarial Machine Unlearningcited by 16
- 2024Imperative Learning: A Self-supervised Neuro-Symbolic Learning Framework for Robot Autonomycited by 16
- 2023Efficient Gradient Approximation Method for Constrained Bilevel Optimizationcited by 15
- 2021Sign-MAML: Efficient Model-Agnostic Meta-Learning by SignSGDcited by 14
- 2023SimFBO: Towards Simple, Flexible and Communication-efficient Federated Bilevel Learningcited by 14
- 2024Guiding Neural Collapse: Optimising Towards the Nearest Simplex Equiangular Tight Framecited by 14
- 2022Asynchronous Distributed Bilevel Optimizationcited by 13
- 2023Backpropagation of Unrolled Solvers with Folded Optimizationcited by 13
- 2023Communication-Efficient Federated Hypergradient Computation via Aggregated Iterative Differentiationcited by 13
- 2022A Primal-Dual Approach to Bilevel Optimization with Multiple Inner Minimacited by 12
- 2021Convergent Graph Solverscited by 11
- 2021Neural Fixed-Point Acceleration for Convex Optimizationcited by 10
- 2022Alternating Implicit Projected SGD and Its Efficient Variants for Equality-constrained Bilevel Optimizationcited by 10
- 2023What Is Missing in IRM Training and Evaluation? Challenges and Solutionscited by 10
- 2024First-Order Methods for Linearly Constrained Bilevel Optimizationcited by 10
50 most cited of 108 All 108 (JSON)
- 2026Recovering Weak Signals with Normalizing Flows6 Sep 2026
- 2026Disciplined Bilevel Programming1 Sep 2026
- 2026Non-Linear Strategic Classification Made Practical26 Jun 2026
- 2026TurboMPC: Fast, Scalable, and Differentiable Model Predictive Control on the GPU23 Jun 2026
- 2026Constrained Variable Projection for Structured Problems22 Jun 2026
- 2026Scaling Decision-Focused Learning to Large Problems with Lagrangian Decomposition7 Jun 2026
- 2026DiffSlack: Learning under Nonlinear Inequality Constraints via Learnable Slack Variables3 Jun 2026
- 2026Constraint-Aware Flow Matching: Decision Aligned End-to-End Training for Constrained Sampling12 May 2026
- 2026InfoGeo: Information-Theoretic Object-Centric Learning for Cross-View Generalizable UAV Geo-Localization8 May 2026
- 2026Cheap Thrills: Effective Amortized Optimization Using Inexpensive Labels5 Mar 2026
- 2026Differentiable Zero-One Loss via Hypersimplex Projections26 Feb 2026
- 2026Fully First-Order Algorithms for Online Bilevel Optimization12 Feb 2026
38 more
- 2026Achieving Better Local Regret Bound for Online Non-Convex Bilevel Optimization6 Feb 2026
- 2026Learning to Optimize by Differentiable Programming23 Jan 2026
- 2025APT: Affine Prototype-Timestamp For Time Series Forecasting Under Distribution Shift17 Nov 2025
- 2025Scalable Principal-Agent Contract Design via Gradient-Based Optimization24 Oct 2025
- 2025Differentiable Model Predictive Control on the GPU7 Oct 2025
- 2025Sharper Convergence Rates for Nonconvex Optimisation via Reduction Mappings10 Jun 2025
- 2025End-to-End Probabilistic Framework for Learning with Hard Constraints8 Jun 2025
- 2025Online Decision-Focused Learning19 May 2025
- 2024Differentiable Convex Optimization Layers in Neural Architectures: Foundations and Perspectives30 Dec 2024
- 2024Guiding Neural Collapse: Optimising Towards the Nearest Simplex Equiangular Tight Frame2 Nov 2024
- 2024Fully First-Order Methods for Decentralized Bilevel Optimization25 Oct 2024
- 2024WAGLE: Strategic Weight Attribution for Effective and Modular Unlearning in Large Language Models23 Oct 2024
- 2024End-to-End Optimization and Learning of Fair Court Schedules22 Oct 2024
- 2024Towards Differentiable Multilevel Optimization: A Gradient-Based Approach15 Oct 2024
- 2024Meta-Reinforcement Learning with Universal Policy Adaptation: Provable Near-Optimality under All-task Optimum Comparator13 Oct 2024
- 2024Differentiation Through Black-Box Quadratic Programming Solvers8 Oct 2024
- 2024Tuning-Free Bilevel Optimization: New Algorithms and Convergence Analysis7 Oct 2024
- 2024Exploiting Missing Data Remediation Strategies using Adversarial Missingness Attacks6 Sep 2024
- 2024Preference-Optimized Pareto Set Learning for Blackbox Optimization19 Aug 2024
- 2024Interventional Causal Structure Discovery over Graphical Models with Convergence and Optimality Guarantees9 Aug 2024
- 2024LPGD: A General Framework for Backpropagation through Embedded Optimization Layers8 Jul 2024
- 2024Imperative Learning: A Self-supervised Neuro-Symbolic Learning Framework for Robot Autonomy23 Jun 2024
- 2024First-Order Methods for Linearly Constrained Bilevel Optimization18 Jun 2024
- 2024Adversarial Machine Unlearning11 Jun 2024
- 2024Triadic-OCD: Asynchronous Online Change Detection with Provable Robustness, Optimality, and Convergence3 May 2024
- 2024Accelerated Fully First-Order Methods for Bilevel and Minimax Optimization1 May 2024
- 2024SOUL: Unlocking the Power of Second-Order Optimization for LLM Unlearning28 Apr 2024
- 2024Functional Bilevel Optimization for Machine Learning29 Mar 2024
- 2024Stability-Aware Training of Machine Learning Force Fields with Differentiable Boltzmann Estimators21 Feb 2024
- 2023Analyzing and Enhancing the Backward-Pass Convergence of Unrolled Optimization28 Dec 2023
- 2023Provably Convergent Federated Trilevel Learning19 Dec 2023
- 2023Achieving O(ε-1.5) Complexity in Hessian/Jacobian-free Stochastic Bilevel Optimization6 Dec 2023
- 2023Behavior Alignment via Reward Function Optimization29 Oct 2023
- 2023Non-Convex Bilevel Optimization with Time-Varying Objective Functions7 Aug 2023
- 2023An Introduction to Bi-level Optimization: Foundations and Applications in Signal Processing and Machine Learning1 Aug 2023
- 2023Doubly Robust Instance-Reweighted Adversarial Training1 Aug 2023
- 2023Decision-Focused Learning: Foundations, State of the Art, Benchmark and Future Opportunities25 Jul 2023
- 2023Landscape Surrogate: Learning Decision Losses for Mathematical Optimization Under Partial Information18 Jul 2023
50 newest of 108 All 108 (JSON)
- 2023Model Sparsity Can Simplify Machine Unlearningcited by 206
- 2024SOUL: Unlocking the Power of Second-Order Optimization for LLM Unlearningcited by 104
- 2024WAGLE: Strategic Weight Attribution for Effective and Modular Unlearning in Large Language Modelscited by 25
- 2024Adversarial Machine Unlearningcited by 16
- 2024WAGLE: Strategic Weight Attribution for Effective and Modular Unlearning in Large Language Models23 Oct 2024
- 2024Adversarial Machine Unlearning11 Jun 2024
- 2024SOUL: Unlocking the Power of Second-Order Optimization for LLM Unlearning28 Apr 2024
- 2023Model Sparsity Can Simplify Machine Unlearning11 Apr 2023
- 2021Graph Contrastive Learning Automatedcited by 287
- 2021Efficient and Modular Implicit Differentiationcited by 199
- 2021Investigating Bi-Level Optimization for Learning and Vision from a Unified Perspective: A Survey and Beyondcited by 151
- 2021Hyperbolic Deep Neural Networks: A Surveycited by 138
- 2021Sinkformers: Transformers with Doubly Stochastic Attentioncited by 136
- 2023Decision-Focused Learning: Foundations, State of the Art, Benchmark and Future Opportunitiescited by 129
- 2021Counterfactual Explanations Can Be Manipulatedcited by 113
- 2021Provably Faster Algorithms for Bilevel Optimizationcited by 106
- 2022Equivariance with Learned Canonicalization Functionscited by 106
- 2022Tutorial on amortized optimizationcited by 83
- 2021Revisiting and Advancing Fast Adversarial Training Through The Lens of Bi-Level Optimizationcited by 75
- 2022Theseus: A Library for Differentiable Nonlinear Optimizationcited by 64
12 most cited of 102 All 102 (JSON)
- 2026Recovering Weak Signals with Normalizing Flows6 Sep 2026
- 2026Disciplined Bilevel Programming1 Sep 2026
- 2026Non-Linear Strategic Classification Made Practical26 Jun 2026
- 2026TurboMPC: Fast, Scalable, and Differentiable Model Predictive Control on the GPU23 Jun 2026
- 2026Constrained Variable Projection for Structured Problems22 Jun 2026
- 2026Scaling Decision-Focused Learning to Large Problems with Lagrangian Decomposition7 Jun 2026
- 2026DiffSlack: Learning under Nonlinear Inequality Constraints via Learnable Slack Variables3 Jun 2026
- 2026InfoGeo: Information-Theoretic Object-Centric Learning for Cross-View Generalizable UAV Geo-Localization8 May 2026
- 2026Cheap Thrills: Effective Amortized Optimization Using Inexpensive Labels5 Mar 2026
- 2026Differentiable Zero-One Loss via Hypersimplex Projections26 Feb 2026
- 2026Fully First-Order Algorithms for Online Bilevel Optimization12 Feb 2026
- 2026Achieving Better Local Regret Bound for Online Non-Convex Bilevel Optimization6 Feb 2026
12 newest of 102 All 102 (JSON)