人工智能
小波
降噪
图像去噪
模式识别(心理学)
计算机科学
变压器
计算机视觉
工程类
电压
电气工程
作者
Juncheng Li,Bodong Cheng,Ying Chen,Guangwei Gao,Jun Shi,Tieyong Zeng
出处
期刊:Neural Networks
[Elsevier]
日期:2024-05-08
卷期号:177: 106378-106378
被引量:5
标识
DOI:10.1016/j.neunet.2024.106378
摘要
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.
科研通智能强力驱动
Strongly Powered by AbleSci AI