Conditional Flow Matching for Transport Between Markov Processes
cs.LG, stat.ML
Submitted: 2026-10-05
Updated: 2026-10-05
Code: https://github.com/Alljoined/ENIGMA
License: http://creativecommons.org/licenses/by/4.0/
The gist: Motivated by sequence-to-sequence transport in the context time-series domain adaptation, we study the problem of transportation between trajectories of Markov processes.
Terminology
Abstract
Motivated by sequence-to-sequence transport in the context time-series domain adaptation, we study the problem of transportation between trajectories of Markov processes. Given a limited number of trajectories from source distribution and the target distribution, we formulate a flow matching based algorithm which learns a transport map from the source to target trajectory distribution, while preserving the Markov structure. We show that this is consistent in the population limit and derive finite-sample error bounds under mixing time assumptions, following the analysis of classical statistical problems including regression (Nagaraj et al., 2020), principal component analysis (Kumar and Sarkar, 2023), and matrix concentration (Neeman et al., 2024) in the Markov setting. We complement that with a lower-bound construction showing that a mixing-time dependent sample complexity is unavoidable even with regular Gaussian conditional transitions. We evaluate on synthetic and real-world data. For image retrieval from electroencephalography (EEG) on THINGS-EEG2 (Gifford et al., 2022), the task is to identify the viewed image from EEG signals captured from human subjects, which suffers from high inter subject variability. We augment the ENIGMA decoder (Kneeland et al., 2026) with a conditional flow before its subject-specific temporal map. This improves mean top-5 retrieval accuracy from 43.87% to 49.05%, an 11.82% relative improvement.
Sources
- A short note on an inequality between KL and TV
- Generative Modeling with Continuous Flows: Sample Complexity of Flow Matching
- ENIGMA: EEG-to-Image in 15 Minutes Using Less Than 1% of the Parameters
- Is Flow Matching Just Trajectory Replay for Sequential Data?
- Rectified Flow: A Marginal Preserving Approach to Optimal Transport
- Low-Dimensional Adaptation of Rectified Flow: A Diffusion and Stochastic Localization Perspective
- Perturbation theory for Markov chains via Wasserstein distance
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks