Predictor-corrector algorithms for stochastic optimization under gradual distribution shift
Subha Maity, Debarghya Mukherjee, Moulinath Banerjee, Yuekai Sun
Abstract
Time-varying stochastic optimization problems frequently arise in machine learning practice (e.g. gradual domain shift, object tracking, strategic classification). Often, the underlying process that drives the distribution shift is continuous in nature. We exploit this underlying continuity by developing predictor-corrector algorithms for time-varying stochastic optimization that anticipates changes in the underlying data generating process through a predictor-corrector term in the update rule. The key challenge is the estimation of the predictor-corrector term; a naive approach based on sample-average approximation may lead to non-convergence. We develop a general moving-average based method to estimate the predictorcorrector term and provide error bounds for the iterates, both in presence of pure and noisy access to the queries from the relevant derivatives of the loss function. Furthermore, we show (theoretically and empirically in several examples) that our method outperforms non-predictor corrector methods that do not anticipate changes in the data generating process. 1
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 902f1a85-ba2e-49c8-8ee6-5bc7e54a8096Cited by top-tier papers3
- Bilevel Optimization with Coupled Decision-Dependent DistributionsSongtao LuICML 2023 · 15 citations
- Learning in reverse causal strategic environments with ramifications on two sided marketsSeamus Somerstep, Yuekai Sun, Yaacov RitovICLR 2024 · 5 citations
- Non-Parametric Optimization for Scalable Learning in Stochastic Decision ProblemsMohsen Amidzade, Lauri Viitasaari, Mario Di FrancescoICML 2026
Builds on2
- Stochastic Optimization for Performative PredictionCelestine Mendler-Dünner, Juan C. Perdomo, Tijana Zrnic, Moritz HardtNeurIPS 2020 · 161 citations
- Stochastic optimization under time drift: iterate averaging, step-decay schedules, and high probability guaranteesJoshua Cutler, Dmitriy Drusvyatskiy, Zaïd HarchaouiNeurIPS 2021 · 19 citations
Related papers
- TRACE: A Generalizable Drift Detector for Streaming Data-Driven OptimizationYuan-Ting Zhong, Ting Huang, Xiaolin Xiao, Yue-Jiao GongAAAI 2026 · 1 citation
- Prospective Learning: Learning for a Dynamic FutureAshwin De Silva, Rahul Ramesh, Rubing Yang, Siyu Yu et al.NeurIPS 2024 · 5 citations
- Adapting to Continuous Covariate Shift via Online Density Ratio EstimationYu-Jie Zhang, Zhen-Yu Zhang, Peng Zhao, Masashi SugiyamaNeurIPS 2023 · 25 citations
- Time-Varying Propensity Score to Bridge the Gap between the Past and PresentRasool Fakoor, Jonas Mueller, Zachary Chase Lipton, Pratik Chaudhari et al.ICLR 2024 · 4 citations
- Optimizing for the Future in Non-Stationary MDPsYash Chandak, Georgios Theocharous, Shiv Shankar, Martha White et al.ICML 2020 · 72 citations
