Y-shaped Generative Flows
By: Arip Asadulaev , Semyon Semenov , Abduragim Shtanchaev and more
Potential Business Impact:
Helps AI learn patterns by moving data together.
Modern continuous-time generative models often induce V-shaped transport: each sample travels independently along nearly straight trajectories from prior to data, overlooking shared structure. We introduce Y-shaped generative flows, which move probability mass together along shared pathways before branching to target-specific endpoints. Our formulation is based on novel velocity-powered transport cost with a sublinear exponent (between zero and one). this concave dependence rewards joint and fast mass movement. Practically, we instantiate the idea in a scalable neural ODE training objective. On synthetic, image, and biology datasets, Y-flows recover hierarchy-aware structure, improve distributional metrics over strong flow-based baselines, and reach targets with fewer integration steps.
Similar Papers
An Eulerian Perspective on Straight-Line Sampling
Machine Learning (CS)
Makes computer art creation faster and simpler.
Adversarial Flow Models
Machine Learning (CS)
Makes AI create realistic pictures faster and better.
Generative Modeling with Continuous Flows: Sample Complexity of Flow Matching
Machine Learning (CS)
Makes AI create better pictures with less data.