arXiv:2410.12628cs.CV2024-10被引 101

用合成数据和感知模块提升文档布局分析的精度与速度。

DocLayout-YOLO: Enhancing Document Layout Analysis through Diverse Synthetic Data and Global-to-Local Adaptive Perception

  • 通过二维装箱算法生成30万张合成文档,增强预训练效果。
  • 在多个数据集上实现更快且更准的布局分析,优于现有方法。
  • 适合需要高效高精度文档理解的工业级应用。

文档布局分析对真实场景中的文档理解系统至关重要,但面临速度与精度的权衡:多模态方法利用文本与视觉特征可获得更高精度,但延迟显著;单模态方法仅依赖视觉特征虽速度快,却牺牲了准确性。为此,我们提出DocLayout-YOLO,通过文档特化的预训练与模型设计,在保持速度优势的同时提升精度。为实现鲁棒的文档预训练,我们引入Mesh-candidate BestFit算法,将文档合成建模为二维装箱问题,生成大规模、多样化的DocSynth-300K数据集。在该数据集上预训练显著提升了跨多种文档类型的微调性能。在模型优化方面,提出全局到局部可控感受野模块,更好处理文档元素的多尺度变化。此外,为验证不同文档类型下的性能,我们构建了复杂且具有挑战性的基准DocStructBench。大量实验表明,DocLayout-YOLO在下游数据集上兼具速度与精度优势。代码、数据与模型已公开于https://github.com/opendatalab/DocLayout-YOLO。

原文摘要 · Abstract (English)

Document Layout Analysis is crucial for real-world document understanding systems, but it encounters a challenging trade-off between speed and accuracy: multimodal methods leveraging both text and visual features achieve higher accuracy but suffer from significant latency, whereas unimodal methods relying solely on visual features offer faster processing speeds at the expense of accuracy. To address this dilemma, we introduce DocLayout-YOLO, a novel approach that enhances accuracy while maintaining speed advantages through document-specific optimizations in both pre-training and model design. For robust document pre-training, we introduce the Mesh-candidate BestFit algorithm, which frames document synthesis as a two-dimensional bin packing problem, generating the large-scale, diverse DocSynth-300K dataset. Pre-training on the resulting DocSynth-300K dataset significantly improves fine-tuning performance across various document types. In terms of model optimization, we propose a Global-to-Local Controllable Receptive Module that is capable of better handling multi-scale variations of document elements. Furthermore, to validate performance across different document types, we introduce a complex and challenging benchmark named DocStructBench. Extensive experiments on downstream datasets demonstrate that DocLayout-YOLO excels in both speed and accuracy. Code, data, and models are available at https://github.com/opendatalab/DocLayout-YOLO.

文档分析合成数据布局检测YOLO

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