arXiv Machine Learning

Beyond Random Couplings: Contrastive Noise Alignment in Generative Flows

arXiv Machine Learning
Sep 4

Beyond Straightness: Non-Crossing Flow Matching via Quantile AlignTree Coupling

The paper introduces Quantile AlignTree Flow Matching (QAT‑FM), a structured coupling method that builds a hierarchical, quantile‑aligned tree to connect a Gaussian prior with a target distribution. QAT‑FM achieves efficient coupling construction in ≠ Nd log N time and allows per‑pair source sampling in ≠ d time, enabling scalable training for high‑dimensional generative tasks. The authors prove that the coupling preserves marginal consistency, produces non‑crossing interpolation paths, and improves path separation compared to independent coupling, while also extending naturally to conditional generation.

By Junyi Lin, Mengyu Li, Jingxuan Hu, Kejun He, Cheng Meng
arXiv AI
Aug 12

Flow Straight to Reality: Perceptually Consistent Flow Matching for Efficient Image Restoration

arXiv:2608. 10544v1 Announce Type: cross Abstract: Image restoration is fundamentally constrained by the tradeoff between distortion and perception: minimizing pixel-wise error yields over-smoothed results, whereas optimizing for perceptual realism often introduces structural deviations.

By Sangwoo Jo, Donggeun Ko, Jayeon Kang, Youngsang Kwak, Jaehwa Kwak, Sungjoon Choi
Hugging Face Trending Papers
Jul 13

Self-Consistent Flow: Unifying Velocity and Endpoint Prediction for Rectified Flow Models

In rectified-flow-based generative models, the neural network can be trained to predict two different targets, such as the instantaneous velocity or the data endpoint, to perform denoising. Although prior work shows that these parameterizations lead to different empirical behaviors, the mechanisms underlying their respective advantages remain to be underexplored, and how to combine them effectively is still unclear.

arXiv Machine Learning
Sep 3

CAT-Flow: Curvature-Adaptive sTeps for Flow Matching

The paper introduces CAT-Flow, a pair of lightweight, training‑free algorithms—CAT‑OV and CAT‑OT—that adapt step‑sizes during Flow Matching inference by estimating curvature in time or state space. These methods avoid extra neural evaluations and achieve constant‑order truncation error bounds. Experiments show that CAT‑OV and CAT‑OT improve image quality metrics across four text‑to‑image Flow Matching models, cutting the required generation steps by up to 40%.

By Qinchan Li, Pedro Cisneros-Velarde, Keru Fu, Samuel Antunes Miranda, Sharan Vaswani, Hao Zhang