Structure-guided Diffusion Transformer for Low-Light Image Enhancement

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Yin, Xiangchen, Yu, Zhenda, Jiang, Longtao, Gao, Xin, Sun, Xiao, Liu, Zhi, Yang, Xun
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866912820755431424
author Yin, Xiangchen
Yu, Zhenda
Jiang, Longtao
Gao, Xin
Sun, Xiao
Liu, Zhi
Yang, Xun
author_facet Yin, Xiangchen
Yu, Zhenda
Jiang, Longtao
Gao, Xin
Sun, Xiao
Liu, Zhi
Yang, Xun
contents While the diffusion transformer (DiT) has become a focal point of interest in recent years, its application in low-light image enhancement remains a blank area for exploration. Current methods recover the details from low-light images while inevitably amplifying the noise in images, resulting in poor visual quality. In this paper, we firstly introduce DiT into the low-light enhancement task and design a novel Structure-guided Diffusion Transformer based Low-light image enhancement (SDTL) framework. We compress the feature through wavelet transform to improve the inference efficiency of the model and capture the multi-directional frequency band. Then we propose a Structure Enhancement Module (SEM) that uses structural prior to enhance the texture and leverages an adaptive fusion strategy to achieve more accurate enhancement effect. In Addition, we propose a Structure-guided Attention Block (SAB) to pay more attention to texture-riched tokens and avoid interference from noisy areas in noise prediction. Extensive qualitative and quantitative experiments demonstrate that our method achieves SOTA performance on several popular datasets, validating the effectiveness of SDTL in improving image quality and the potential of DiT in low-light enhancement tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2504_15054
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Structure-guided Diffusion Transformer for Low-Light Image Enhancement
Yin, Xiangchen
Yu, Zhenda
Jiang, Longtao
Gao, Xin
Sun, Xiao
Liu, Zhi
Yang, Xun
Computer Vision and Pattern Recognition
While the diffusion transformer (DiT) has become a focal point of interest in recent years, its application in low-light image enhancement remains a blank area for exploration. Current methods recover the details from low-light images while inevitably amplifying the noise in images, resulting in poor visual quality. In this paper, we firstly introduce DiT into the low-light enhancement task and design a novel Structure-guided Diffusion Transformer based Low-light image enhancement (SDTL) framework. We compress the feature through wavelet transform to improve the inference efficiency of the model and capture the multi-directional frequency band. Then we propose a Structure Enhancement Module (SEM) that uses structural prior to enhance the texture and leverages an adaptive fusion strategy to achieve more accurate enhancement effect. In Addition, we propose a Structure-guided Attention Block (SAB) to pay more attention to texture-riched tokens and avoid interference from noisy areas in noise prediction. Extensive qualitative and quantitative experiments demonstrate that our method achieves SOTA performance on several popular datasets, validating the effectiveness of SDTL in improving image quality and the potential of DiT in low-light enhancement tasks.
title Structure-guided Diffusion Transformer for Low-Light Image Enhancement
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2504.15054