Skip to main navigation Skip to search Skip to main content

Large-Scale News Classification using BERT Language Model: Spark NLP Approach

Research output: Chapter in Book/Report/Conference proceeding › Conference contribution › peer-review

Abstract

The rise of big data analytics on top of NLP increasing the computational burden for text processing at scale. The problems faced in NLP are very high dimensional text, so it takes a high computation resource. The MapReduce allows parallelization of large computations and can improve the efficiency of text processing. This research aims to study the effect of big data processing on NLP tasks based on a deep learning approach. We classify a big text of news topics with fine-tuning BERT used pre-trained models. Five pre-trained models with a different number of parameters were used in this study. To measure the efficiency of this method, we compared the performance of the BERT with the pipelines from Spark NLP. The result shows that BERT without Spark NLP gives higher accuracy compared to BERT with Spark NLP. The accuracy average and training time of all model's using BERT is 0.9187 and 35 minutes while using BERT with Spark NLP pipeline is 0.8444 and 9 minutes. The bigger model will take more computation resources and need a longer time to complete the tasks. However, the accuracy of BERT with Spark NLP only decreased by an average of 5.7%, while the training time was reduced significantly by 62.9% compared to BERT without Spark NLP.

Original languageEnglish
Title of host publicationProceedings of 2021 International Conference on Sustainable Information Engineering and Technology, SIET 2021
PublisherAssociation for Computing Machinery
Pages240-246
Number of pages7
ISBN (Electronic)9781450384070
DOIs
Publication statusPublished - 13 Sept 2021
Event6th International Conference on Sustainable Information Engineering and Technology, SIET 2021 - Virtual, Online, Indonesia
Duration: 13 Sept 2021 → 14 Sept 2021

Publication series

NameACM International Conference Proceeding Series

Conference

Conference6th International Conference on Sustainable Information Engineering and Technology, SIET 2021
Country/TerritoryIndonesia
CityVirtual, Online
Period13/09/21 → 14/09/21

Keywords

  • BERT language model
  • distributed NLP architectures
  • Large-scale text classification
  • Spark NLP

Fingerprint

Dive into the research topics of 'Large-Scale News Classification using BERT Language Model: Spark NLP Approach'. Together they form a unique fingerprint.

Cite this