TDiR: Transformer based Diffusion for Image Restoration Tasks
On the map: Diffusion models → Inverse problems and image restoration
Abstract (arXiv)
Images captured in challenging environments often experience various types of degradation, such as noise, color cast, blur, and light scattering. These issues significantly lower image quality, thereby reducing their usefulness in downstream tasks such as object detection, mapping, and classification. Our transformer-based diffusion model was developed to address image restoration challenges and enhance the quality of degraded images. Our methodology is assessed across three primary image restoration tasks, including underwater enhancement, denoising, and deraining, utilizing five standard benchmarks. It is then compared to 18 state-of-the-art techniques, employing four evaluation metrics. Our results show that the diffusion model, combined with transformers, outperforms current methods. The findings highlight the effectiveness of diffusion models and transformers in improving degraded image quality, thereby broadening their application in downstream tasks that demand high-fidelity visual data
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Inverse problems and image restoration, one of 14 directions in Diffusion models. Builds on 27 of its 70 references that are in the map. First posted on arXiv 25 Jun 2025.
Builds on
27 of 70 references.
References · 27
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,761
- 2016Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoisingcited by 1,358
- 2020Pre-Trained Image Processing Transformercited by 893
- 2021Uformer: A General U-Shaped Transformer for Image Restorationcited by 866
- 2021Multi-Stage Progressive Image Restorationcited by 816
- 2017Learning Deep CNN Denoiser Prior for Image Restorationcited by 408
- 2020Learning Texture Transformer Network for Image Super-Resolutioncited by 347
- 2019An Underwater Image Enhancement Benchmark Dataset and Beyondcited by 253
- 2021TransWeather: Transformer-based Restoration of Images Degraded by Adverse Weather Conditionscited by 244
- 2020CycleISP: Real Image Restoration via Improved Data Synthesiscited by 213
15 more
- 2019Fast Underwater Image Enhancement for Improved Visual Perceptioncited by 183
- 2018Residual Dense Network for Image Restorationcited by 181
- 2019Real Image Denoising with Feature Attentioncited by 176
- 2021Underwater Image Enhancement via Medium Transmission-Guided Multi-Color Space Embeddingcited by 134
- 2017WaterGAN: Unsupervised Generative Network to Enable Real-time Color Correction of Monocular Underwater Imagescited by 112
- 2021U-shape Transformer for Underwater Image Enhancementcited by 112
- 2020FD-GAN: Generative Adversarial Networks with Fusion-discriminator for Single Image Dehazingcited by 80
- 2019A General Decoupled Learning Framework for Parameterized Image Operatorscited by 72
- 2025Degradation-Aware Feature Perturbation for All-in-One Image Restorationcited by 41
- 2022DRT: A Lightweight Single Image Deraining Recursive Transformercited by 26
- 2024WildSAT: Learning Satellite Image Representations from Wildlife Observationscited by 13
- 2022Unsupervised Image Deraining: Optimization Model Driven Deep CNNcited by 12
- 2025Channel Consistency Prior and Self-Reconstruction Strategy Based Unsupervised Image Derainingcited by 11
- 2020Physical Model Guided Deep Image Derainingcited by 7
- 2023MetaUE: Model-based Meta-learning for Underwater Image Enhancementcited by 2
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Exploiting Diffusion Prior for Task-driven Image Restoration
- 2024Taming Diffusion Models for Image Restoration: A Review
- 2023Diffusion Models for Image Restoration and Enhancement: A Comprehensive Survey
- 2026Stop Denoising Your Blurs
- 2025Single-Step Latent Diffusion for Underwater Image Restoration
- 2023Residual Denoising Diffusion Models
- 2024Diffusion Posterior Proximal Sampling for Image Restoration
- 2024Efficient Diffusion Model for Image Restoration by Residual Shifting