cs.LGOct 20, 2026

Minimax-Optimal Predictive Flow Matching under Distribution Drift

Authors: Aadithya Srikanth, Vaneet Aggarwal

Organizations: Purdue University, West Lafayette

Abstract

Flow matching learns a generative velocity field by regression, but under distribution drift a field must be deployed before the current data arrive. We study this predictive problem for a fixed, possibly nonconvex, neural class, with dynamic regret against the best represented field of each round. We propose DyProd-FM, which refits flow matching on dyadic historical prefixes and selects among the fits with a sleeping Prod master. Without knowing any drift budget, it attains regret O~(min⁡{Nω+1,1+TΩ})\widetilde{O}(\min\{N_ω+1,1+\sqrt{TΩ}\}) in the relative velocity risk variation ΩΩ and O~(min⁡{Nμ+1,1+T1/3Vμ2/3})\widetilde{O}(\min\{N_μ+1,1+T^{1/3}V_μ^{2/3}\}) in the target-law variation Vμ=∑tW2(μt+1,μt)V_μ=\sum_t W_2(μ_{t+1},μ_t), up to approximation and fitting costs. Matching lower bounds for fixed classes show that both rates are optimal in TT and in the budgets up to logarithmic factors. Prior optimal dynamic-regret guarantees rely on a convex decision set with strongly convex or exp-concave losses, whereas our class may be nonconvex and no curvature in its parameters is assumed. The target-law rate rests on a second-order link between the two variations, which the base noise of flow matching provides.

Figures & tables

Appendix figures & tables2 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Velocity Scheduled Flow Matching

    Jul 13, 2026Vitalii BondarFlow MatchingConditional Flow Matching

  2. Flow Matching under Noisy Latent Structure: Beyond Exact Low-Dimensional Support

    Sep 30, 2026Lifeng Hao, Shaolin JiFlow MatchingFlow-Based Generative Modeling

  3. Same Flow, Different Paths: Variance Reduction in Flow Matching

    Sep 15, 2026Alexander TyurinFlow MatchingStochastic Optimization Convergence