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 之间的联系。