DocBERT: BERT for Document Classification
Not on the map. Why?
Abstract (arXiv)
We present, to our knowledge, the first application of BERT to document classification. A few characteristics of the task might lead one to think that BERT is not the most appropriate model: syntactic structures matter less for content categories, documents can often be longer than typical BERT input, and documents often have multiple labels. Nevertheless, we show that a straightforward classification model using BERT is able to achieve the state of the art across four popular datasets. To address the computational expense associated with BERT inference, we distill knowledge from BERT-large to small bidirectional LSTMs, reaching BERT-base parity on multiple datasets using 30x fewer parameters. The primary contribution of our paper is improved baselines that can provide the foundation for future work.
Key facts. Cited by 68 papers in the map (data through 7 Oct 2026). First posted on arXiv 17 Apr 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 36; 2022: 11; 2023: 12; 2024: 5; 2025: 2; 2026: 2.
68 papers.
Sort:
By area:
- 2023A Survey of Large Language Models in Medicine: Progress, Application, and Challengecited by 123
- 2022Revisiting Transformer-based Models for Long Document Classificationcited by 48
- 2021Edge-Cloud Polarization and Collaboration: A Comprehensive Survey for AIcited by 38
- 2021Scaling Federated Learning for Fine-tuning of Large Language Modelscited by 36
- 2023DRG-LLaMA : Tuning LLaMA Model to Predict Diagnosis-related Group for Hospitalized Patientscited by 29
- 2021Exploring Transformers in Natural Language Generation: GPT, BERT, and XLNetcited by 27
- 2021Automated scholarly paper review: Concepts, technologies, and challengescited by 21
- 2021FakeFlow: Fake News Detection by Modeling the Flow of Affective Informationcited by 20
- 2021Enhancing Label Correlation Feedback in Multi-Label Text Classification via Multi-Task Learningcited by 15
- 2021Incorporating Uncertainty in Learning to Defer Algorithms for Safe Computer-Aided Diagnosiscited by 15
- 2021"Short is the Road that Leads from Fear to Hate": Fear Speech in Indian WhatsApp Groupscited by 13
- 2021Automated Quality Assessment of Cognitive Behavioral Therapy Sessions Through Highly Contextualized Language Representationscited by 13
38 more
- 2021Comparative Study of Long Document Classificationcited by 12
- 2021Goldilocks: Just-Right Tuning of BERT for Technology-Assisted Reviewcited by 11
- 2021You too Brutus! Trapping Hateful Users in Social Media: Challenges, Solutions & Insightscited by 11
- 2022Improving Short Text Classification With Augmented Data Using GPT-3cited by 11
- 2021Comparison of Czech Transformers on Text Classification Taskscited by 10
- 2023FedTherapist: Mental Health Monitoring with User-Generated Linguistic Expressions on Smartphones via Federated Learningcited by 10
- 2022Automated Identification of Toxic Code Reviews Using ToxiCRcited by 8
- 2021Evaluating Pre-Trained Models for User Feedback Analysis in Software Engineering: A Study on Classification of App-Reviewscited by 7
- 2021Deep Transfer Learning & Beyond: Transformer Language Models in Information Systems Researchcited by 7
- 2021Unsupervised Matching of Data and Textcited by 7
- 2022Hierarchical Neural Network Approaches for Long Document Classificationcited by 7
- 2022Less is More: Parameter-Free Text Classification with Gzipcited by 7
- 2021Three-level Hierarchical Transformer Networks for Long-sequence and Multiple Clinical Documents Classificationcited by 6
- 2021PyEuroVoc: A Tool for Multilingual Legal Document Classification with EuroVoc Descriptorscited by 6
- 2021Human Guided Exploitation of Interpretable Attention Patterns in Summarization and Topic Segmentationcited by 6
- 2022Towards Understanding Large-Scale Discourse Structures in Pre-Trained and Fine-Tuned Language Modelscited by 6
- 2022Protoformer: Embedding Prototypes for Transformerscited by 6
- 2023ATLAS: Automatically Detecting Discrepancies Between Privacy Policies and Privacy Labelscited by 5
- 2023Well-calibrated Confidence Measures for Multi-label Text Classification with a Large Number of Labelscited by 5
- 2021Lacking the embedding of a word? Look it up into a traditional dictionarycited by 4
- 2024Machine Unlearning for Document Classificationcited by 4
- 2021Classifying Long Clinical Documents with Pre-trained Transformerscited by 3
- 2021Sentiment analysis in tweets: an assessment study from classical to modern text representation modelscited by 3
- 2021Efficient Document Image Classification Using Region-Based Graph Neural Networkcited by 3
- 2021Large-Scale News Classification using BERT Language Model: Spark NLP Approachcited by 3
- 2022Can Language Representation Models Think in Bets?cited by 3
- 2023Ticket-BERT: Labeling Incident Management Tickets with Language Modelscited by 3
- 2024PecSched: Preemptive and Efficient Cluster Scheduling for LLM Inferencecited by 3
- 2025Unsupervised Document and Template Clustering using Multimodal Embeddingscited by 3
- 2021On-Device Document Classification using multimodal featurescited by 2
- 2021One System to Rule them All: a Universal Intent Recognition System for Customer Service Chatbotscited by 2
- 2022Understanding Prior Bias and Choice Paralysis in Transformer-based Language Representation Models through Four Experimental Probescited by 2
- 2023Multi label classification of Artificial Intelligence related patents using Modified D2SBERT and Sentence Attention mechanismcited by 2
- 2024TacoERE: Cluster-aware Compression for Event Relation Extractioncited by 2
- 2024FDC: Fast KV Dimensionality Compression for Efficient LLM Inferencecited by 2
- 2025Label-semantics Aware Generative Approach for Domain-Agnostic Multilabel Classificationcited by 2
- 2021Does a Hybrid Neural Network based Feature Selection Model Improve Text Classification?cited by 1
- 2021Digital Twin Based Disaster Management System Proposal: DT-DMScited by 1
50 most cited of 68 All 68 (JSON)
- 2026DocHRL: A Hierarchical Reinforcement Learning Framework for Cost-Optimised Document Classification24 Jun 2026
- 2026CzechTopic: A Benchmark for Zero-Shot Topic Localization in Historical Czech Documents4 Mar 2026
- 2025Unsupervised Document and Template Clustering using Multimodal Embeddings13 Jun 2025
- 2025Label-semantics Aware Generative Approach for Domain-Agnostic Multilabel Classification7 Jun 2025
- 2024Text Classification using Graph Convolutional Networks: A Comprehensive Survey12 Oct 2024
- 2024PecSched: Preemptive and Efficient Cluster Scheduling for LLM Inference23 Sep 2024
- 2024FDC: Fast KV Dimensionality Compression for Efficient LLM Inference7 Aug 2024
- 2024TacoERE: Cluster-aware Compression for Event Relation Extraction11 May 2024
- 2024Machine Unlearning for Document Classification29 Apr 2024
- 2023Well-calibrated Confidence Measures for Multi-label Text Classification with a Large Number of Labels14 Dec 2023
- 2023Summarization-based Data Augmentation for Document Classification1 Dec 2023
- 2023A Survey of Large Language Models in Medicine: Progress, Application, and Challenge9 Nov 2023
38 more
- 2023FedTherapist: Mental Health Monitoring with User-Generated Linguistic Expressions on Smartphones via Federated Learning25 Oct 2023
- 2023DRG-LLaMA : Tuning LLaMA Model to Predict Diagnosis-related Group for Hospitalized Patients22 Sep 2023
- 2023Power-up! What Can Generative Models Do for Human Computation Workflows?5 Jul 2023
- 2023Ticket-BERT: Labeling Incident Management Tickets with Language Models30 Jun 2023
- 2023ATLAS: Automatically Detecting Discrepancies Between Privacy Policies and Privacy Labels24 May 2023
- 2023Leveraging BERT Language Model for Arabic Long Document Classification4 May 2023
- 2023Label Dependencies-aware Set Prediction Networks for Multi-label Text Classification14 Apr 2023
- 2023Multi label classification of Artificial Intelligence related patents using Modified D2SBERT and Sentence Attention mechanism3 Mar 2023
- 2023Topics in Contextualised Attention Embeddings11 Jan 2023
- 2022Less is More: Parameter-Free Text Classification with Gzip19 Dec 2022
- 2022Can Language Representation Models Think in Bets?14 Oct 2022
- 2022Understanding Prior Bias and Choice Paralysis in Transformer-based Language Representation Models through Four Experimental Probes3 Oct 2022
- 2022Multimodal E-Commerce Product Classification Using Hierarchical Fusion7 Jul 2022
- 2022Protoformer: Embedding Prototypes for Transformers25 Jun 2022
- 2022Improving Short Text Classification With Augmented Data Using GPT-323 May 2022
- 2022CTM -- A Model for Large-Scale Multi-View Tweet Topic Classification3 May 2022
- 2022Revisiting Transformer-based Models for Long Document Classification14 Apr 2022
- 2022Towards Understanding Large-Scale Discourse Structures in Pre-Trained and Fine-Tuned Language Models8 Apr 2022
- 2022Automated Identification of Toxic Code Reviews Using ToxiCR26 Feb 2022
- 2022Hierarchical Neural Network Approaches for Long Document Classification18 Jan 2022
- 2021Attention-based Bidirectional LSTM for Deceptive Opinion Spam Classification29 Dec 2021
- 2021LAME: Layout Aware Metadata Extraction Approach for Research Articles23 Dec 2021
- 2021Unsupervised Matching of Data and Text16 Dec 2021
- 2021One System to Rule them All: a Universal Intent Recognition System for Customer Service Chatbots15 Dec 2021
- 2021Human Guided Exploitation of Interpretable Attention Patterns in Summarization and Topic Segmentation10 Dec 2021
- 2021Automated scholarly paper review: Concepts, technologies, and challenges15 Nov 2021
- 2021Edge-Cloud Polarization and Collaboration: A Comprehensive Survey for AI11 Nov 2021
- 2021Comparative Study of Long Document Classification1 Nov 2021
- 2021DeepHelp: Deep Learning for Shout Crisis Text Conversations25 Oct 2021
- 2021Deep Transfer Learning & Beyond: Transformer Language Models in Information Systems Research18 Oct 2021
- 2021OPAD: An Optimized Policy-based Active Learning Framework for Document Content Analysis1 Oct 2021
- 2021COVID-19 Fake News Detection Using Bidirectional Encoder Representations from Transformers Based Models30 Sep 2021
- 2021Lacking the embedding of a word? Look it up into a traditional dictionary24 Sep 2021
- 2021Incorporating Uncertainty in Learning to Defer Algorithms for Safe Computer-Aided Diagnosis17 Aug 2021
- 2021PyEuroVoc: A Tool for Multilingual Legal Document Classification with EuroVoc Descriptors2 Aug 2021
- 2021You too Brutus! Trapping Hateful Users in Social Media: Challenges, Solutions & Insights1 Aug 2021
- 2021Comparison of Czech Transformers on Text Classification Tasks21 Jul 2021
- 2021Large-Scale News Classification using BERT Language Model: Spark NLP Approach14 Jul 2021
50 newest of 68 All 68 (JSON)
- 2024Machine Unlearning for Document Classificationcited by 4
- 2024Machine Unlearning for Document Classification29 Apr 2024
- 2023A Survey of Large Language Models in Medicine: Progress, Application, and Challengecited by 123
- 2022Revisiting Transformer-based Models for Long Document Classificationcited by 48
- 2021Edge-Cloud Polarization and Collaboration: A Comprehensive Survey for AIcited by 38
- 2021Scaling Federated Learning for Fine-tuning of Large Language Modelscited by 36
- 2023DRG-LLaMA : Tuning LLaMA Model to Predict Diagnosis-related Group for Hospitalized Patientscited by 29
- 2021Exploring Transformers in Natural Language Generation: GPT, BERT, and XLNetcited by 27
- 2021Automated scholarly paper review: Concepts, technologies, and challengescited by 21
- 2021FakeFlow: Fake News Detection by Modeling the Flow of Affective Informationcited by 20
- 2021Enhancing Label Correlation Feedback in Multi-Label Text Classification via Multi-Task Learningcited by 15
- 2021Incorporating Uncertainty in Learning to Defer Algorithms for Safe Computer-Aided Diagnosiscited by 15
- 2021"Short is the Road that Leads from Fear to Hate": Fear Speech in Indian WhatsApp Groupscited by 13
- 2021Automated Quality Assessment of Cognitive Behavioral Therapy Sessions Through Highly Contextualized Language Representationscited by 13
12 most cited of 64 All 64 (JSON)
- 2026DocHRL: A Hierarchical Reinforcement Learning Framework for Cost-Optimised Document Classification24 Jun 2026
- 2026CzechTopic: A Benchmark for Zero-Shot Topic Localization in Historical Czech Documents4 Mar 2026
- 2025Unsupervised Document and Template Clustering using Multimodal Embeddings13 Jun 2025
- 2025Label-semantics Aware Generative Approach for Domain-Agnostic Multilabel Classification7 Jun 2025
- 2024Text Classification using Graph Convolutional Networks: A Comprehensive Survey12 Oct 2024
- 2024PecSched: Preemptive and Efficient Cluster Scheduling for LLM Inference23 Sep 2024
- 2024TacoERE: Cluster-aware Compression for Event Relation Extraction11 May 2024
- 2023Well-calibrated Confidence Measures for Multi-label Text Classification with a Large Number of Labels14 Dec 2023
- 2023Summarization-based Data Augmentation for Document Classification1 Dec 2023
- 2023A Survey of Large Language Models in Medicine: Progress, Application, and Challenge9 Nov 2023
- 2023FedTherapist: Mental Health Monitoring with User-Generated Linguistic Expressions on Smartphones via Federated Learning25 Oct 2023
- 2023DRG-LLaMA : Tuning LLaMA Model to Predict Diagnosis-related Group for Hospitalized Patients22 Sep 2023
12 newest of 64 All 64 (JSON)