Efficient JPEG Restoration in the Wavelet Domain via Mean Flows

📅 2026-08-28
📈 Citations: 0
Influential: 0
📄 PDF
🤖 AI Summary
该研究提出了一种65M参数的生成式恢复器,通过使用Haar变换和改进的MeanFlow目标函数,在保持高效处理速度的同时实现高质量JPEG图像恢复。
📝 Abstract
Latest JPEG restoration systems achieve strong quality with large models, yet often remain too slow and expensive for efficient on-device deployment. We present a 65M-parameter generative restorer that attains the lowest LPIPS at QF 10 and 20 on LIVE-1, Urban100, and DIV2K-val while sustaining 8.05 images/s at $1024\times1024$ on a single RTX 3090, roughly $4.9\times$ the reported throughput of one-step SODiff at one-twentieth of its parameters. Trained from scratch, the model replaces the learned VAE encoder-decoder with an exactly invertible two-level Haar transform, predicts a clean wavelet-domain residual through a rank-enhanced linear-attention DiT that estimates compression severity internally, and is optimized with an improved MeanFlow objective that enables inference in one or two network evaluations without distillation. Large pretrained priors remain stronger under severe compression (QF 5), whereas our model prioritizes throughput for deployment-constrained restoration.
Problem

Research questions and friction points this paper is trying to address.

JPEG restoration
on-device deployment
efficiency
Innovation

Methods, ideas, or system contributions that make the work stand out.

Wavelet Domain
MeanFlows
Invertible Haar Transform
Rank-Enhanced Linear-Attention DiT
🔎 Similar Papers
2024-09-16Philosophical transactions. Series A, Mathematical, physical, and engineering sciencesCitations: 8
S
Stefan-Alexandru Asandei
Independent Researcher
M
Mihai-Alexandru Radu
Faculty of Physics, "Alexandru Ioan Cuza" University of Iași