Abstract:Forecasting stochastic nonlinear dynamical systems under the influence of conditioning variables is a fundamental challenge repeatedly encountered across the biological and physical sciences. While flow-based models can impressively predict the temporal evolution of probability distributions representing possible outcomes of a specific process, existing frameworks cannot satisfactorily account for the impact of conditioning variables on these dynamics. Amongst several limitations, existing methods require training data with paired conditions and are developed for discrete conditioning variables. We propose Conditional Variable Flow Matching (CVFM), a framework for learning flows transforming conditional distributions with amortization across continuous conditioning variables - permitting predictions across the conditional density manifold. This is accomplished through several novel advances, in particular, simultaneous sample conditioned flows over the main and conditioning variables, alongside a conditional Wasserstein distance and kernel facilitating conditional optimal transport. Collectively, these advances allow for learning system dynamics provided measurement data whose states and conditioning variables are not in correspondence. We demonstrate CVFM on a suite of increasingly challenging problems, including discrete and continuous conditional mapping benchmarks, image-to-image domain transfer, and modeling the temporal evolution of materials internal structure during manufacturing processes. We observe that CVFM results in improved performance and convergence characteristics over alternative conditional variants.

Predictive Flows for Faster Ford-Fulkerson

Flow Prediction in Spatio-Temporal Networks Based on Multitask Deep Learning.

Learning-Augmented Maximum Flow

UnSAMFlow: Unsupervised Optical Flow Guided by Segment Anything Model

SAMFlow: Eliminating Any Fragmentation in Optical Flow with Segment Anything Model

FLOWPROPHET: Generic and Accurate Traffic Prediction for Data-Parallel Cluster Computing

VectorFlow: Combining Images and Vectors for Traffic Occupancy and Flow Prediction

VectorFlow: Combining Images and Vectors for Traffic Occupancy and Flow Prediction

MemFlow: Optical Flow Estimation and Prediction with Memory

Flow Matching for Scalable Simulation-Based Inference

Sinkhorn-Flow: Predicting Probability Mass Flow in Dynamical Systems Using Optimal Transport

SAFIT: Segmentation-Aware Scene Flow with Improved Transformer

Fisher Flow Matching for Generative Modeling over Discrete Data

FlowSDF: Flow Matching for Medical Image Segmentation Using Distance Transforms

Flow Matching in Latent Space

BUFF: Boosted Decision Tree based Ultra-Fast Flow matching

Conditional Variable Flow Matching: Transforming Conditional Densities with Amortized Conditional Optimal Transport

TransFlow: Transformer as Flow Learner

GMFlow: Learning Optical Flow Via Global Matching

Optical Flow as Spatial-Temporal Attention Learners

Moving Object Segmentation: All You Need Is SAM (and Flow)