拡散モデルの時間割当を最適化するCBS、FID約35%改善

要約

拡散モデルの生成プロセスを複数の特化サブネットワークに分割する「Complexity-Balanced Splitting（CBS）」フレームワークが提案された。de Boorの等分布原理に基づき、モデリングが困難な時間領域により多くの表現容量を割り当てる設計となっている。複雑度の推定にはフローのDirichletエネルギーと軌跡加速度という2つの指標を活用する。SiT・JiT・UNetの複数アーキテクチャで検証した結果、推論コストを増やさずにSiT-XLのFIDを約35%改善した。