EWT: Efficient Wavelet-Transformer for single image denoising

Transformer-based image denoising methods have shown remarkable potential but suffer from high computational cost and large memory footprint due to their linear operations for capturing long-range dependencies. In this work, we aim to develop a more resource-efficient Transformer-based image denoisi...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Veröffentlicht in:Neural networks 2024-09, Vol.177, p.106378, Article 106378
Hauptverfasser: Li, Juncheng, Cheng, Bodong, Chen, Ying, Gao, Guangwei, Shi, Jun, Zeng, Tieyong
Format: Artikel
Sprache:eng
Schlagworte:
Online-Zugang:Volltext
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
Beschreibung
Zusammenfassung:Transformer-based image denoising methods have shown remarkable potential but suffer from high computational cost and large memory footprint due to their linear operations for capturing long-range dependencies. In this work, we aim to develop a more resource-efficient Transformer-based image denoising method that maintains high performance. To this end, we propose an Efficient Wavelet Transformer (EWT), which incorporates a Frequency-domain Conversion Pipeline (FCP) to reduce image resolution without losing critical features, and a Multi-level Feature Aggregation Module (MFAM) with a Dual-stream Feature Extraction Block (DFEB) to harness hierarchical features effectively. EWT achieves a faster processing speed by over 80% and reduces GPU memory usage by more than 60% compared to the original Transformer, while still delivering denoising performance on par with state-of-the-art methods. Extensive experiments show that EWT significantly improves the efficiency of Transformer-based image denoising, providing a more balanced approach between performance and resource consumption. •We propose a novel Efficient Wavelet-Transformer (EWT) for SID.•EWT increases the speed of Transformer by more than 80% and reduces GPU by more than 60%.•We propose an efficient Multi-level Feature Aggregation Module (MFAM).•We demonstrate the effectiveness of wavelets in Transformer models.
ISSN:0893-6080
1879-2782
1879-2782
DOI:10.1016/j.neunet.2024.106378