arrow
Return

Improved Residual Learning in Diffusion Models

delete2026-05-04
delete0
PRE
AI
J
Junyu Zhang
D
Daochang Liu
E
Eunbyung Park
S
Shichao Zhang
C
Chang Xu
DOI:10.1109/tpami.2026.3689939delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Diffusion models (DMs) have profoundly transformed the field of generative modeling, delivering exceptional image generation quality. Nevertheless, residuals in the generated images, arising from modeling errors that accumulate along the practical sampling trajectories of pre-trained DMs, remain an inevitable challenge. To mitigate this, we propose a novel residual learning framework built upon a parameterized correction function, designed to improve modeling performance through residual correction. Most notably, our framework exhibits remarkable transferable residual correction capabilities, enabling a correction function optimized for a specific pre-trained DM on a given dataset to enhance the performance of other DMs trained on the same dataset. To further improve our framework, we introduce an advanced training approach that designs an ODE sampler bank to refine residual simulation and incorporates a score consistency maintenance technique to enhance model convergence. Building on this approach, the optimized correction function achieves substantial improvements in residual correction. Extensive experiments performed on four widely used datasets and multiple pre-trained DMs validate the effectiveness and superiority of our residual learning framework.
Keywords:
Diffusion models
residual learning
residual correction
ODE sampler bank
transferable residual correction

Journal

IEEE Transactions on Pattern Analysis and Machine Intelligence cover
IEEE Transactions on Pattern Analysis and Machine Intelligence
IF:
18.6
Papers:
1.1K
Citations:
9.8W

Organization

U
university of sydney
Scholars:
6.4K
Papers: 2.9K
Citations: 0
S
sungkyunkwan university
Scholars:
4.4K
Papers: 1.6K
Citations: 0
Y
Yonsei University
Scholars:
4.8W
Papers: 4.6W
Citations: 5.2W
U
university of western australia
Scholars:
2.5K
Papers: 1.2K
Citations: 0
G
guangxi normal university
Scholars:
1.9K
Papers: 635
Citations: 0
researcher View more organizations
Cited Papers

Cited Papers

Emerging Properties in Self-Supervised Vision Transformers
err2021-10-01
err0
errOAAI
errMathilde Caron; Hugo Touvron; Ishan Misra; Herve Jegou; Julien Mairal; Piotr Bojanowski; Armand Joulin
errShare
errSave
High-Resolution Image Synthesis with Latent Diffusion Models
err2022-06-01
err0
errOAAI
errRobin Rombach; Andreas Blattmann; Dominik Lorenz; Patrick Esser; Bjorn Ommer
errShare
errSave
Perception Prioritized Training of Diffusion Models
err2022-06-01
err0
errOAAI
errJooyoung Choi; Jungbeom Lee; Chaehun Shin; Sungwon Kim; Hyunwoo Kim; Sungroh Yoon
errShare
errSave
researcher View more