arXiv:2508.04055cs.CV2025-08被引 3

一个统一模型搞定文档修复,效果媲美专用模型。

Uni-DocDiff: A Unified Document Restoration Model Based on Diffusion

  • 用可学习的任务提示实现多任务统一建模。
  • 在多个数据集上性能接近甚至超过专用模型。
  • 适合需要灵活扩展新修复任务的场景。

去除文档的各种退化因素对数字化、下游分析和可读性均有重要价值。以往方法通常为每项修复任务单独设计模型,导致处理系统复杂繁琐。尽管近期研究尝试统一多任务,但常受限于人工设计的提示词和繁重预处理,难以充分挖掘任务间的协同效应。为此,我们提出基于扩散模型的统一文档修复框架Uni-DocDiff。该模型采用可学习的任务提示设计,实现优异的可扩展性。为增强多任务能力并缓解任务干扰,我们提出一种新颖的先验池(Prior Pool),融合局部高频特征与全局低频特征。同时设计先验融合模块(PFM),使模型能自适应选择最相关的先验信息。大量实验表明,Uni-DocDiff在性能上可媲美甚至超越专用专家模型,且具备无缝适配新任务的强可扩展性。

原文摘要 · Abstract (English)

Removing various degradations from damaged documents greatly benefits digitization, downstream document analysis, and readability. Previous methods often treat each restoration task independently with dedicated models, leading to a cumbersome and highly complex document processing system. Although recent studies attempt to unify multiple tasks, they often suffer from limited scalability due to handcrafted prompts and heavy preprocessing, and fail to fully exploit inter-task synergy within a shared architecture. To address the aforementioned challenges, we propose Uni-DocDiff, a Unified and highly scalable Document restoration model based on Diffusion. Uni-DocDiff develops a learnable task prompt design, ensuring exceptional scalability across diverse tasks. To further enhance its multi-task capabilities and address potential task interference, we devise a novel \textbf{Prior \textbf{P}ool}, a simple yet comprehensive mechanism that combines both local high-frequency features and global low-frequency features. Additionally, we design the \textbf{Prior \textbf{F}usion \textbf{M}odule (PFM)}, which enables the model to adaptively select the most relevant prior information for each specific task. Extensive experiments show that the versatile Uni-DocDiff achieves performance comparable or even superior performance compared with task-specific expert models, and simultaneously holds the task scalability for seamless adaptation to new tasks.

文档修复扩散模型多任务学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。