NPG 魔法報紙

每日深稿 第 133 期

SpecKCast 讓 AI 影片生成快 6 倍,記憶體只多 0.05 GiB

用頻域預測取代重複運算,在不需訓練的情況下大幅加速 Diffusion Transformer。

文/魔法報紙編輯部 · 依 Hugging Face Papers 報導整理 · 看原文

SpecKCast 讓 AI 影片生成快 6 倍,記憶體只多 0.05 GiB
圖片來源:Hugging Face Papers(原報導配圖)

三個重點

  1. 在 FLUX.1-dev 等模型上實現最高 6 倍加速
  2. 額外記憶體開銷低於 0.05 GiB
  3. 採用免訓練的頻域卡爾曼濾波預測機制

頻域預測破解「重複運算」瓶頸

Diffusion Transformer (DiT, 一種用於生成高保真影像與影片的 AI 架構) 的核心痛點在於推論成本。為了生成一張圖或一段影片,模型需要在多個去噪步驟中反覆執行 Transformer 評估,這導致了巨大的計算負擔。現有的特徵快取 (Feature Caching) 技術試圖透過重用特徵來降低成本,而較新的預測方法則嘗試外推特徵以跳過部分步驟。然而,這些方法存在一個根本缺陷:它們假設特徵在所有空間頻率成分上的變化是均勻的,且去噪過程是線性的。事實上,不同頻段的變化速率差異巨大,且去噪進度並非均勻分佈。這種錯配導致跳過的步驟越多,預測誤差越大,最終限制了加速的潛力。

每個頻段擁有自己的「時鐘」

SpecKCast 提出了一種免訓練 (Training-free) 的解決方案,核心思想是在頻域進行預測。該方法對模型輸出應用二維離散餘弦變換 (2D DCT, 一種將空間訊號轉換為頻率係數的數學方法),並為每個頻率帶建立輕量級的卡爾曼濾波器 (Kalman Filter, 一種用於追蹤動態系統狀態的演算法)。關鍵創新在於,每個濾波器都沿著一個基於雜訊排程 (Noise Schedule) 的「頻帶特定時間座標」進行追蹤。這意味著高頻與低頻成分不再被強制同步更新,而是根據它們各自的變化速率獨立預測。濾波器同時追蹤係數的值與變化率,只有當預測不確定性超過特定閾值時,才會觸發 Transformer 的實際評估。

實測數據:6 倍加速與極低記憶體負擔

在 FLUX.1-dev、Qwen-Image、Wan2.1-14B 和 HunyuanVideo 等主流模型上的實驗顯示,SpecKCast 在加速比與生成保真度之間取得了優於現有快取與預測方法的平衡。具體數據顯示,該方法可實現最高 6 倍的推論速度提升,而額外佔用的記憶體不到 0.05 GiB。這一結果證明了頻域預測不僅能顯著降低計算開銷,還能將資源佔用控制在極低水平,使其在硬體資源受限的環境中具備高度的實用性。

主筆觀點

我認為 SpecKCast 的真正價值不在於「加速」本身,而在於它揭示了 Diffusion Transformer 推論優化的一個盲點:我們長期以來錯誤地假設去噪過程是「時間上均勻」的。現有的快取技術之所以在跳過步驟時容易崩潰,是因為它們用同一把尺去量所有頻段的變化,而 SpecKCast 承認了不同頻率成分有各自的「節奏」。這是一種更符合物理直覺的建模方式。

反過來看,有人可能會質疑「免訓練」意味著它無法針對特定模型做深度優化,上限可能不如端到端訓練的方法。但對於白領與開發者而言,「即插即用」且記憶體開銷極低 (0.05 GiB) 的特性,意味著它可以無縫嵌入現有的生產管線,而不需要重新訓練或調整超參數。這降低了技術門檻,讓中小團隊也能享受大模型加速的紅利。

接下來該關注的訊號是:各大模型廠商 (如 Stability AI 或阿里通義) 是否會在官方推理引擎中整合類似的頻域預測機制。如果主流框架開始預設啟用這種「自適應跳步」策略,那麼 Diffusion 模型的部署成本將出現斷層式下降,這將直接影響到 AI 影像生成服務的定價模型與市場格局。

以上為魔法報紙編輯部觀點,不代表原報導立場。

原始報導:Hugging Face Papers《SpecKCast: Spectral Kalman Forecasting for Diffusion Acceleration》
https://huggingface.co/papers/2610.26837
事實段落只寫原文有的內容;「編輯觀察」是本報的判斷,不是原文內容。

回到今天的魔法報紙 →