arXiv Machine Learning

Multi-Marginal Flow Matching with Adversarially Learnt Interpolants

arXiv Machine Learning
Oct 23

Learning Domain Trajectories with Flow Matching for Gradual Domain Adaptation

The paper introduces FlowGDA, a flow‑matching framework for gradual domain adaptation that models adaptation as continuous transport on a statistical manifold. FlowGDA learns time‑dependent velocity fields to generate smooth domain trajectories, avoiding adversarial Min–Max optimization by using a simulation‑free regression objective. The method incorporates source‑anchored transport, semantic consistency loss, and a two‑stage optimization scheme to prevent semantic drift, achieving superior performance over state‑of‑the‑art techniques.

By Yubo Huang, Yushe Cao, Zixi Wang, Jingzehua Xu, Xin Lai, Xiongyu Hu, Shuai Zhang
arXiv AI
Oct 9

Finsler Flow Matching: Dynamics-Aware Geodesic Interpolation for Single-Snapshot Trajectory Inference

The paper introduces Finsler Flow Matching (FFM), a method for learning continuous stochastic dynamics from discrete Markov transition graphs in single‑cell snapshot data. FFM constructs a Finsler geometry using first and second local moments, learns directed geodesics via neural networks, and employs these to build geometry‑aware stochastic paths that can be simulated without explicit score or flow matching. Benchmark tests show that FFM better recovers intermediate cellular states, especially when dynamics are directional or anisotropic.

By Niklas Canova, Jonas Simon Fleck
arXiv Machine Learning
Aug 5

Trajectory inference via Acceleration Matching

arXiv:2608. 03916v1 Announce Type: new Abstract: Trajectory inference is a fundamental problem in many scientific domains: given a collection of unpaired snapshots of observations at discrete time points, the goal is to generate smooth trajectories that best resemble and interpolate the data.

By Bartolo Dazzini, Giovanni Conforti, Alain Durmus, Aram-Alexandre Pooladian
arXiv Machine Learning
Oct 2

A Flow Matching Algorithm for Many-Shot Adaptation to Unseen Distributions

The paper introduces Function Projection for Flow Matching (FP‑FM), an algorithm that conditions generative models directly on samples from a target distribution. FP‑FM learns basis functions spanning velocity fields for training distributions and adapts to new distributions via a simple least‑squares projection, enabling efficient sample generation without additional training at inference time. Variants of the method trade expressivity for computational cost by enriching coefficient calculations, and the approach shows markedly better precision and recall than baselines, especially on unseen distributions.

By Tyler Ingebrand, Ruihan Zhao, Kushagra Gupta, David Fridovich-Keil, Sandeep P. Chinchali, Ufuk Topcu
arXiv Machine Learning
Sep 30

ALICE: In-context, Zero-shot, Mutual Information Estimation

ALICE is a foundation model that estimates mutual information (MI) without per‑distribution training. Trained only on synthetic distributions, it acts as an in‑context estimator of rectified‑flow velocity fields, producing MI via a fixed identity that integrates squared differences between joint and conditional fields. The authors validate ALICE on a challenging benchmark and demonstrate its applicability to unseen data in biology, genetics, and neuroscience, achieving performance comparable to neural estimators trained separately for each distribution.

By Giulio Franzese, Simone Rossi, Pietro Michiardi
arXiv Machine Learning
Aug 5

Information-Geometric Forward Policy Training in GFlowNets

arXiv:2608. 03967v1 Announce Type: cross Abstract: Generative Flow Networks (GFlowNets) have emerged as a flexible framework for amortised inference over discrete and mixed discrete-continuous objects, requiring only an unnormalised target density specified through a reward.

By Yordan Raykov, Rodrigo Veiga
arXiv Machine Learning
4d ago

Coarse-Graining Optimal Transport for Rollout-Free Post-Training of Flow Matching

The paper introduces ReCouple, a framework that constructs coarse-grained optimal transport (OT) couplings to improve flow matching without requiring rollouts. Extensions ReCouple+ and ReCouple++ address continuous-source settings and add spatial regularization, respectively. Experiments show that these methods balance coupling optimality, neural network realizability, and generation quality, especially in low NFE regimes.

By Soon Hoe Lim, Aditi Gupta, Michael W. Mahoney, N. Benjamin Erichson