跳到主要內容
AI News HubLIVE
站內改寫1 分鐘閱讀

動態少步生成:統一動態計算與少步蒸餾的高效影片生成框架

文章摘要

影片擴散模型雖生成質量高,但計算成本高昂。現有少步蒸餾技術加速推理,卻忽略了不同噪聲水平下的計算需求差異。本文提出一種後訓練加速框架,將動態結構稀疏化直接融入蒸餾過程,聯合最佳化去噪步驟與模型稀疏性,將預訓練模型轉化為緊湊、步驟特定的混合模型(MoM)。透過漸進訓練策略和輸出滾動機制解決聯合最佳化不穩定性,並開發專用推理引擎。在Wan-14B模型上,該方法在4步蒸餾基礎上額外減少24%的每步FLOPs,實現1.2倍實際加速,相比50步教師模型達到30倍加速,同時保持競爭性生成質量。

來源arXiv Computer Vision作者: Yu Cheng, Siyue Yao, Zhongang Qi, Shanyan Guan, Wei Li, Fajie Yuan
動態少步生成:統一動態計算與少步蒸餾的高效影片生成框架
回報錯誤

更正管道尚未開通,可先複製下方文章資訊留存。

查看更正說明
直接讀正文

影片擴散模型(VDMs)在生成高質量影片方面表現出色,但其高昂的計算成本嚴重限制了實際應用。近年來,少步蒸餾技術透過減少推理所需的去噪步驟,顯著加速了生成過程。然而,這些方法通常在整個去噪階段採用靜態模型架構,忽略了不同噪聲水平下計算需求的差異。例如,在早期高噪聲階段可能需要更多的計算資源來重建結構,而在後期低噪聲階段,細節最佳化所需的計算量相對較小。針對這一冗餘問題,來自研究團隊的最新論文《Dynamic-in-Few-Step: Unifying Dynamic Computation and Few-Step Distillation for Efficient Video Generation》提出了一種創新的後訓練加速框架,透過將動態結構稀疏化直接融入蒸餾過程,充分利用了不同時間步的計算需求差異。

與傳統的後處理壓縮方法(如在固定擴散流水線上應用剪枝或量化)不同,該框架聯合最佳化去噪步驟和結構化模型稀疏性。具體而言,它將預訓練的VDM轉化為一個緊湊的、步驟特定的混合模型(MoM)。MoM的核心思想是為每個去噪步驟學習一個專門的稀疏子網路,從而在保持生成質量的同時減少計算量。為了應對聯合最佳化帶來的訓練不穩定性,研究人員引入了一種漸進訓練策略,並結合輸出滾動機制,確保跨時間步的結構決策能夠連貫學習,避免因稀疏模式突變導致的生成質量下降。此外,他們還開發了一個專門的推理引擎,以高效部署生成的MoM,實現實際推理加速。

該方法的優勢在於其正交性,可以與現有的少步蒸餾、模型剪枝或量化等技術相結合,效果顯著。在Wan-14B模型上的實驗表明,在4步蒸餾的基礎上,該方法額外移除了24%的每步FLOPs,帶來了1.2倍的實際牆鍾加速,相比原始的50步教師模型實現了30倍的加速,同時保持了具有競爭力的生成質量。這一工作為高效影片生成開闢了新途徑,有望推動擴散模型在即時影片生成、互動式應用等場景中的部署。

展開要點與分析

文章情報

工程師進階

要點

  • 提出動態少步生成框架,統一動態計算與少步蒸餾,實現高效影片生成。
  • 將動態結構稀疏化整合到蒸餾過程中,建立步驟特定的混合模型(MoM)。
  • 採用漸進訓練策略和輸出滾動機制,確保跨時間步的結構決策穩定學習。
  • 在Wan-14B上實現24%的每步FLOPs削減,1.2倍實際加速,30倍總加速比。

要點與分析由自動化流程生成,可能有誤,請結合原始來源核實。