atlas

Research map / Architectures and efficiency / Attention and transformers

Adaptive Window Pruning for Efficient Local Motion Deblurring

ICLR 2024 · Architectures and efficiency › Attention and transformers · approach: Token merging and masked pretraining · 1 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes LMD-ViT, a U-shaped Transformer for local motion deblurring that adaptively prunes non-blurry windows via a learned blurriness confidence predictor (Gumbel-Softmax training, mask-supervised pruning loss), cutting FLOPs by 66% and roughly doubling speed over Transformer deblurring baselines.

Problem

Local motion blur (moving objects against static backgrounds) is handled poorly by global deblurring methods, which distort sharp regions and waste computation on sharp pixels, especially for high-resolution images.

Approach

An adaptive window pruning Transformer block (AdaWPT) predicts per-window blurriness confidence, samples keep/drop decisions (Gumbel-Softmax in training, thresholding at test), and runs window-based self-attention and feed-forward only on kept windows, with the confidence predictor supervised by newly annotated blur masks.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
apply adaptive window pruningLMD-ViT · ReLoBlur testFLOPs: 4.376T -> 1.485THelped
apply adaptive window pruningLMD-ViT · ReLoBlur testinference time: 1.30s -> 0.56sHelped
remove window pruning (ablation)LMD-ViT · ReLoBlur testweighted PSNR: 30.25 -> 30.22Hurt
use our annotated blur masks instead of LBFMG masks (ablation)LMD-ViT · ReLoBlur testweighted PSNR: 30.14 -> 30.25Helped
use our annotated blur masks instead of LBFMG masks (ablation)LBAG · ReLoBlur testweighted PSNR: 28.31 -> 29.30Helped
replace LeFF with W-LeFF (ablation)LMD-ViT · ReLoBlur testinference time: 0.86s -> 0.56sHelped
replace LeFF with W-LeFF (ablation)LMD-ViT · ReLoBlur testPSNR: 35.45 -> 35.42No clear effect
increase feature channels from 16-32-64-128-256 to 32-64-128-256-512 (ablation)LMD-ViT · ReLoBlur testPSNR: 34.98 -> 35.42Helped
set pruning threshold β to 0.7 instead of 0.5 (ablation)LMD-ViT · ReLoBlur testweighted PSNR: 30.25 -> 30.14Hurt
set pruning threshold β to 0.2 instead of 0.5 (ablation)LMD-ViT · ReLoBlur testweighted PSNR: 30.25 -> 30.21Hurt
use AdaWPT blocks 2–8 instead of 1–9 (ablation)LMD-ViT · ReLoBlur testweighted PSNR: 30.25 -> 30.34Helped
use AdaWPT blocks 3–7 instead of 1–9 (ablation)LMD-ViT · ReLoBlur testweighted PSNR: 30.25 -> 30.38Helped

Surprising findings

Limitations

Experimental setup

Datasets: ReLoBlur GoPro

Benchmarks: ReLoBlur GoPro

Metrics: PSNR SSIM weighted PSNR weighted SSIM precision inference time FLOPs parameters

Baselines: DeepDeblur DeblurGAN-v2 HINet MIMO-UNet LBAG Restormer Uformer-B

Builds on: Restormer Uformer LBAG Gumbel-Softmax Swin Transformer

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy