无需提示词的文档修复框架,自动识别退化类型并修复多种图像质量问题。
DocPure: Prompt-Free Unified Document Restoration via Degradation-Aware Structure-Guided Wavelet Modulation

- 通过退化感知结构编码器预测清晰结构先验,无需人工提示。
- 在多个任务上超越现有方法,如去模糊、去噪、压缩伪影去除等。
- 适合需要统一处理多种文档退化的实际场景,如档案数字化。
高质量文档图像对信息存档和下游自动化处理至关重要,但常因非受控采集与传输导致多种退化。现有统一文档修复方法往往需训练多套退化特定模型、依赖人工任务提示或跨任务数据配对。为此,我们提出DocPure,一种无提示词的统一框架,实现退化感知文档修复。设计退化感知结构自编码器,结合退化引导路由正则化,从退化输入中预测干净结构先验;推理时完全无提示,退化标签仅在训练中作为路由正则化辅助监督。进一步引入结构引导小波交互机制,桥接频域特征与空间语义;其中跨频自适应调制利用低频子带调制高频恢复,保障结构一致性。大量实验表明,DocPure在去模糊、去噪、压缩伪影减少和去阴影等多种任务上均优于当前最优方法。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。