アーカイブ / Generative AI / Computer Vision
ラベル別の分布から画像を生成し、経路の曲がりを抑えるBlock Flow
Block Flow
Block Flowの原論文から掲載した図です。細部はクリックして拡大できます。
原論文の図の説明を読む
Figure 1: Forward and reverse trajectories of 1-rectified flow [ 16 ] , fast ODE [ 18 ] , and block flow on a toy example. Intersections in forward trajectories lead to averaging effects, causing the resulting reverse trajectories to deviate from the true data distribution. We partition the toy example into two blocks and match each block with a separate Gaussian distribution.By employing block matching, block flow significantly reduces cross-label intersections in forward trajectories, resulting in straighter reverse trajectories.
概要
- ラベル別の画像群を学習可能なガウス分布と対応付け、生成経路を学ぶ。
- 著者らは、出発点の分散が学習用経路の曲がり具合の上限に関わることを示し、分散を調整する正則化を検討した。
- CIFAR-10の表では、FABR方式がRK45、NFE113でFID2.29、IS9.66と報告された。
補足：編集した模式図で仕組みを確認する
VISUAL EXPLAINER
図でつかむ、Block Flowの仕組み
- 01ラベル付き画像
- 02ラベル別に分割
- 03対応分布と向きを学習
- 04分布から画像生成
原論文の説明をもとにした模式図です。処理の細部は省略しています。
研究の背景
Block Flowは、ノイズに相当する出発点から画像へ移る経路を直線に近づける画像生成方法である。基になるフローマッチングは、経路の各位置で進む向きを学ぶ手法を指す。生成時にはその向きを繰り返し計算するため、経路の曲がりが計算誤差に関わる。
無関係な出発点と画像を組み合わせると、学習用の経路が交差しやすい。交差地点では異なる向きが平均され、生成経路が目標の画像分布から外れうる。著者らは画像のラベルを対応付けに使う。
手法
学習では画像とラベルを受け取り、ラベルごとに画像群を分ける。各群にガウス分布を対応させ、小さな符号化器で分布の中心と広がりを学ぶ。ガウス分布は中心と広がりで表す確率分布である。例えばCIFAR-10では、画像のラベルに対応する分布から出発点を取り、画像との途中位置で進む向きを学習する。
生成時はラベルに対応する分布から出発点を取り、学んだ向きを繰り返し計算して画像を得る。正則化は学習する値に制約を加えることで、ここでは出発点の分散を調整する。分散を小さくしすぎると生成画像の多様性が失われるため、著者らは複数の正則化方法を検討した。
新規性
著者らの提案は、画像をラベル別の群に分け、各群を学習可能なガウス分布と対応付ける「ブロック対応付け」である。異なるラベル間で学習用の経路が交差するのを減らし、生成経路を直線に近づける狙いがある。
出発点の分散と学習用経路の曲がり具合の上限との関係も示した。その関係を使い、分散を調整する複数の正則化方法を設計している。
従来手法との違い
通常のフローマッチングでは、出発点と画像を結ぶ途中位置で、進む向きを学ぶ。両者を独立に選ぶと経路が交差し、その場所で向きを平均するような学習が起きる。Block Flowはラベルで対応先を絞り、交差を減らす。
関連手法のRectified Flowは、再学習によって経路を直線に近づける方法として本文で説明される。Block Flowは、ラベルに対応する出発点の分布を学び、その分散を正則化で調整する。CIFAR-10の比較表には他手法も載るが、全手法の学習条件が同じとは確認できない。
実験結果
著者らはCIFAR-10とMNISTの画像生成を評価した。CIFAR-10の表にあるFIDは画像群の特徴分布の距離で低いほどよく、ISは生成画像の分類結果に基づく指標で高いほどよい。NFEは生成時に向きを推定する関数を呼んだ回数を表す。
著者らのCIFAR-10の結果では、正則化係数が1のFABR方式はRK45計算でNFE113、IS9.66、FID2.29だった。同じ表のFast OdeはRK45でNFE118、IS9.55、FID2.45。8回のEuler計算ではFABRのFIDが12.95、Fast Odeが13.52だった。確認できた本文の範囲では、データ分割や反復測定の条件は分からない。
応用の可能性
編集上の応用案 · 論文が実証した用途とは区別しています。
編集上の応用案：製造業で、製品カテゴリ別の外観画像を社内の検討用に生成する。入力は撮影済みの製品画像とカテゴリラベルで、画像を集めるセンサーとしてカメラを想定する。カテゴリに対応する分布から画像を出力し、担当者が外観を確認する。
適用には、対象カテゴリの画像と正しいラベルを用意し、撮影条件を整理したうえでモデルをそのデータに適応させる必要がある。生成画像を実際の検査に使えるかは、この研究結果からは判断できない。
応用先の候補からNauhに相談できます。
導入時の検証
導入時の検証案 · 対象データでの再評価が必要です。
導入時の検証案：同じ撮影用カメラで得た製品画像を集め、製品カテゴリのラベルを点検する。画像の大きさと撮影条件をそろえ、その画像とラベルでBlock Flowを適応させる。同じデータを使う通常のフローマッチングを比較対象に置く。
両方式で同数の画像を生成し、NFEをそろえたときのFIDと、目標FIDに達するNFEを測る。カテゴリごとの見た目の多様性と不自然な画像の割合も人が記録する。ラベルの誤りや撮影条件の違いを分けて集計し、実務画像でも結果が続くか確認する。
限界と課題
著者らは、画像に結び付く事前知識、通常はラベルが必要だと述べる。ラベルのないデータにはクラスタリングによる付与を提案する。また、ラベル別のガウス分布が大きく重なり、ラベル固有の特徴を十分に表せない点や、画像とラベルを併用した方式で狙った性能に達しなかった点を報告している。
編集上の検証事項は、誤ラベルや撮影条件の違いへの強さ、生成画像の多様性、実務画像への適応である。確認できた本文の範囲では、これらの実務条件での評価結果は示されていない。
出典
元のタイトル・要旨を確認する（英語）
Block Flow: Learning Straight Flow on Data Blocks
Flow-matching models provide a powerful framework for various applications, offering efficient sampling and flexible probability path modeling. These models are characterized by flows with low curvature in learned generative trajectories, which results in reduced truncation error at each sampling step. To further reduce curvature, we propose block matching. This novel approach leverages label information to partition the data distribution into blocks and match them with a prior distribution parameterized using the same label information, thereby learning straighter flows. We demonstrate that the variance of the prior distribution can control the curvature upper bound of forward trajectories in flow-matching models. By designing flexible regularization strategies to adjust this variance, we achieve optimal generation performance, effectively balancing the trade-off between maintaining diversity in generated samples and minimizing numerical solver errors. Our results demonstrate competitive performance with models of the same parameter scale.Code is available at \url{https://github.com/wpp13749/block_flow}.
論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新：2026-10-01