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

在流形假設下可證明學習擴散模型:崩潰與細化

文章摘要

本文提出崩潰與細化機制,解釋擴散模型如何在低維流形上高效學習得分函數,並基於此設計SiLD框架,理論證明樣本複雜度取決於內在維度,實驗驗證性能優於VAE基潛在擴散模型。

來源arXiv Machine Learning作者: Wei Huang, Andi Han, Mingyuan Bai, Huanjian Zhou, Qixin Zhang, Taiji Suzuki, Kenji Fukumizu
在流形假設下可證明學習擴散模型:崩潰與細化
報告錯誤

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

查看更正說明
直接讀正文

近年來,擴散模型在生成高維數據方面展現出卓越能力,但其訓練過程如何有效學習得分函數並避開維度災難(當數據位於低維流形上時)一直缺乏理論解釋。最新研究《Provably Learning Diffusion Models under the Manifold Hypothesis: Collapse and Refine》揭示了這一謎題背後的崩潰與細化機制,並提出了名為Score-induced Latent Diffusion (SiLD)的兩階段框架。該研究由Wei Huang等七位作者完成,於2026年5月16日提交至arXiv。

通過深入分析擴散模型的訓練動力學,作者發現得分函數自身的幾何特性驅動了訓練過程。在小噪聲尺度下,得分的發散奇異性迫使去噪映射快速“崩潰”到數據流形投影上,實現維度降低;在中等噪聲尺度下,訓練則專注於“細化”所學流形上的內在密度分佈。這一過程無需顯式的流形學習,而是通過單一的去噪得分匹配目標自然湧現,從而替代了VAE基潛在擴散模型中啓發式的KL正則化,實現了更簡潔高效的兩階段學習框架。

理論分析表明,SiLD的樣本複雜度僅依賴於數據的固有維度,而非環境維度,這為擴散模型在高維數據上的高效性提供了嚴格證明,是連接流形假設與可學習性的重要理論進展。實驗部分,研究者在Stacked MNIST、CelebA變體以及分子生成基準上驗證了SiLD的性能。結果顯示,SiLD在生成質量上與VAE基潛在擴散模型相當或更優,且在重構任務上持續改進,充為支持了理論預測。該工作不僅為理解擴散模型的訓練動力學提供了新視角,也為基於得分幾何設計更高效的生成模型開闢了道路。

展開要點與分析

文章情報

工程師進階

要點

  • 發現擴散模型訓練中的崩潰與細化機制
  • 提出Score-induced Latent Diffusion (SiLD)框架
  • 證明樣本複雜度取決於數據內在維度
  • 在多個基準上匹配或超越VAE基模型

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