Scan Clusters, Not Pixels: A Cluster-Centric Paradigm for Efficient Ultra-high-definition Image Restoration
Organizations: National University of Defense Technology · HKUST(GZ) · Qilu University of Technology · Technical University of Munich ·
Abstract
Ultra-High-Definition (UHD) image restoration is trapped in a scalability crisis: existing models, bound to pixel-wise operations, demand unsustainable computation. While state space models (SSMs) like Mamba promise linear complexity, their pixel-serial scanning remains a fundamental bottleneck for the millions of pixels in UHD content. We ask: must we process every pixel to understand the image? This paper introduces CSSM, a visual state space model that breaks this taboo by shifting from pixel-serial to cluster-serial scanning. Our core discovery is that the rich feature distribution of a UHD image can be distilled into a sparse set of semantic centroids via a neural-parameterized mixture model. CSSM leverages this to reformulate global modeling into a novel dual-path process: it scans and reasons over a handful of cluster centers, then diffuses the global context back to all pixels through a principled similarity distribution, all while a lightweight modulator preserves fine details. This cluster-centric paradigm achieves a decisive leap in efficiency, slashing computational costs while establishing new state-of-the-art results across five UHD restoration tasks. More than a solution, CSSM charts a new course for efficient large-scale vision: scan clusters, not pixels.
Figures & tables
| Methods | Type | Venue | PSNR | SSIM | Param |
|---|---|---|---|---|---|
| Z_DCE++ [ 17 ] | non-UHD | TPAMI’21 | 15.58 | 0.934 | 79.42K |
| Uformer [ 33 ] | CVPR’22 | 29.98 | 0.980 | 20.63M | |
| Restormer [ 42 ] | CVPR’22 | 36.90 | 0.988 | 26.11M | |
| NSEN [ 40 ] | UHD | MM’23 | 29.49 | 0.980 | 2.67M |
| UHDFour [ 16 ] | ICLR’23 | 36.12 | 0.990 | 17.54M | |
| LLFormer [ 32 ] | AAAI’23 | 37.33 | 0.988 | 24.52M |
| Methods | Type | Venue | PSNR | SSIM | Param |
|---|---|---|---|---|---|
| Z_DCE++ [ 17 ] | non-UHD | TPAMI’21 | 16.41 | 0.630 | 10.56K |
| Uformer [ 33 ] | CVPR’22 | 19.28 | 0.849 | 20.63M | |
| Restormer [ 42 ] | CVPR’22 | 22.25 | 0.871 | 26.11M | |
| UHDFour [ 16 ] | UHD | ICLR’23 | 26.22 | 0.900 | 17.54M |
| LLFormer [ 32 ] | AAAI’23 | 22.79 | 0.853 | 13.15M | |
| UHDformer [ 29 ] | AAAI’24 | 27.11 | 0.927 | 0.34M |
| Methods | Type | Venue | PSNR | SSIM | Param |
|---|---|---|---|---|---|
| RCDNet [ 30 ] | non-UHD | CVPR’20 | 30.83 | 0.921 | 3.17M |
| SPDNet [ 39 ] | ICCV’21 | 31.81 | 0.922 | 3.04M | |
| IDT [ 38 ] | TPAMI’22 | 32.91 | 0.948 | 16.41M | |
| Restormer [ 42 ] | CVPR’22 | 33.02 | 0.934 | 26.12M | |
| DRSformer [ 3 ] | CVPR’23 | 32.96 | 0.933 | 33.65M | |
| UDR-S2Former [ 2 ] | ICCV’23 | 33.36 | 0.946 | 8.53M |
| Methods | RCDNet [ 30 ] | SPDNet [ 39 ] | IDT [ 38 ] | Restormer [ 42 ] | DRSformer [ 3 ] | NeRD-Rain [ 4 ] | MambaIRv2 [ 10 ] | ERR [ 43 ] | Ours |
|---|---|---|---|---|---|---|---|---|---|
| NIQE | 9.997 | 9.917 | 9.067 | 8.636 | 8.49 | 9.139 | 9.397 | 9.136 | 8.198 |
| PIQE | 63.816 | 64.774 | 55.049 | 60.335 | 60.441 | 60.070 | 62.913 | 59.735 | 54.90 |
| Methods | Type | Venue | PSNR | SSIM | Param |
|---|---|---|---|---|---|
| MIMO-Unet++ [ 5 ] | non-UHD | ICCV’21 | 25.03 | 0.752 | 16.1M |
| Restormer [ 42 ] | CVPR’22 | 25.21 | 0.752 | 26.1M | |
| Uformer [ 33 ] | CVPR’22 | 25.27 | 0.752 | 20.6M | |
| Stripformer [ 26 ] | ECCV’22 | 25.05 | 0.750 | 19.7M | |
| FFTformer [ 13 ] | CVPR’23 | 25.41 | 0.757 | 16.6M | |
| UHDformer [ 29 ] | UHD | AAAI’24 | 28.82 | 0.844 | 0.34M |
| Methods | Type | Venue | PSNR | SSIM | Param |
|---|---|---|---|---|---|
| Restormer [ 42 ] | non-UHD | CVPR’22 | 12.72 | 0.693 | 26.11M |
| Uformer [ 33 ] | CVPR’22 | 19.83 | 0.737 | 20.63M | |
| DehazeFormer [ 25 ] | TIP’23 | 15.37 | 0.725 | 2.5M | |
| MB-TaylorFormer [ 24 ] | ICCV’23 | 20.99 | 0.919 | 2.7M | |
| UHD [ 45 ] | UHD | ICCV’21 | 18.04 | 0.811 | 34.5M |
| UHDformer [ 29 ] | AAAI’24 | 22.59 | 0.942 | 0.34M |
| Methods | Type | Venue | PSNR | SSIM | Param |
|---|---|---|---|---|---|
| Uformer [ 33 ] | non-UHD | CVPR’22 | 23.71 | 0.871 | 20.63M |
| Restormer [ 42 ] | CVPR’22 | 24.14 | 0.869 | 26.12M | |
| SFNet [ 8 ] | ICLR’23 | 23.63 | 0.845 | 7.05M | |
| UHD [ 45 ] | UHD | ICCV’21 | 29.29 | 0.949 | 34.5M |
| UHDformer [ 29 ] | AAAI’24 | 36.61 | 0.988 | 0.34M | |
| UHDDIP [ 31 ] | TCSVT’25 | 41.56 | 0.990 | 0.81M |
| Variants | PSNR | SSIM | Param |
|---|---|---|---|
| (a) replace CCSM with ResBlock | 37.32 | 0.989 | 2.38M |
| (b) replace CCSM with Vanilla Mamba [ 11 ] | 37.43 | 0.990 | 3.11M |
| (c) replace CCSM with ASSM [ 10 ] | 38.71 | 0.991 | 2.96M |
| (d) replace SCFM with ResBlock | 39.25 | 0.992 | 2.69M |
| (e) remove CCSM | 35.87 | 0.987 | 2.21M |
| (f) remove SCFM | 39.05 | 0.992 | 2.53M |
| Number of Centers | 2 | 4 | 6 | 8 |
|---|---|---|---|---|
| UHD-LOL4K [ 32 ] | 37.91/0.992 | 39.61/0.992 | 39.23/0.992 | 39.19/0.992 |
| UHD-Blur [ 29 ] | 30.42/0.881 | 31.53/0.890 | 31.72/0.890 | 31.65/0.889 |
| UHD-Haze [ 29 ] | 23.27/0.940 | 24.08/0.942 | 24.03/0.942 | 23.86/0.942 |
| Method | MambaIR [ 11 ] | Wave-Mamba [ 48 ] | EVSSM [ 14 ] | MambaIRv2 [ 10 ] | Ours |
|---|---|---|---|---|---|
| Parm | 1.32M | 1.25M | 17.13M | 0.84M | 2.71M |
| FLOPs | 4.774G | 0.881G | 7.893G | 4.940G | 0.407G |
| Venue | ECCV’24 | MM’24 | CVPR’25 | CVPR’25 | - |
Image Restoration
Image restoration uses computational methods to recover useful visual content from noisy, blurred, low-resolution, or otherwise degraded images.
See all 372 papers