English

Advancing Scientific Text Classification: Fine-Tuned Models with Dataset Expansion and Hard-Voting

Computation and Language 2025-09-09 v2 Artificial Intelligence

Abstract

Efficient text classification is essential for handling the increasing volume of academic publications. This study explores the use of pre-trained language models (PLMs), including BERT, SciBERT, BioBERT, and BlueBERT, fine-tuned on the Web of Science (WoS-46985) dataset for scientific text classification. To enhance performance, we augment the dataset by executing seven targeted queries in the WoS database, retrieving 1,000 articles per category aligned with WoS-46985's main classes. PLMs predict labels for this unlabeled data, and a hard-voting strategy combines predictions for improved accuracy and confidence. Fine-tuning on the expanded dataset with dynamic learning rates and early stopping significantly boosts classification accuracy, especially in specialized domains. Domain-specific models like SciBERT and BioBERT consistently outperform general-purpose models such as BERT. These findings underscore the efficacy of dataset augmentation, inference-driven label prediction, hard-voting, and fine-tuning techniques in creating robust and scalable solutions for automated academic text classification.

Keywords

Cite

@article{arxiv.2504.19021,
  title  = {Advancing Scientific Text Classification: Fine-Tuned Models with Dataset Expansion and Hard-Voting},
  author = {Zhyar Rzgar K Rostam and Gábor Kertész},
  journal= {arXiv preprint arXiv:2504.19021},
  year   = {2025}
}

Comments

6 pages, 1 figure, 8 tables

R2 v1 2026-06-28T23:12:33.282Z