arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.09800cs.CV
收藏

用于引导式文档表格生成的潜在扩散模型

Latent Diffusion for Guided Document Table Generation

Smarte Daten and Wissensdienste (SDS)(智能数据与知识服务公司(SDS));Deutsches Forschungszentrum für Künstliche Intelligenz GmbH (DFKI)(德国人工智能研究中心(DFKI));Department of Computer Science, RPTU Kaiserslautern-Landau(凯撒斯劳滕-兰道工业大学计算机科学系);DeepReader GmbH(DeepReader有限公司)

Syed Jawwad Haider Hamdani, Saifullah Saifullah, Stefan Agne, Andreas Dengel, Sheraz Ahmed

AI总结:

针对复杂表格标注数据稀缺的问题,提出基于潜在扩散模型、利用行列条件掩码引导生成带标注表格图像的方法,可提升合成数据质量,使YOLOv5在pubtables-1m上的性能接近SOTA。

摘要:

由于真实世界文档布局固有的多样性与复杂性,获取复杂表格的带标注表格结构数据是一项颇具挑战的任务。针对复杂表格结构提供全面标注的公开数据集十分稀缺,这阻碍了面向此类场景的模型的开发与评估。本研究论文提出了一种生成表格结构带标注图像的新方法,该方法通过潜在扩散模型(latent diffusion models),利用行与列的条件掩码图像来实现生成过程。所提方法旨在提升用于训练目标检测模型的合成数据的质量。具体而言,本研究采用一种条件机制来引导复杂文档表格图像的生成,确保表格布局的呈现具有真实感。为评估生成数据的有效性,我们采用广泛使用的YOLOv5目标检测模型进行训练。生成的表格图像可作为有价值的训练样本,以多样的表格结构丰富数据集。随后,我们在极具挑战性的pubtables-1m测试集上对模型进行测试,该测试集是复杂文档布局中表格结构识别的基准。实验结果表明,所提出的方法显著提升了训练用合成数据的质量,使得YOLOv5模型的性能得到增强。在pubtables-1m测试集上得到的平均精度均值(mean Average Precision, mAP)显示,其结果与当前最优方法(state-of-the-art methods)十分接近。此外,合成数据上得到的较低的Fréchet inception distance(FID)结果进一步验证了所提方法在生成表格结构带标注图像方面的有效性。

英文摘要:

Obtaining annotated table structure data for complex tables is a challenging task due to the inherent diversity and complexity of real-world document layouts. The scarcity of publicly available datasets with comprehensive annotations for intricate table structures hinders the development and evaluation of models designed for such scenarios. This research paper introduces a novel approach for generating annotated images for table structure by leveraging conditioned mask images of rows and columns through the application of latent diffusion models. The proposed method aims to enhance the quality of synthetic data used for training object detection models. Specifically, the study employs a conditioning mechanism to guide the generation of complex document table images, ensuring a realistic representation of table layouts. To evaluate the effectiveness of the generated data, we employ the popular YOLOv5 object detection model for training. The generated table images serve as valuable training samples, enriching the dataset with diverse table structures. The model is subsequently tested on the challenging pubtables-1m testset, a benchmark for table structure recognition in complex document layouts. Experimental results demonstrate that the introduced approach significantly improves the quality of synthetic data for training, leading to YOLOv5 models with enhanced performance. The mean Average Precision (mAP) values obtained on the pubtables-1m testset showcase results closely aligned with state-of-the-art methods. Furthermore, low FID results obtained on the synthetic data further validate the efficacy of the proposed methodology in generating annotated images for table structure.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑