摘要

近年来，单图像超分辨率技术的发展主要得益于令牌混合器（token mixer）与Transformer架构的推动。WaveMixSR采用了WaveMix架构，利用二维离散小波变换实现空间域的令牌混合，在超分辨率任务中取得了卓越性能，同时展现出出色的资源效率。在本研究中，我们对WaveMixSR架构进行了优化，主要包含两点改进：（1）用像素重排（pixel shuffle）操作替代传统的转置卷积层；（2）引入多阶段设计，以应对更高分辨率的重建任务（如4×超分辨率）。实验结果表明，所提出的增强模型WaveMixSR-V2在多个超分辨率任务中均优于现有架构，在BSD100数据集上达到了当前最优（state-of-the-art）性能，同时具备更低的资源消耗、更高的参数效率、更短的延迟以及更高的吞吐量。相关代码已开源，地址为：https://github.com/pranavphoenix/WaveMixSR。