Multi-Scale Generative Modeling with Heat Dissipation Flow Matching
Abstract
Diffusion models are widely used in image generation, with most relying on noise-based corruption and denoising. A distinct branch instead uses blur as the main corruption, preserving better color budgets and multi-scale detail by providing multi-scale priors. However, blur-based models remain in SDE-based frameworks and are not integrated into ODE-based frameworks, such as Flow Matching (FM). Meanwhile, in the blur-based formulation, the classical inverse heat-dissipation (IHD) process faces an ill-posed challenge. Moreover, under the data-manifold assumption, regressing blurred images from high-dimensional noise (or velocity) space is also difficult. We propose Heat Dissipation Flow Matching (HDFM), which introduces a continuous blurred (heat-dissipation) process into FM to inject multi-scale priors. HDFM aligns an interpolated heat-dissipation path to address ill-posedness and adopts -prediction to mitigate high-dimensional regression difficulty. Toy experiments and ablation studies show that HDFM consistently benefits from both blur and -prediction. The performance of HDFM outperforms most baseline methods on all datasets.
Keywords
Cite
@article{arxiv.2605.19371,
title = {Multi-Scale Generative Modeling with Heat Dissipation Flow Matching},
author = {Jun Ma and Hanquan Zhang and Yanjun Qin and Haoyuan Guan and Ke Zhang},
journal= {arXiv preprint arXiv:2605.19371},
year = {2026}
}