Biased Stochastic First-Order Methods for Conditional Stochastic Optimization and Applications in Meta Learning
Not on the map. Why?
Abstract (arXiv)
Conditional stochastic optimization covers a variety of applications ranging from invariant learning and causal inference to meta-learning. However, constructing unbiased gradient estimators for such problems is challenging due to the composition structure. As an alternative, we propose a biased stochastic gradient descent (BSGD) algorithm and study the bias-variance tradeoff under different structural assumptions. We establish the sample complexities of BSGD for strongly convex, convex, and weakly convex objectives under smooth and non-smooth conditions. Our lower bound analysis shows that the sample complexities of BSGD cannot be improved for general convex objectives and nonconvex objectives except for smooth nonconvex objectives with Lipschitz continuous gradient estimator. For this special setting, we propose an accelerated algorithm called biased SpiderBoost (BSpiderBoost) that matches the lower bound complexity. We further conduct numerical experiments on invariant logistic regression and model-agnostic meta-learning to illustrate the performance of BSGD and BSpiderBoost.
Key facts. Cited by 46 papers in the map (data through 7 Oct 2026). First posted on arXiv 25 Feb 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 7; 2022: 9; 2023: 12; 2024: 9; 2025: 5; 2026: 4.
46 papers.
Sort:
By area:
- 2023A Guide Through the Zoo of Biased SGDcited by 39
- 2021Sinkhorn Distributionally Robust Optimizationcited by 38
- 2021Randomized Stochastic Variance-Reduced Methods for Multi-Task Stochastic Bilevel Optimizationcited by 32
- 2022Finite-Sum Coupled Compositional Stochastic Optimization: Theory and Applicationscited by 27
- 2022Large-scale Stochastic Optimization of NDCG Surrogates for Deep Learning with Provable Convergencecited by 21
- 2022Multi-block-Single-probe Variance Reduced Estimator for Coupled Compositional Optimizationcited by 16
- 2021Momentum Accelerates the Convergence of Stochastic AUPRC Maximizationcited by 14
- 2023Contextual Stochastic Bilevel Optimizationcited by 14
- 2022Federated Learning and Meta Learning: Approaches, Applications, and Directionscited by 12
- 2022FeDXL: Provable Federated Learning for Deep X-Risk Optimizationcited by 10
- 2023Non-Smooth Weakly-Convex Finite-sum Coupled Compositional Optimizationcited by 10
- 2025MPO: An Efficient Post-Processing Framework for Mixing Diverse Preference Alignmentcited by 10
34 more
- 2025Advantage Shaping as Surrogate Reward Maximization: Unifying Pass@K Policy Gradientscited by 10
- 2023Post-processing Private Synthetic Data for Improving Utility on Selected Measurescited by 9
- 2023Wasserstein Distributionally Robust Policy Evaluation and Learning for Contextual Banditscited by 9
- 2022A Projection-free Algorithm for Constrained Stochastic Multi-level Composition Optimizationcited by 8
- 2022Algorithmic Foundations of Empirical X-risk Minimizationcited by 7
- 2021Memory-Based Optimization Methods for Model-Agnostic Meta-Learning and Personalized Federated Learningcited by 6
- 2021KOALA: A Kalman Optimization Algorithm with Loss Adaptivitycited by 6
- 2022Constructing unbiased gradient estimators with finite variance for conditional stochastic optimizationcited by 6
- 2023Debiasing Conditional Stochastic Optimizationcited by 6
- 2021Online Estimation and Optimization of Utility-Based Shortfall Riskcited by 5
- 2024CoBo: Collaborative Learning via Bilevel Optimizationcited by 5
- 2023Federated Conditional Stochastic Optimizationcited by 4
- 2024Non-asymptotic Analysis of Biased Adaptive Stochastic Approximationcited by 4
- 2024Non-Convex Robust Hypothesis Testing using Sinkhorn Uncertainty Setscited by 4
- 2024Statistical and Computational Guarantees of Kernel Max-Sliced Wasserstein Distancescited by 4
- 2024Stochastic Optimization Algorithms for Instrumental Variable Regression with Streaming Datacited by 4
- 2024Deep Learning for Computing Convergence Rates of Markov Chainscited by 4
- 2023Provable Multi-instance Deep AUC Maximization with Stochastic Poolingcited by 3
- 2023Serverless Federated AUPRC Optimization for Multi-Party Collaborative Imbalanced Data Miningcited by 3
- 2023Optimal Guarantees for Algorithmic Reproducibility and Gradient Complexity in Convex Optimizationcited by 3
- 2023A Near-Optimal Single-Loop Stochastic Algorithm for Convex Finite-Sum Coupled Compositional Optimizationcited by 3
- 2024A Functional Model Method for Nonconvex Nonsmooth Conditional Stochastic Optimizationcited by 3
- 2021A Variance Controlled Stochastic Method with Biased Estimation for Faster Non-convex Optimizationcited by 2
- 2024Parallel Momentum Methods Under Biased Gradient Estimationscited by 2
- 2023Stochastic Nested Compositional Bi-level Optimization for Robust Feature Learningcited by 1
- 2024On the KL-Divergence-based Robust Satisficing Modelcited by 1
- 2025Iterative Sampling Methods for Sinkhorn Distributionally Robust Optimizationcited by 1
- 2026A Geometry-Aware Efficient Algorithm for Compositional Entropic Risk Minimizationcited by 1
- 2022Fairness via Adversarial Attribute Neighbourhood Robust Learningcited by 0
- 2025Quantum speedup of non-linear Monte Carlo problemscited by 0
- 2025Stochastic Momentum Methods for Non-smooth Non-Convex Finite-Sum Coupled Compositional Optimizationcited by 0
- 2026Convergence Rate of a Functional Learning Method for Contextual Stochastic Optimizationcited by 0
- 2026UB-SMoE: Universally Balanced Sparse Mixture-of-Experts for Resource-adaptive Federated Fine-tuning of Foundation Modelscited by 0
- 2026Convex Optimization Is Free When Accuracy Is Expensivecited by 0
- 2026Convex Optimization Is Free When Accuracy Is Expensive28 Sep 2026
- 2026UB-SMoE: Universally Balanced Sparse Mixture-of-Experts for Resource-adaptive Federated Fine-tuning of Foundation Models15 May 2026
- 2026Convergence Rate of a Functional Learning Method for Contextual Stochastic Optimization13 Mar 2026
- 2026A Geometry-Aware Efficient Algorithm for Compositional Entropic Risk Minimization2 Feb 2026
- 2025Iterative Sampling Methods for Sinkhorn Distributionally Robust Optimization14 Dec 2025
- 2025Advantage Shaping as Surrogate Reward Maximization: Unifying Pass@K Policy Gradients27 Oct 2025
- 2025Stochastic Momentum Methods for Non-smooth Non-Convex Finite-Sum Coupled Compositional Optimization3 Jun 2025
- 2025MPO: An Efficient Post-Processing Framework for Mixing Diverse Preference Alignment25 Feb 2025
- 2025Quantum speedup of non-linear Monte Carlo problems7 Feb 2025
- 2024CoBo: Collaborative Learning via Bilevel Optimization9 Sep 2024
- 2024On the KL-Divergence-based Robust Satisficing Model17 Aug 2024
- 2024Deep Learning for Computing Convergence Rates of Markov Chains30 May 2024
34 more
- 2024Stochastic Optimization Algorithms for Instrumental Variable Regression with Streaming Data29 May 2024
- 2024Statistical and Computational Guarantees of Kernel Max-Sliced Wasserstein Distances24 May 2024
- 2024A Functional Model Method for Nonconvex Nonsmooth Conditional Stochastic Optimization17 May 2024
- 2024Non-Convex Robust Hypothesis Testing using Sinkhorn Uncertainty Sets21 Mar 2024
- 2024Parallel Momentum Methods Under Biased Gradient Estimations29 Feb 2024
- 2024Non-asymptotic Analysis of Biased Adaptive Stochastic Approximation5 Feb 2024
- 2023A Near-Optimal Single-Loop Stochastic Algorithm for Convex Finite-Sum Coupled Compositional Optimization4 Dec 2023
- 2023Contextual Stochastic Bilevel Optimization27 Oct 2023
- 2023Optimal Guarantees for Algorithmic Reproducibility and Gradient Complexity in Convex Optimization26 Oct 2023
- 2023Non-Smooth Weakly-Convex Finite-sum Coupled Compositional Optimization5 Oct 2023
- 2023Federated Conditional Stochastic Optimization4 Oct 2023
- 2023Wasserstein Distributionally Robust Policy Evaluation and Learning for Contextual Bandits15 Sep 2023
- 2023Serverless Federated AUPRC Optimization for Multi-Party Collaborative Imbalanced Data Mining6 Aug 2023
- 2023Stochastic Nested Compositional Bi-level Optimization for Robust Feature Learning11 Jul 2023
- 2023A Guide Through the Zoo of Biased SGD25 May 2023
- 2023Post-processing Private Synthetic Data for Improving Utility on Selected Measures24 May 2023
- 2023Provable Multi-instance Deep AUC Maximization with Stochastic Pooling14 May 2023
- 2023Debiasing Conditional Stochastic Optimization20 Apr 2023
- 2022FeDXL: Provable Federated Learning for Deep X-Risk Optimization26 Oct 2022
- 2022Federated Learning and Meta Learning: Approaches, Applications, and Directions24 Oct 2022
- 2022Fairness via Adversarial Attribute Neighbourhood Robust Learning12 Oct 2022
- 2022Multi-block-Single-probe Variance Reduced Estimator for Coupled Compositional Optimization18 Jul 2022
- 2022Constructing unbiased gradient estimators with finite variance for conditional stochastic optimization4 Jun 2022
- 2022Algorithmic Foundations of Empirical X-risk Minimization1 Jun 2022
- 2022Finite-Sum Coupled Compositional Stochastic Optimization: Theory and Applications24 Feb 2022
- 2022Large-scale Stochastic Optimization of NDCG Surrogates for Deep Learning with Provable Convergence24 Feb 2022
- 2022A Projection-free Algorithm for Constrained Stochastic Multi-level Composition Optimization9 Feb 2022
- 2021Online Estimation and Optimization of Utility-Based Shortfall Risk16 Nov 2021
- 2021Sinkhorn Distributionally Robust Optimization24 Sep 2021
- 2021KOALA: A Kalman Optimization Algorithm with Loss Adaptivity7 Jul 2021
- 2021Momentum Accelerates the Convergence of Stochastic AUPRC Maximization2 Jul 2021
- 2021Memory-Based Optimization Methods for Model-Agnostic Meta-Learning and Personalized Federated Learning9 Jun 2021
- 2021Randomized Stochastic Variance-Reduced Methods for Multi-Task Stochastic Bilevel Optimization5 May 2021
- 2021A Variance Controlled Stochastic Method with Biased Estimation for Faster Non-convex Optimization19 Feb 2021
- 2023A Guide Through the Zoo of Biased SGDcited by 39
- 2021Sinkhorn Distributionally Robust Optimizationcited by 38
- 2021Randomized Stochastic Variance-Reduced Methods for Multi-Task Stochastic Bilevel Optimizationcited by 32
- 2022Finite-Sum Coupled Compositional Stochastic Optimization: Theory and Applicationscited by 27
- 2022Large-scale Stochastic Optimization of NDCG Surrogates for Deep Learning with Provable Convergencecited by 21
- 2022Multi-block-Single-probe Variance Reduced Estimator for Coupled Compositional Optimizationcited by 16
- 2021Momentum Accelerates the Convergence of Stochastic AUPRC Maximizationcited by 14
- 2023Contextual Stochastic Bilevel Optimizationcited by 14
- 2022Federated Learning and Meta Learning: Approaches, Applications, and Directionscited by 12
- 2022FeDXL: Provable Federated Learning for Deep X-Risk Optimizationcited by 10
- 2023Non-Smooth Weakly-Convex Finite-sum Coupled Compositional Optimizationcited by 10
- 2023Post-processing Private Synthetic Data for Improving Utility on Selected Measurescited by 9
12 most cited of 43 All 43 (JSON)
- 2026Convex Optimization Is Free When Accuracy Is Expensive28 Sep 2026
- 2026Convergence Rate of a Functional Learning Method for Contextual Stochastic Optimization13 Mar 2026
- 2026A Geometry-Aware Efficient Algorithm for Compositional Entropic Risk Minimization2 Feb 2026
- 2025Iterative Sampling Methods for Sinkhorn Distributionally Robust Optimization14 Dec 2025
- 2025Stochastic Momentum Methods for Non-smooth Non-Convex Finite-Sum Coupled Compositional Optimization3 Jun 2025
- 2025Quantum speedup of non-linear Monte Carlo problems7 Feb 2025
- 2024CoBo: Collaborative Learning via Bilevel Optimization9 Sep 2024
- 2024On the KL-Divergence-based Robust Satisficing Model17 Aug 2024
- 2024Deep Learning for Computing Convergence Rates of Markov Chains30 May 2024
- 2024Stochastic Optimization Algorithms for Instrumental Variable Regression with Streaming Data29 May 2024
- 2024Statistical and Computational Guarantees of Kernel Max-Sliced Wasserstein Distances24 May 2024
- 2024A Functional Model Method for Nonconvex Nonsmooth Conditional Stochastic Optimization17 May 2024
12 newest of 43 All 43 (JSON)