Non-Parametric Optimization for Scalable Learning in Stochastic Decision Problems
Mohsen Amidzade, Lauri Viitasaari, Mario Di Francesco
Abstract
Stochastic optimization (SO) plays a central role in decision-making under uncertainty. Among SO problems, time-varying stochastic optimization (TV-SO) is particularly important due to its applications in adaptive control and machine learning. Non-parametric approaches have been proposed for time-varying deterministic optimization; however, they have not been developed for their stochastic counterparts. This work addresses that gap by developing a stochastic variational framework based on Malliavin calculus. This framework yields non-parametric optimality conditions for SO problems with stochastic decisions and supports the design of a scalable deep-learning algorithm that is insensitive to the parameterization dimension. This algorithm, called the Stochastic Path Follower (SPF), is applied to two important problems under distribution drift, namely least-squares recovery and logistic regression. Experimental results show that the proposed approach outperforms state-of-the-art learning-based and gradient-based methods in both performance and scalability.
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 2b99e744-b493-448a-91e4-23d2850abac8Builds on4
- Neural SDEs as Infinite-Dimensional GANsPatrick Kidger, James Foster, Xuechen Li, Terry J. LyonsICML 2021 · 214 citations
- Stochastic optimization under time drift: iterate averaging, step-decay schedules, and high probability guaranteesJoshua Cutler, Dmitriy Drusvyatskiy, Zaïd HarchaouiNeurIPS 2021 · 19 citations
- Predictor-corrector algorithms for stochastic optimization under gradual distribution shiftSubha Maity, Debarghya Mukherjee, Moulinath Banerjee, Yuekai SunICLR 2023 · 1 citation
- Efficient Training of Neural Stochastic Differential Equations by Matching Finite Dimensional DistributionsJianxin Zhang, Josh Viktorov, Doosan Jung, Emily PitlerICLR 2025
Related papers
- Provable Learning-based Algorithm For Sparse RecoveryXinshi Chen, Haoran Sun, Le SongICLR 2022
- D-SPIDER-SFO: A Decentralized Optimization Algorithm with Faster Convergence Rate for Nonconvex ProblemsTaoxing Pan, Jun Liu, Jie WangAAAI 2020 · 19 citations
- Adapting to Mixing Time in Stochastic Optimization with Markovian DataRon Dorfman, Kfir Yehuda LevyICML 2022 · 41 citations
- Stochastic Decentralized Optimization of Non-Smooth Convex and Convex-Concave Problems over Time-Varying NetworksMaxim Divilkovskiy, Alexander GasnikovAAAI 2026
- SAPD+: An Accelerated Stochastic Method for Nonconvex-Concave Minimax ProblemsXuan Zhang, Necdet Serhat Aybat, Mert GürbüzbalabanNeurIPS 2022 · 55 citations
