TDiR: Transformer based Diffusion for Image Restoration Tasks

arXiv:2506.20302v2 Announce Type: replace Abstract: 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

Related

• M2Retinexformer: Multi-Modal Retinexformer for Low-Light Image Enhancement
• Fast and Accurate Image Restoration and Generation with Rank Enhanced Linear Attention
• Joint Medical Image Enhancement and Segmentation with Diffusion-based Symbiotic Information Interaction
• VSANet: View-aware Sparse Attention Network for Light Field Image Denoising

Source: arXiv cs.CV | 2026-07-27