huggingface / huggingface/diffusers

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

Abierto
#12,903 1 comentario 0 reacciones 0 asignados Ver en GitHub
stale
Lenguaje dominante
Python
Estrellas
34.5k
Forks
7.3k
Merge medio
3 d 3 h
PR fusionados (30 d)
91

Descripción

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

Guía de contribución

Abrir la guía de contribución

Línea de trabajo

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.

Escrito por el modelo de indexación a partir del texto del issue.

Evaluación

Stack tecnológico
python, pytorch
Área
machine-learning
Tipo de issue
Nueva funcionalidad
Dificultad
5/5
Tiempo estimado
Más de una semana
Estado de actividad
Estancado
Claridad
Necesita aclaración
Aptitud para principiantes
25/100

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.