异构窗口Transformer用于图像去噪

Heterogeneous Window Transformer for Image Denoising

IEEE Transactions on Systems, Man, and Cybernetics: Systems · 2024
被引 55 · 同刊同年前 3%
ABS 3

中文导读

提出一种异构窗口Transformer(HWformer),通过设计异构全局窗口和稀疏前馈网络,在保持去噪效果的同时将去噪时间缩短至主流恢复Transformer的30%。

Abstract

Deep networks can usually depend on extracting more structural information to improve denoising results. However, they may ignore correlation between pixels from an image to pursue better-denoising performance. Window Transformer can use long- and short-distance modeling to interact pixels to address mentioned problem. To make a tradeoff between distance modeling and denoising time, we propose a heterogeneous window Transformer (HWformer) for image denoising. HWformer first designs heterogeneous global windows to capture global context information for improving denoising effects. To build a bridge between long and short-distance modeling, global windows are horizontally and vertically shifted to facilitate diversified information without increasing denoising time. To prevent the information loss phenomenon of independent patches, sparse idea is guided a feed-forward network to extract local information of neighboring patches. The proposed HWformer only takes 30% of popular restoration Transformer in terms of denoising time. Its codes can be obtained at <uri xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">https://github.com/hellloxiaotian/HWformer</uri>.

图像去噪计算机视觉深度学习Transformer