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 の間の接続を実証します。