Search papers, labs, and topics across Lattice.
This paper introduces Expanding Flow Maps (EFMs), a novel class of flow-based generative models that allow for dynamic expansion of state dimensions through an innovative interpolant approach. By employing an expand operator that augments the state space with conditional noise and a transport map that advances the state along the interpolant, EFMs enable efficient few-step generation across both continuous and discrete modalities. The results demonstrate that this framework not only recovers existing flow models as special cases but also facilitates variable-size graph and sequence generation, significantly enhancing the flexibility of generative modeling.
Expanding Flow Maps redefine generative modeling by allowing output size to be a learned, controllable parameter, enabling unprecedented flexibility in generation tasks.
Flow-based generative models have enabled remarkable progress in fast and controllable generation across continuous and discrete state spaces, yet existing parameterizations are constrained to fixed dimensions or fixed sequence lengths. Here, we introduce Expanding Generative Flows (EFlows), which define flows between distributions of increasing dimensionality along an expanding interpolant that grows the state by augmenting it with conditional noise. Building on this construction, we propose Expanding Flow Maps (EFMs), a new class of flow maps that distill the expanding interpolant into efficient few-step generative models. Each EFM factors the map between any two timesteps into two learnable operations: an expand operator, which augments the state space with new coordinates or tokens conditioned on the current state, and a transport map, which pushes the expanded state forward along the interpolant. Composing these operators yields a single map that jointly expands and denoises the state, recovering existing fixed-canvas flows and flow maps as the special case in which the expand operator is the identity. We further extend the framework to the discrete simplex, enabling variable-size graph generation and variable-length sequence generation. Across both continuous and discrete modalities, we establish EFlows and EFMs as a principled framework for settings in which output size is itself a learned, controllable degree of freedom.
