DiT4SR: Taming Diffusion Transformer for Real-World Image Super-Resolution

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Duan, Zheng-Peng, Zhang, Jiawei, Jin, Xin, Zhang, Ziheng, Xiong, Zheng, Zou, Dongqing, Ren, Jimmy S., Guo, Chun-Le, Li, Chongyi
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912467654803456
author Duan, Zheng-Peng
Zhang, Jiawei
Jin, Xin
Zhang, Ziheng
Xiong, Zheng
Zou, Dongqing
Ren, Jimmy S.
Guo, Chun-Le
Li, Chongyi
author_facet Duan, Zheng-Peng
Zhang, Jiawei
Jin, Xin
Zhang, Ziheng
Xiong, Zheng
Zou, Dongqing
Ren, Jimmy S.
Guo, Chun-Le
Li, Chongyi
contents Large-scale pre-trained diffusion models are becoming increasingly popular in solving the Real-World Image Super-Resolution (Real-ISR) problem because of their rich generative priors. The recent development of diffusion transformer (DiT) has witnessed overwhelming performance over the traditional UNet-based architecture in image generation, which also raises the question: Can we adopt the advanced DiT-based diffusion model for Real-ISR? To this end, we propose our DiT4SR, one of the pioneering works to tame the large-scale DiT model for Real-ISR. Instead of directly injecting embeddings extracted from low-resolution (LR) images like ControlNet, we integrate the LR embeddings into the original attention mechanism of DiT, allowing for the bidirectional flow of information between the LR latent and the generated latent. The sufficient interaction of these two streams allows the LR stream to evolve with the diffusion process, producing progressively refined guidance that better aligns with the generated latent at each diffusion step. Additionally, the LR guidance is injected into the generated latent via a cross-stream convolution layer, compensating for DiT's limited ability to capture local information. These simple but effective designs endow the DiT model with superior performance in Real-ISR, which is demonstrated by extensive experiments. Project Page: https://adam-duan.github.io/projects/dit4sr/.
format Preprint
id arxiv_https___arxiv_org_abs_2503_23580
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle DiT4SR: Taming Diffusion Transformer for Real-World Image Super-Resolution
Duan, Zheng-Peng
Zhang, Jiawei
Jin, Xin
Zhang, Ziheng
Xiong, Zheng
Zou, Dongqing
Ren, Jimmy S.
Guo, Chun-Le
Li, Chongyi
Computer Vision and Pattern Recognition
Large-scale pre-trained diffusion models are becoming increasingly popular in solving the Real-World Image Super-Resolution (Real-ISR) problem because of their rich generative priors. The recent development of diffusion transformer (DiT) has witnessed overwhelming performance over the traditional UNet-based architecture in image generation, which also raises the question: Can we adopt the advanced DiT-based diffusion model for Real-ISR? To this end, we propose our DiT4SR, one of the pioneering works to tame the large-scale DiT model for Real-ISR. Instead of directly injecting embeddings extracted from low-resolution (LR) images like ControlNet, we integrate the LR embeddings into the original attention mechanism of DiT, allowing for the bidirectional flow of information between the LR latent and the generated latent. The sufficient interaction of these two streams allows the LR stream to evolve with the diffusion process, producing progressively refined guidance that better aligns with the generated latent at each diffusion step. Additionally, the LR guidance is injected into the generated latent via a cross-stream convolution layer, compensating for DiT's limited ability to capture local information. These simple but effective designs endow the DiT model with superior performance in Real-ISR, which is demonstrated by extensive experiments. Project Page: https://adam-duan.github.io/projects/dit4sr/.
title DiT4SR: Taming Diffusion Transformer for Real-World Image Super-Resolution
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2503.23580