Search papers, labs, and topics across Lattice.
This paper introduces DocPure, a novel prompt-free framework for unified document restoration that effectively addresses the challenges posed by diverse image degradations. By employing a degradation-aware structure auto-encoder with routing regularization, the model predicts clean structural priors from degraded inputs without the need for manual prompts during inference. Extensive experiments show that DocPure outperforms existing state-of-the-art methods in tasks such as deblurring, denoising, and compression artifact reduction, demonstrating its versatility and effectiveness in document restoration.
Achieving high-quality document restoration without manual prompts, DocPure outperforms traditional methods by leveraging degradation-aware structural insights.
High-quality document images are pivotal for information archiving and downstream automatic processing. However, they are frequently compromised by diverse degradations during uncontrolled acquisition and transmission. While unified document restoration techniques have been proposed to restore images from multiple degradations, they often struggle with training multiple degradation-specific models, reliance on manual task-specific prompts, or cross-task data pairing. To address these limitations, we propose DocPure, a prompt-free unified framework that achieves degradation-aware document restoration. We design a degradation-aware structure auto-encoder with degradation-informed routing regularization to predict clean structural priors from degraded inputs. The model is prompt-free at inference, and degradation labels are only used as auxiliary supervision for the routing regularization during training. Furthermore, we introduce a structure-guided wavelet interaction mechanism to bridge frequency-domain features and spatial semantics. Within the structure-guided wavelet interaction mechanism, a cross-frequency adaptive modulation utilizes low-frequency sub-bands to modulate high-frequency recovery, ensuring structural consistency. Extensive experiments demonstrate that DocPure achieves strong performance compared with state-of-the-art methods across various tasks, including deblurring, denoising, compression artifact reduction, and deshadowing.
