arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2006.09141cs.CVcs.DCcs.LG
收藏

Improving accuracy and speeding up Document Image Classification through parallel systems

Barcelona Supercomputing Center - Centro Nacional de Supercomputación(巴塞罗那超级计算中心-国家超级计算中心);Universitat Politècnica de Catalunya, UPC-BarcelonaTech(加泰罗尼亚理工大学(UPC-巴塞罗那理工));Serimag Media - TAAD;CaixaBank(凯克萨银行)

Javier Ferrando, Juan Luis Dominguez, Jordi Torres, Raul Garcia, David Garcia, Daniel Garrido, Jordi Cortada, Mateo Valero

英文摘要:

This paper presents a study showing the benefits of the EfficientNet models compared with heavier Convolutional Neural Networks (CNNs) in the Document Classification task, essential problem in the digitalization process of institutions. We show in the RVL-CDIP dataset that we can improve previous results with a much lighter model and present its transfer learning capabilities on a smaller in-domain dataset such as Tobacco3482. Moreover, we present an ensemble pipeline which is able to boost solely image input by combining image model predictions with the ones generated by BERT model on extracted text by OCR. We also show that the batch size can be effectively increased without hindering its accuracy so that the training process can be sped up by parallelizing throughout multiple GPUs, decreasing the computational time needed. Lastly, we expose the training performance differences between PyTorch and Tensorflow Deep Learning frameworks.

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

↑