mlsanityData to 7 Oct 2026

Multi-class Text Classification using BERT-based Active Learning

arXiv 2104.14289 ↗First posted 27 Apr 2021cs.IR cs.LGcited by 11

Not on the map. Why?

Abstract (arXiv)

Text Classification finds interesting applications in the pickup and delivery services industry where customers require one or more items to be picked up from a location and delivered to a certain destination. Classifying these customer transactions into multiple categories helps understand the market needs for different customer segments. Each transaction is accompanied by a text description provided by the customer to describe the products being picked up and delivered which can be used to classify the transaction. BERT-based models have proven to perform well in Natural Language Understanding. However, the product descriptions provided by the customers tend to be short, incoherent and code-mixed (Hindi-English) text which demands fine-tuning of such models with manually labelled data to achieve high accuracy. Collecting this labelled data can prove to be expensive. In this paper, we explore Active Learning strategies to label transaction descriptions cost effectively while using BERT to train a transaction classification model. On TREC-6, AG's News Corpus and an internal dataset, we benchmark the performance of BERT across different Active Learning strategies in Multi-Class Text Classification.

Key facts. Cited by 11 papers in the map (data through 7 Oct 2026). Builds on 20 of its 49 references that are in the map. First posted on arXiv 27 Apr 2021.

Builds on

20 of 49 references.

References · 20

  1. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  2. 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
  3. 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,854
  4. 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,243
  5. 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,115
  6. 2015Character-level Convolutional Networks for Text Classificationcited by 2,293
  7. 2018Know What You Don't Know: Unanswerable Questions for SQuADcited by 1,538
  8. 2017RACE: Large-scale ReAding Comprehension Dataset From Examinationscited by 851
  9. 2017Deep Bayesian Active Learning with Image Datacited by 812
  10. 2019BatchBALD: Efficient and Diverse Batch Acquisition for Deep Bayesian Active Learningcited by 428
  11. 2017Snorkel: Rapid Training Data Creation with Weak Supervisioncited by 370
  12. 2019Learning Loss for Active Learningcited by 335
8 more
  1. 2015Bayesian Convolutional Neural Networks with Bernoulli Approximate Variational Inferencecited by 159
  2. 2019Discriminative Active Learningcited by 97
  3. 2018Deep Bayesian Active Learning for Natural Language Processing: Results of a Large-Scale Empirical Studycited by 90
  4. 2019Diverse mini-batch Active Learningcited by 88
  5. 2018Practical Obstacles to Deploying Active Learningcited by 68
  6. 2019Sampling Bias in Deep Active Classification: An Empirical Studycited by 36
  7. 2018Active Learning with Partial Feedbackcited by 27
  8. 2016Active Learning for Speech Recognition: the Power of Gradientscited by 25

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2022: 2; 2023: 4; 2024: 0; 2025: 4; 2026: 1.

11 papers.