Oct, 2023

上下文随机双层优化

TL;DR我们引入了上下文随机双层优化(CSBO)—— 一种在下层问题最小化给定某些上下文信息和上层决策变量条件下的期望的随机双层优化框架。该框架扩展了经典的随机双层优化,使得下层决策者不仅对上层决策者的决策做出最优响应,还对一些侧面信息做出响应,并且存在多个甚至无数个追随者。它涵盖了元学习、个性化联邦学习、端到端学习和侧面信息的 Wasserstein 分布鲁棒优化等重要应用。由于存在上下文信息,现有的经典随机双层优化的单循环方法无法收敛。为了克服这个挑战,我们引入了一种基于多层蒙特卡罗(MLMC)技术的高效双循环梯度方法,并建立了其样本复杂度和计算复杂度。当专门应用于随机非凸优化时,我们的方法与现有下界相匹配。对于元学习,我们的方法的复杂度不依赖于任务数量。数值实验进一步验证了我们的理论结果。