Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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

About

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.

Zhiyuan Zhao, Hengrui Kang, Bin Wang, Conghui He• 2024

Related benchmarks

TaskDatasetResultRank
Document Text GenerationOHR-Bench Generation
Text Score41.9
14
Textual RAGOHR-Bench (Overall)
TXT Score0.385
14
Document RetrievalOHR-Bench Retrieval
Accuracy (Text)63.5
14
Document Layout AnalysisDocLayNet
mAP79.7
13
Document Layout AnalysisD4LA
mAP70.3
7
Visual RAGAllganize
TXT Score69.5
5
Textual RAGBizMMRAG Japanese (test)
TXT Score61.7
5
Textual RAGAllganize Japanese (test)
TXT Score49.3
5
Visual RAGOHR-Bench (test)
TXT Score72.2
5
Visual RAGBizMMRAG
Score (TXT)55
5
Showing 10 of 10 rows

Other info

Code

Follow for update