Comparative Study of Undersampling Techniques in Text Classification
Imbalanced datasets pose critical challenges in text classification, often leading to biased predictions and reduced model performance. This study compares three undersampling techniques: Random UnderSampler, Near Miss, and Tomek Links, to evaluate their impact on five machine learning classifiers: Logistic Regression, Support Vector Machine, Random Forest, Naive Bayes, and Neural Network. The results demonstrate that the Random Forest classifier achieves the highest performance with precision of $\mathrm{9 7 \%}$, recall of $\mathrm{9 9 \%}$, and F1-score of $\mathrm{9 8 \%}$ across all methods, showcasing its robustness against data imbalance. The Near Miss technique improves Neural Network performance to a recall of $83 \%$ and F1-score of $\mathrm{7 1 \%}$, while Tomek Links further boosts Neural Network precision to $\mathrm{6 8 \%}$, recall to $\mathrm{7 2 \%}$, and F1-score to $\mathrm{7 0 \%}$. In contrast, simpler models like Naive Bayes exhibit limited improvements, with precision and recall scores around 47-51%. These findings underscore the importance of selecting appropriate undersampling strategies to enhance text classification outcomes and address data imbalance effectively.
Authors
- Albertus Dwiyoga Widiantoro (ORCID: https://orcid.org/0000-0001-6168-0968)
- Faiz Zulkifli (ORCID: https://orcid.org/0000-0002-2508-3231)
- Muhammad Zainul Arifin (ORCID: https://orcid.org/0000-0001-7611-8134)
- Nahdatul Akma Ahmad (ORCID: https://orcid.org/0000-0002-1079-6116)
- Usman Ependi (ORCID: https://orcid.org/0000-0002-5814-4045)
- Ade Putra
Institutions
- Soegijapranata Catholic University (ID)
- Universitas Bina Darma (ID)
- Universiti Teknologi MARA System (MY)
- Muria Kudus University (ID)
- Universiti Teknologi MARA (MY)
Publication Details
- Published
- 2024-10-24
- DOI
- https://doi.org/10.1109/icic64337.2024.10956648
- Citations
- 1
- Primary Topic
- Text and Document Classification Technologies
- Type
- article
- Field-Weighted Citation Impact
- 0.35