arrow
Return

Resistive memory-based neural differential equation solver for score-based diffusion model

delete2026-05-12
delete0
delete
OA
AI
J
Jichang Yang
H
Hegan Chen
J
Jia Chen *
S
Songqi Wang
Q
Qifan Zhu
X
Xinyuan Zhang
Y
Yan Zeng
S
Shaocong Wang
M
Mingrui Yang
Y
Yifei Yu
X
Xi Chen
B
Bo Wang
B
Binbin Cui
Y
Yi Li
林宁 cover
林宁 (Ning Lin)
M
Meng Xu
Y
Yi Li
X
Xiaoxin Xu
X
Xiaojuan Qi
X
Xumeng Zhang *
D
Dashan Shang *
Z
Zhongrui Wang *
H
Han Wang
刘祺 cover
刘祺 (Qi Liu)
K
Kwang‐Ting Cheng
刘鸣 cover
刘鸣 (Ming Liu)
DOI:10.1038/s41467-026-72900-zdelete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
While AI-generated content (AIGC) strives to replicate human imagination, current models like score-based diffusion remain slow and energy-intensive. This inefficiency stems from conventional digital computers, where physically separated storage and processing units cause data-transfer bottlenecks, and discrete operations disrupt naturally continuous generation dynamics. Here we show a brain-inspired, analog in-memory computing system that overcomes these limitations. By employing resistive memory, our system integrates storage and computation to act as a time-continuous neural differential equation solver. We experimentally validate our solution with 180 nm resistive memory in-memory computing macros. While maintaining generative quality equivalent to the software baseline, our system accelerated both unconditional and conditional generation tasks, by factors of 69.0 and 116.5, respectively, compared to advanced digital hardware. Furthermore, it reduced energy consumption by 31.5% and 52.0%, respectively. Our approach expands the horizon for hardware solutions in edge computing for generative AI applications. Replicating the human brain’s imaginative ability has long been a core AI goal, yet current systems remain limited in speed and efficiency. Yang et al. present a hardware platform based on resistive memory, enabling high-speed, low-power AI content generation while fully preserving output quality.
Keywords:
Resistive memory
In-memory computing
Neural differential equation
Score-based diffusion model
Generative AI
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

Nature Communications cover
Nature Communications
IF:
15.7
Papers:
9.2W
Citations:
91.2W

Organization

F
fudan university
Scholars:
11.4W
Papers: 7.6W
Citations: 121
T
the hong kong university of science and technology
Scholars:
1.6K
Papers: 764
Citations: 0
T
The University of Hong Kong
Scholars:
5.9K
Papers: 2.9K
Citations: 7
P
peking university
Scholars:
11.6W
Papers: 8.6W
Citations: 146
S
southern university of science and technology
Scholars:
3.9K
Papers: 1.4K
Citations: 0
H
huazhong university of science and technology
Scholars:
2.5W
Papers: 7.4K
Citations: 5
C
chinese academy of sciences
Scholars:
55.3W
Papers: 44.6W
Citations: 704
researcher View more organizations