mlsanityData to 14 Oct 2026

TDiR: Transformer based Diffusion for Image Restoration Tasks

arXiv 2506.20302 ↗First posted 25 Jun 2025cs.CVcited by 0

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

  1. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  2. 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
  3. 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,761
  4. 2016Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoisingcited by 1,358
  5. 2020Pre-Trained Image Processing Transformercited by 893
  6. 2021Uformer: A General U-Shaped Transformer for Image Restorationcited by 866
  7. 2021Multi-Stage Progressive Image Restorationcited by 816
  8. 2017Learning Deep CNN Denoiser Prior for Image Restorationcited by 408
  9. 2020Learning Texture Transformer Network for Image Super-Resolutioncited by 347
  10. 2019An Underwater Image Enhancement Benchmark Dataset and Beyondcited by 253
  11. 2021TransWeather: Transformer-based Restoration of Images Degraded by Adverse Weather Conditionscited by 244
  12. 2020CycleISP: Real Image Restoration via Improved Data Synthesiscited by 213
15 more
  1. 2019Fast Underwater Image Enhancement for Improved Visual Perceptioncited by 183
  2. 2018Residual Dense Network for Image Restorationcited by 181
  3. 2019Real Image Denoising with Feature Attentioncited by 176
  4. 2021Underwater Image Enhancement via Medium Transmission-Guided Multi-Color Space Embeddingcited by 134
  5. 2017WaterGAN: Unsupervised Generative Network to Enable Real-time Color Correction of Monocular Underwater Imagescited by 112
  6. 2021U-shape Transformer for Underwater Image Enhancementcited by 112
  7. 2020FD-GAN: Generative Adversarial Networks with Fusion-discriminator for Single Image Dehazingcited by 80
  8. 2019A General Decoupled Learning Framework for Parameterized Image Operatorscited by 72
  9. 2025Degradation-Aware Feature Perturbation for All-in-One Image Restorationcited by 41
  10. 2022DRT: A Lightweight Single Image Deraining Recursive Transformercited by 26
  11. 2024WildSAT: Learning Satellite Image Representations from Wildlife Observationscited by 13
  12. 2022Unsupervised Image Deraining: Optimization Model Driven Deep CNNcited by 12
  13. 2025Channel Consistency Prior and Self-Reconstruction Strategy Based Unsupervised Image Derainingcited by 11
  14. 2020Physical Model Guided Deep Image Derainingcited by 7
  15. 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

  1. 2025Exploiting Diffusion Prior for Task-driven Image Restoration
  2. 2024Taming Diffusion Models for Image Restoration: A Review
  3. 2023Diffusion Models for Image Restoration and Enhancement: A Comprehensive Survey
  4. 2026Stop Denoising Your Blurs
  5. 2025Single-Step Latent Diffusion for Underwater Image Restoration
  6. 2023Residual Denoising Diffusion Models
  7. 2024Diffusion Posterior Proximal Sampling for Image Restoration
  8. 2024Efficient Diffusion Model for Image Restoration by Residual Shifting