Adapting Noise to Data: Generative Flows from learned 1D Processes

ICLR 2026 Conference Submission13384 Authors

Keywords: generative modelling, flow matching, noise learning, optimal transport

Abstract: We introduce a general framework for constructing generative models using one-dimensional noising processes. Beyond diffusion processes, we outline examples that demonstrate the flexibility of our approach. Motivated by this, we propose a novel framework in which the 1D processes themselves are learnable, achieved by parameterizing the noise distribution through quantile functions that adapt to the data. Our construction integrates seamlessly with standard objectives, including Flow Matching and consistency models. Learning quantile-based noise naturally captures heavy tails and compact supports when present. Numerical experiments highlight both the flexibility and the effectiveness of our method.

Primary Area: generative models

Submission Number: 13384