Blockwise Flow Matching: Improving Flow Matching Models For Efficient High-Quality Generation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Park, Dogyun, Lee, Taehoon, Joo, Minseok, Kim, Hyunwoo J.
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911229484728320
author Park, Dogyun
Lee, Taehoon
Joo, Minseok
Kim, Hyunwoo J.
author_facet Park, Dogyun
Lee, Taehoon
Joo, Minseok
Kim, Hyunwoo J.
contents Recently, Flow Matching models have pushed the boundaries of high-fidelity data generation across a wide range of domains. It typically employs a single large network to learn the entire generative trajectory from noise to data. Despite their effectiveness, this design struggles to capture distinct signal characteristics across timesteps simultaneously and incurs substantial inference costs due to the iterative evaluation of the entire model. To address these limitations, we propose Blockwise Flow Matching (BFM), a novel framework that partitions the generative trajectory into multiple temporal segments, each modeled by smaller but specialized velocity blocks. This blockwise design enables each block to specialize effectively in its designated interval, improving inference efficiency and sample quality. To further enhance generation fidelity, we introduce a Semantic Feature Guidance module that explicitly conditions velocity blocks on semantically rich features aligned with pretrained representations. Additionally, we propose a lightweight Feature Residual Approximation strategy that preserves semantic quality while significantly reducing inference cost. Extensive experiments on ImageNet 256x256 demonstrate that BFM establishes a substantially improved Pareto frontier over existing Flow Matching methods, achieving 2.1x to 4.9x accelerations in inference complexity at comparable generation performance. Code is available at https://github.com/mlvlab/BFM.
format Preprint
id arxiv_https___arxiv_org_abs_2510_21167
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Blockwise Flow Matching: Improving Flow Matching Models For Efficient High-Quality Generation
Park, Dogyun
Lee, Taehoon
Joo, Minseok
Kim, Hyunwoo J.
Computer Vision and Pattern Recognition
Recently, Flow Matching models have pushed the boundaries of high-fidelity data generation across a wide range of domains. It typically employs a single large network to learn the entire generative trajectory from noise to data. Despite their effectiveness, this design struggles to capture distinct signal characteristics across timesteps simultaneously and incurs substantial inference costs due to the iterative evaluation of the entire model. To address these limitations, we propose Blockwise Flow Matching (BFM), a novel framework that partitions the generative trajectory into multiple temporal segments, each modeled by smaller but specialized velocity blocks. This blockwise design enables each block to specialize effectively in its designated interval, improving inference efficiency and sample quality. To further enhance generation fidelity, we introduce a Semantic Feature Guidance module that explicitly conditions velocity blocks on semantically rich features aligned with pretrained representations. Additionally, we propose a lightweight Feature Residual Approximation strategy that preserves semantic quality while significantly reducing inference cost. Extensive experiments on ImageNet 256x256 demonstrate that BFM establishes a substantially improved Pareto frontier over existing Flow Matching methods, achieving 2.1x to 4.9x accelerations in inference complexity at comparable generation performance. Code is available at https://github.com/mlvlab/BFM.
title Blockwise Flow Matching: Improving Flow Matching Models For Efficient High-Quality Generation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2510.21167