helblazer811/Diffusion-Explorer

Interactive visualizations of the geometric intuition behind diffusion models.

What it solves

Diffusion Explorer 是一個教育工具，旨在幫助用戶理解擴散模型與流式生成模型的幾何直覺。它通過提供這些模型如何生成數據的互動式視覺化，彌補了複雜數學概念與視覺直覺之間的差距。

How it works

該工具實現了幾種訓練目標，包括 Flow Matching 和 Denoising Score Matching。它允許用戶視覺化預訓練模型生成的樣本隨時間變化的動態過程，並觀察樣本在訓練過程中的演變。此外，它還支持在自定義的、手繪的分佈上進行訓練，使學習過程更加具體化。

Who it's for

此工具是為學生、研究人員和開發者設計的，他們希望視覺化地探索擴散模型與流式生成 AI 的概念，特別是那些對學習 Rectified Flow 和 optimal transport 感興趣的人。

Highlights

  • 互動式視覺化生成模型動態。
  • 支持 Flow Matching 和 Denoising Score Matching 目標。
  • 能夠在自定義的手繪分佈上訓練模型。
  • 專門為 Rectified Flow 提供解釋器，以展示直線路徑與 optimal transport 之間的聯繫。