新着研究 / Generative AI
再計算までの余裕で生成の歩幅を変えるSoftCache、画像・動画の高速化で元出力への忠実さを改善
SoftCache
SoftCacheの原論文から掲載した図です。細部はクリックして拡大できます。
原論文の図の説明を読む
Figure 1: SoftCache: how far, not just whether. Hard caching walks the fixed grid of scheduled nodes one node at a time, deciding only whether to refresh or reuse . SoftCache also sets the step size: a reused step the cache still supports skips several nodes and lands between them (inset).
概要
- キャッシュ更新の閾値までの余裕を、再利用するかの判定だけでなく生成ステップ幅へ使う。
- 余裕が大きい区間を長く進み、更新が近づくと元の刻みへ戻す学習不要の方式を提案した。
- FLUX、HunyuanVideo、Wan2.1で、約3.5〜6.9倍の高速化帯における元出力との一致を比較した。
補足:編集した模式図で仕組みを確認する
VISUAL EXPLAINER
図でつかむ、SoftCacheの仕組み
- 01生成途中の状態
- 02キャッシュの余裕を計算
- 03歩幅と再利用値を調整
- 04次の状態へ更新
原論文の説明をもとにした模式図です。処理の細部は省略しています。
研究の背景
拡散モデルやフローモデルは、何度も大きなネットワークを通して画像や動画を作る。近い生成段階の計算が似ているため、過去の中間値を保存して再利用すれば処理を速くできる。
従来の動的キャッシュは、古さを示す信号が閾値を超えたら再計算し、それまでは保存値を使う。ただし再利用する段階も生成の歩幅は通常どおりである。SoftCacheは、この信号が示す余裕を次にどこまで進むかへ利用し、元の刻みを超えて移動する。
手法
再計算の閾値と現在の古さ信号の差を、ヘッドルームという余裕として扱う。余裕が大きければ保存値を信用できるとして長いステップを取り、通常なら訪れる複数の時点を飛ばす。閾値へ近づくと歩幅を元の大きさへ戻す。
長い移動先が当初の生成時点の間へ入るため、残りのスケジュールも組み直す。保存した残差は直近二回の新しい評価を使い直線的に外挿する。例えば変化の少ない区間は大きく進み、キャッシュが古くなる区間で細かく進む。追加学習や基盤ごとの新たな設定を増やさない構成として提案されている。
新規性
キャッシュの信頼度を再計算と再利用の二択へ潰さず、再利用するステップの長さへ連続的に反映する。更新する時点だけでなく、生成が次に到着する時点も入力に応じて変わる。
固定された時間格子上で中間値を予測する方式とも異なり、格子の間に着地できる。計算を省くことと生成経路を調整することを結び付け、同程度の実時間短縮で元の生成結果を保つことを狙う。
従来手法との違い
FLUXではTeaCache、SeaCache、DPCache、TaylorSeerと比較した。動画では計算予算の都合からSeaCacheとDPCacheを比較対象とし、画像と同じ全方式を評価したわけではない。
速度を近づけた帯域で、同じ指示と乱数種によるキャッシュなしの出力への忠実度を比べる。著者らは四つの忠実度指標すべてで優位と報告するが、生成物単体の好ましさや映像品質の得点差はそれほど大きくないとも述べる。
実験結果
FLUX.1-devはDrawBenchの200指示で1024四方の画像、HunyuanVideoはVBenchの946指示、Wan2.1は100指示で動画を生成した。基準はいずれも50回のEulerステップである。
PSNR、SSIM、LPIPS、DreamSimを用いて元出力との一致を測り、約3.5〜6.9倍の高速化帯を比較した。例えばHunyuanVideoの約5.4倍の帯ではPSNR22.36、LPIPS0.246を報告する。全く同一の画像を保った高速化ではなく、近似による品質と速度の交換である。
応用の可能性
編集上の応用案 · 論文が実証した用途とは区別しています。
編集上の応用案は、社内の画像・短動画試作サービスで応答時間を短くする用途である。指示と乱数種を固定した通常出力を基準にし、キャッシュ付き生成の待ち時間と差分を管理する。
利用者には高速試作で構図を検討してもらい、細部の一致が必要な案件は別の基準で確認する運用が考えられる。元モデルの出力へ近いことを、指示への正確さや品質の高さそのものと同一視しない設計が必要になる。
応用先の候補からNauhに相談できます。
導入時の検証
導入時の検証案 · 対象データでの再評価が必要です。
導入時の検証案では、実際の指示集合を固定し、キャッシュなしと既存キャッシュを基準にする。同じGPU、解像度、ステップ数、乱数種で実時間、最大メモリ、元出力との差を測定する。
細かい文字、繰り返し模様、速い動きを含む例を分け、平均の忠実度が良くても重要な対象が壊れていないかを人が確認する。指示ごとの遅延のばらつきと再計算回数を記録し、閾値を変えた際の品質曲線も残す。
限界と課題
評価の中心はキャッシュなしの同じ生成器への忠実さであり、元生成器自身の誤りを直す研究ではない。画素や知覚特徴が近くても、用途で重要な小さな文字や数の違いが残る可能性はある。
編集上の確認点は、別のサンプラーや追加条件を使う設定でも余裕信号が誤差を予測するかである。報告された速度域を任意の環境へそのまま当てはめず、使う基盤と生成条件で実測して許容する近似範囲を決める必要がある。
出典
元のタイトル・要旨を確認する(英語)
SoftCache: Adaptive Step Size Caching for Diffusion Transformers
Diffusion and flow models generate a sample over iterative steps, each a full pass through a large network, and that pass dominates generation cost. Training-free caching accelerates them by reusing activations from recent evaluations instead of recomputing them. Dynamic caching methods decide when to refresh by accumulating a cheap probe of the network input into a staleness signal and spending a fresh evaluation once it crosses a threshold. They perform hard caching: the signal is a binary switch between refresh and reuse, and every step, fresh or cached, advances by the prescribed step size. We observe that the signal carries much more than this switch. How far it sits below the threshold, its headroom, predicts the error induced by reuse. We propose SoftCache, which performs soft caching: instead of a binary switch, the headroom sets how far each cached step goes. Where the headroom is large, the sampler jumps past several scheduled nodes, and as a refresh nears, its steps shrink back toward the original step. These adaptive jumps may land between the scheduled nodes, taking the sampler off its fixed grid. SoftCache requires no training and adds no per-backbone settings: on FLUX for images and on HunyuanVideo and Wan2.1 for video, at speedups from $3.5\times$ to $6.9\times$, it is ahead of every baseline on all four fidelity metrics.
論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。収集:2026-10-17




