COMPARISON OF GAUSSIAN NAIVE BAYES AND RANDOM FOREST FOR ANEMIA CLASSIFICATION USING HEMATOLOGICAL PARAMETERS
Abstract
Anemia is a global health problem affecting approximately 1.92 billion people, or 24% of the population, according to the WHO. Accurate early detection is crucial for data-driven healthcare. This study evaluates two machine learning algorithms, Gaussian Naive Bayes (GNB) and Random Forest (RF). The classification is based on four key hematological parameters: Hemoglobin (Hb), Mean Corpuscular Volume (MCV), Mean Corpuscular Hemoglobin (MCH), and Mean Corpuscular Hemoglobin Concentration (MCHC). GNB relies on Bayes' Theorem with Gaussian distribution assumptions, whereas RF is a decision-tree-based ensemble method capable of capturing non-linear patterns without specific distributional assumptions. Evaluated using 5-fold cross-validation and standard metrics (accuracy, precision, recall, F1-score), results showed that RF outperformed GNB. RF achieved 94.2% accuracy (CV 94.9% ± 1.1%), compared to GNB's 90.5% (CV 90.1% ± 1.3%). RF feature importance confirmed Hb as the dominant predictor (score 0.562), aligning with its strong correlation to anemia (r = −0.80). Although not surpassing larger-scale studies, these results remain highly competitive. Ultimately, this research provides evidence to support the development of automated, data-driven clinical decision support systems for anemia detection.
Downloads
References
K. Abbotts and P. Sriskandarajah, “Anemia of Chronic Disease: Pathophysiology, Diagnosis and Management,” Hematol. Rep., vol. 18, no. 4, p. 48, Jul. 2026, doi: 10.3390/hematolrep18040048.
World Health Organization, “Anaemia,” 2024, World Health Organization, Geneva, Switzerland. [Online]. Available: https://www.who.int/news-room/fact-sheets/detail/anaemia
G. A. Stevens et al., “National, regional, and global estimates of anaemia by severity in women and children for 2000–19: a pooled analysis of population-representative data,” Lancet Glob. Health, vol. 10, no. 5, pp. e627–e639, 2022, doi: 10.1016/S2214-109X(22)00084-5.
A. G. Mugdha, F. T. Pinki, and S. K. Talukdhar, “Hemoglobin Estimation and Anemia Severity Prediction Using Machine Learning Algorithms,” in 2023 5th International Conference on Sustainable Technologies for Industry 5.0 (STI), 2023, pp. 1–6. doi: 10.1109/STI59863.2023.10464565.
Md. M. Rahman, M. U. Mojumdar, H. A. Shifa, N. R. Chakraborty, N. P. Stenin, and Md. A. Hasan, “Anemia Disease Prediction using Machine Learning Techniques and Performance Analysis,” in 2024 11th International Conference on Computing for Sustainable Global Development (INDIACom), 2024, pp. 1276–1282. doi: 10.23919/INDIACom61295.2024.10498962.
A. Rajkomar, J. Dean, and I. Kohane, “Machine Learning in Medicine,” New England Journal of Medicine, vol. 380, no. 14, pp. 1347–1358, 2019, doi: 10.1056/NEJMra1814259.
A. Badhoutiya, R. P. Verma, A. Shrivastava, K. Laxminarayanamma, A. L. N. Rao, and A. K. Khan, “Random Forest Classification in Healthcare Decision Support for Disease Diagnosis,” in 2023 International Conference on Artificial Intelligence for Innovations in Healthcare Industries (ICAIIHI), 2023, pp. 1–7. doi: 10.1109/ICAIIHI57871.2023.10489244.
M. Maydanchi et al., “Comparative Study of Decision Tree, AdaBoost, Random Forest, Naïve Bayes, KNN, and Perceptron for Heart Disease Prediction,” in SoutheastCon 2023, 2023, pp. 204–208. doi: 10.1109/SoutheastCon51012.2023.10115189.
H. Kamel, D. Abdulah, and J. M. Al-Tuwaijari, “Cancer Classification Using Gaussian Naive Bayes Algorithm,” in 2019 International Engineering Conference (IEC), 2019, pp. 165–170. doi: 10.1109/IEC47844.2019.8950650.
M. Maindola et al., “Utilizing Random Forests for High-Accuracy Classification in Medical Diagnostics,” in 2024 7th International Conference on Contemporary Computing and Informatics (IC3I), 2024, pp. 1679–1685. doi: 10.1109/IC3I61595.2024.10828609.
I. As’ad, “Advancing Healthcare Diagnostics: A Study on Gaussian Naive Bayes Classification of Blood Samples,” International Journal of Artificial Intelligence in Medical Issues, vol. 1, no. 2, pp. 115–123, Nov. 2023, doi: 10.56705/ijaimi.v1i2.120.
A. H. Jahromi and M. Taheri, “A non-parametric mixture of Gaussian naive Bayes classifiers based on local independent features,” in 2017 Artificial Intelligence and Signal Processing Conference (AISP), 2017, pp. 209–212. doi: 10.1109/AISP.2017.8324083.
F.-J. Yang, “An Implementation of Naive Bayes Classifier,” in 2018 International Conference on Computational Science and Computational Intelligence (CSCI), 2018, pp. 301–306. doi: 10.1109/CSCI46756.2018.00065.
X. Hu, M. Cui, and B. Chen, “Feature selection based on random forest and application in correlation analysis of symptom and disease,” in 2009 IEEE International Symposium on IT in Medicine & Education, 2009, pp. 120–124. doi: 10.1109/ITIME.2009.5236450.
D. M. Reif, A. A. Motsinger, B. A. McKinney, J. E. Crowe, and J. H. Moore, “Feature Selection using a Random Forests Classifier for the Integrated Analysis of Multiple Data Types,” in 2006 IEEE Symposium on Computational Intelligence and Bioinformatics and Computational Biology, 2006, pp. 1–8. doi: 10.1109/CIBCB.2006.330987.
K. Swarupa, V. H. Sree, S. Nookambika, Y. K. S. Kishore, and U. R. Teja, “Disease Prediction: Smart Disease Prediction System using Random Forest Algorithm,” in 2021 IEEE International Conference on Intelligent Systems, Smart and Green Technologies (ICISSGT), 2021, pp. 48–51. doi: 10.1109/ICISSGT52025.2021.00021.
S. Paul, P. Ranjan, S. Kumar, and A. Kumar, “Disease Predictor Using Random Forest Classifier,” in 2022 International Conference for Advancement in Technology (ICONAT), 2022, pp. 1–4. doi: 10.1109/ICONAT53423.2022.9726023.
B. R. Rao, “Anemia Dataset,” 2022, Kaggle. [Online]. Available: https://www.kaggle.com/datasets/biswaranjanrao/anemia-dataset
N. K. Mangalapu, L. T K, G. J, G. Kalyani, and B. K S, “Revolutionizing Anaemia Detection: Leveraging Eye Condition Data and Machine Learning,” in 2025 8th International Conference on Electronics, Materials Engineering & Nano-Technology (IEMENTech), 2025, pp. 1–6. doi: 10.1109/IEMENTech65115.2025.10959598.
D. E. Alarcón-Yaquetto et al., “Hematological Parameters and Iron Status in Adult Men and Women Using Altitude Adjusted and Unadjusted Hemoglobin Values for Anemia Diagnosis in Cusco, Peru (3400 MASL),” Physiologia, vol. 2, no. 1, pp. 1–19, 2022, doi: 10.3390/physiologia2010001.
J. G. Gómez, C. Parra Urueta, D. S. Álvarez, V. Hernández Riaño, and G. Ramirez-Gonzalez, “Anemia Classification System Using Machine Learning,” Informatics, vol. 12, no. 1, 2025, doi: 10.3390/informatics12010019.
Sathya, G. S. Priyatharsini, A. Kumar. V, I. Vasudevan, and M. Umapathy, “Machine Learning Models for Predicting Anemia: Evaluation and Performance Insights,” in 2024 First International Conference on Innovations in Communications, Electrical and Computer Engineering (ICICEC), 2024, pp. 1–7. doi: 10.1109/ICICEC62498.2024.10808776.
A. M, J. R, and Sasirekha. E, “Data to Diagnosis Machine Learning Models for Accurate Anemia Classifications,” in 2025 International Conference on Data Science and Business Systems (ICDSBS), 2025, pp. 1–6. doi: 10.1109/ICDSBS63635.2025.11031746.
Baik Budi and Rizky Sarhans, “Hybrid Intelligence For Precision Milk Quality Assessment Integrating Naïve Bayes And K-Means Clustering,” Jurnal Informatika dan Teknik Elektro Terapan, vol. 14, no. 2, Apr. 2026, doi: 10.23960/jitet.v14i2.9357.
Mohanarathinam, and J. Velusamy, “Optimized SMOTE for Imbalanced Data Handling in Machine Learning,” in 2025 3rd International Conference on Advancement in Computation & Computer Technologies (InCACCT), 2025, pp. 349–354. doi: 10.1109/InCACCT65424.2025.11011340.
J. Wang, M. Xu, H. Wang, and J. Zhang, “Classification of Imbalanced Data by Using the SMOTE Algorithm and Locally Linear Embedding,” in 2006 8th international Conference on Signal Processing, 2006. doi: 10.1109/ICOSP.2006.345752.
K. Pal and Biraj. V Patel, “Data Classification with k-fold Cross Validation and Holdout Accuracy Estimation Methods with 5 Different Machine Learning Techniques,” in 2020 Fourth International Conference on Computing Methodologies and Communication (ICCMC), 2020, pp. 83–87. doi: 10.1109/ICCMC48092.2020.ICCMC-00016.
This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.
