huggingface / huggingface/diffusers

[Feature Request] Add Implement of 'Kuramoto Orientation Diffusion'

オープン
#12,903 コメント 1 件 リアクション 0 件 担当者 0 名 GitHub で見る
stale
主要言語
Python
スター
34.5k
フォーク
7.3k
平均マージ
3日 3時間
マージ済み PR(30日)
91

説明

Source code: https://github.com/KingJamesSong/OrientationDiffusion (the official implement use pure `torch`)

This paper introduces a nonlinear score-based generative framework that incorporates stochastic Kuramoto dynamics to model orientation-rich data. By formulating the forward process as synchronization and the reverse process as desynchronization, our method brings biologically inspired inductive biases into the diffusion process. Through wrapped Gaussian transitions and periodicityaware networks, the model naturally captures the geometry of angular data. Experiments show that our approach outperforms conventional baselines on orientation-dense datasets and remains competitive on general image generation tasks. This work highlights the potential of integrating biologically inspired synchronization dynamics as structured priors in generative modeling of orientation-dense data, paving the way for incorporating nonlinear dynamics into generative models.

Image

Governing stochastic differential equations (SDEs) and representative image samples from our globally and locally coupled Kuramoto orientation diffusion models.

Image

[60] Yang Song, Jascha Sohl-Dickstein, Diederik P Kingma, Abhishek Kumar, Stefano Ermon, and Ben Poole. Score-based generative modeling through stochastic differential equations. ICLR, 2021.
[68] Jia Xue, Hang Zhang, and Kristin Dana. Deep texture manifold for ground terrain recognition. In CVPR, 2018.

---
```bibtex
@inproceedings{songKuramotoOrientationDiffusion2025a,
title = {Kuramoto {{Orientation Diffusion Models}}},
booktitle = {{{NeurIPS}}},
author = {Song, Yue and Keller, T. Anderson and Brodjian, Sevan and Miyato, Takeru and Yue, Yisong and Perona, Pietro and Welling, Max},
year = 2025,
url={\url{https://openreview.net/forum?id=dxK2QgEKvz}}
}
```

コントリビューションガイド

コントリビューションガイドを開く

調査の方向性

Start by reviewing the linked official implementation at github.com/KingJamesSong/OrientationDiffusion and the cited Kuramoto Orientation Diffusion paper. Compare its pure PyTorch components with the diffusers repository to determine the required integration surface and validation approach. Done is not defined by the issue; the implementation scope and expected tests need maintainer clarification.

索引モデルが issue の本文から書いたものです。

評価

技術スタック
python, pytorch
領域
machine-learning
issue の種類
機能追加
難易度
5/5
見積もり時間
1週間以上
活発さ
停滞
明瞭さ
説明が足りない
初心者へのやさしさ
25/100

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。