通过高频先验自适应掩码,加速图像超分且不降低性能。
High-Frequency Prior-Driven Adaptive Masking for Accelerating Image Super-Resolution
- 基于高斯模糊差分提取高频区域,用K均值生成动态掩码。
- 在多个模型上减少24%~43%计算量,性能基本不变或提升。
- 无需训练,支持不同退化场景,适合部署在资源受限设备。
图像超分辨率加速的核心挑战在于减少计算量的同时保持性能与适应性。受高频区域(如边缘和纹理)对重建最关键这一观察启发,我们提出一种无需训练的自适应掩码模块,可动态聚焦于这些关键区域。首先通过高斯模糊减法提取高频成分,再利用K均值聚类自动生成二值掩码,识别需密集处理的区域。该方法可轻松集成至CNN与Transformer架构中:对基于CNN的模型,以展开操作结合1×1卷积替代标准3×3卷积,实现掩码引导的像素级稀疏计算;对Transformer模型,将掩码划分为非重叠窗口,依据平均值选择性处理令牌。推理时,无需计算的像素或窗口被剪枝,显著降低计算开销。此外,支持通过膨胀调整掩码范围,无需重新训练,且对未见过的退化(如噪声、压缩)具有鲁棒性。在多个基准测试上,该方法使先进模型(如CARN、SwinIR)的FLOPs降低24%~43%,同时保持或优于原有定量指标。代码已开源。
原文摘要 · Abstract (English)
The primary challenge in accelerating image super-resolution lies in reducing computation while maintaining performance and adaptability. Motivated by the observation that high-frequency regions (e.g., edges and textures) are most critical for reconstruction, we propose a training-free adaptive masking module for acceleration that dynamically focuses computation on these challenging areas. Specifically, our method first extracts high-frequency components via Gaussian blur subtraction and adaptively generates binary masks using K-means clustering to identify regions requiring intensive processing. Our method can be easily integrated with both CNNs and Transformers. For CNN-based architectures, we replace standard $3 \times 3$ convolutions with an unfold operation followed by $1 \times 1$ convolutions, enabling pixel-wise sparse computation guided by the mask. For Transformer-based models, we partition the mask into non-overlapping windows and selectively process tokens based on their average values. During inference, unnecessary pixels or windows are pruned, significantly reducing computation. Moreover, our method supports dilation-based mask adjustment to control the processing scope without retraining, and is robust to unseen degradations (e.g., noise, compression). Extensive experiments on benchmarks demonstrate that our method reduces FLOPs by 24--43% for state-of-the-art models (e.g., CARN, SwinIR) while achieving comparable or better quantitative metrics. The source code is available at https://github.com/shangwei5/AMSR
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。