huggingface / huggingface/diffusers

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

Ouverte
#12,903 1 commentaire 0 réactions 0 personnes assignées Voir sur GitHub
stale
Langage dominant
Python
Étoiles
34.5k
Forks
7.3k
Merge moyen
3 j 3 h
PR mergées (30 j)
91

Description

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}}
}
```

Guide de contribution

Ouvrir le guide de contribution

Piste de recherche

Commencez par examiner l’implémentation officielle liée à github.com/KingJamesSong/OrientationDiffusion ainsi que le Kuramoto Orientation Diffusion paper cité. Comparez ses composants PyTorch purs avec le dépôt diffusers afin de déterminer la surface d’intégration requise et l’approche de validation. La définition de « Done » n’est pas précisée dans l’issue ; le périmètre de l’implémentation et les tests attendus doivent être clarifiés par les mainteneurs.

Rédigé par le modèle d'indexation à partir du texte de l'issue.

Évaluation

Stack technique
python, pytorch
Domaine
machine-learning
Type d'issue
Fonctionnalité
Difficulté
5/5
Temps estimé
Plus d'une semaine
Activité
À l'abandon
Clarté
À clarifier
Accessibilité débutants
25/100

Recevez les nouvelles issues par e-mail

Un résumé court des issues GitHub adaptées aux débutants.