Multi-class Text Classification using BERT-based Active Learning
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
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,854
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,243
- 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,115
- 2015Character-level Convolutional Networks for Text Classificationcited by 2,293
- 2018Know What You Don't Know: Unanswerable Questions for SQuADcited by 1,538
- 2017RACE: Large-scale ReAding Comprehension Dataset From Examinationscited by 851
- 2017Deep Bayesian Active Learning with Image Datacited by 812
- 2019BatchBALD: Efficient and Diverse Batch Acquisition for Deep Bayesian Active Learningcited by 428
- 2017Snorkel: Rapid Training Data Creation with Weak Supervisioncited by 370
- 2019Learning Loss for Active Learningcited by 335
8 more
- 2015Bayesian Convolutional Neural Networks with Bernoulli Approximate Variational Inferencecited by 159
- 2019Discriminative Active Learningcited by 97
- 2018Deep Bayesian Active Learning for Natural Language Processing: Results of a Large-Scale Empirical Studycited by 90
- 2019Diverse mini-batch Active Learningcited by 88
- 2018Practical Obstacles to Deploying Active Learningcited by 68
- 2019Sampling Bias in Deep Active Classification: An Empirical Studycited by 36
- 2018Active Learning with Partial Feedbackcited by 27
- 2016Active Learning for Speech Recognition: the Power of Gradientscited by 25
Built on by
Citing papers per year: 2022: 2; 2023: 4; 2024: 0; 2025: 4; 2026: 1.
11 papers.
- 2023Understanding Telecom Language Through Large Language Modelscited by 32
- 2023Stance Detection: A Practical Guide to Classifying Political Beliefs in Textcited by 20
- 2023ActiveGLAE: A Benchmark for Deep Active Learning with Transformerscited by 14
- 2022Multilingual Detection of Personal Employment Status on Twittercited by 8
- 2025Reasoner for Real-World Event Detection: Scaling Reinforcement Learning via Adaptive Perplexity-Aware Sampling Strategycited by 3
- 2025Improving the Efficiency of Long Document Classification using Sentence Ranking Approachcited by 1
- 2025Advancing Text Classification with Large Language Models and Neural Attention Mechanismscited by 1
- 2022Nearest Neighbor Classifier with Margin Penalty for Active Learningcited by 0
- 2023Utilizing Weak Supervision To Generate Indonesian Conservation Datasetcited by 0
- 2025Efficient Zero-Shot Long Document Classification by Reducing Context Through Sentence Rankingcited by 0
- 2026Automating the Detection of Requirement Dependencies Using Large Language Modelscited by 0
- 2026Automating the Detection of Requirement Dependencies Using Large Language Models25 Feb 2026
- 2025Advancing Text Classification with Large Language Models and Neural Attention Mechanisms10 Dec 2025
- 2025Efficient Zero-Shot Long Document Classification by Reducing Context Through Sentence Ranking24 Aug 2025
- 2025Reasoner for Real-World Event Detection: Scaling Reinforcement Learning via Adaptive Perplexity-Aware Sampling Strategy2 Jul 2025
- 2025Improving the Efficiency of Long Document Classification using Sentence Ranking Approach8 Jun 2025
- 2023Utilizing Weak Supervision To Generate Indonesian Conservation Dataset17 Oct 2023
- 2023ActiveGLAE: A Benchmark for Deep Active Learning with Transformers16 Jun 2023
- 2023Understanding Telecom Language Through Large Language Models9 Jun 2023
- 2023Stance Detection: A Practical Guide to Classifying Political Beliefs in Text2 May 2023
- 2022Multilingual Detection of Personal Employment Status on Twitter17 Mar 2022
- 2022Nearest Neighbor Classifier with Margin Penalty for Active Learning17 Mar 2022