High-Frequency Prior-Driven Adaptive Masking for Accelerating Image Super-Resolution

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Shang, Wei, Ren, Dongwei, Zhang, Wanying, Zhu, Pengfei, Hu, Qinghua, Zuo, Wangmeng
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915282923028480
author Shang, Wei
Ren, Dongwei
Zhang, Wanying
Zhu, Pengfei
Hu, Qinghua
Zuo, Wangmeng
author_facet Shang, Wei
Ren, Dongwei
Zhang, Wanying
Zhu, Pengfei
Hu, Qinghua
Zuo, Wangmeng
contents 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
format Preprint
id arxiv_https___arxiv_org_abs_2505_06975
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle High-Frequency Prior-Driven Adaptive Masking for Accelerating Image Super-Resolution
Shang, Wei
Ren, Dongwei
Zhang, Wanying
Zhu, Pengfei
Hu, Qinghua
Zuo, Wangmeng
Computer Vision and Pattern Recognition
I.4.3
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
title High-Frequency Prior-Driven Adaptive Masking for Accelerating Image Super-Resolution
topic Computer Vision and Pattern Recognition
I.4.3
url https://arxiv.org/abs/2505.06975