Conference Paper2024

Predictive Modeling of Infant Birth Weight with Kolmogorov-Arnold Networks

Mohammed Tashfiq Islam, Md. Tahmid Hossain, Solayman Alam Dip, Jahanggir Hossain Setu, Nabarun Halder, Ashraful Islam, M Ashraful Amin

2024 27th International Conference on Computer and Information Technology (ICCIT)

IEEE, pp. 2974-2979

DOI: 10.1109/iccit64611.2024.11021802URL

Abstract

Birth weight is an important indicator of neonatal health. Because it influences immediate and long-term health outcomes. This study aims to predict infant birth weight using advanced Machine Learning (ML) techniques, focusing on four classification models, namely, Random Forest, KolmogorovArnold Networks (KAN), Multi-Layer Perceptron (MLP), and eXtreme Gradient Boosting (XGBoost). A dataset from the Centers for Disease Control and Prevention (CDC)’s Natality Records, containing 108,082 entries and 37 attributes, was utilized to train and evaluate these models. Feature selection was conducted using Mutual Information. It reveals key predictors, such as maternal morbidity, intervals since the last live birth, and maternal health factors. The models were evaluated based on F1-score, recall, precision, accuracy, and Receiver Operating Characteristics-Area Under Curve (ROC-AUC). Results demonstrated that KAN outperformed all other classifiers. It achieves an impressive micro and macro average AUC score of 97%. XGBoost followed closely with an AUC of 95%, while MLP and Random Forest exhibited scores of 94% and 86%, respectively. This study not only emphasizes the efficacy of ML in predicting infant birth weight but also lays the groundwork for further research in enhancing neonatal health outcomes.

Keywords

  • Kolmogorov-Arnold networks
  • maternal morbidity
  • interpregnancy interval
  • Feature selection
  • Mutual information
  • neonatal outcomes
  • XGBoost
  • Random forest
  • Multilayer perceptron

CCDS Authors

Nabarun Halder
Nabarun Halder

Nabarun Halder

  • Research Assistant, CCDS

Published at

UbiComp, IJCNN, HCII, Pattern Recognition, IEEE Access

Ashraful Islam, PhD
Ashraful Islam

Ashraful Islam, PhD

  • Center Director, CCDS
  • Director, HCI Wing

Published at

CHI, UIST, UbiComp, IJCNN, ICIP, HCII, ICDMW, BLP, Pattern Recognition, IEEE Internet of Things Journal, JMIR, JMIR Human Factors, JMIR Formative Research, IEEE Access, Smart Agricultural Technology, Diabetes Research and Clinical Practice, PLOS ONE

Prof. M Ashraful Amin, PhD
M Ashraful Amin, PhD

Prof. M Ashraful Amin, PhD

  • Founder Director, CCDS
  • Director, AI/ML, HCI & DS Wings

Published at

ACL, CHI, UbiComp, ECAI, WACV, PAKDD, IJCNN, ICPR, ICIP, HCII, CLEF, BLP, Scientific Reports, Pattern Recognition, JMIR, IEEE Access, Smart Agricultural Technology, Frontiers in Computational Neuroscience, Diabetes Research and Clinical Practice, PLOS ONE

References

  1. 1.Tianqi Chen, Carlos Guestrin. (2016). XGBoost. ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD), 785–794[10.1145/2939672.2939785]
  2. 2.H. Taud, J.F. Mas. (2017). Multilayer Perceptron (MLP). Lecture notes in geoinformation and cartography, 451–455[10.1007/978-3-319-60801-3_27]
  3. 3.Aakash Parmar, Rakesh Katariya, Vatsal Patel. (2018). A Review on Random Forest: An Ensemble Classifier. Lecture notes on data engineering and communications technologies, 758–763[10.1007/978-3-030-03146-6_86]
  4. 4.Željko Ð. Vujovic. (2021). Classification Model Evaluation Metrics. International Journal of Advanced Computer Science and Applications, 12(6)[10.14569/ijacsa.2021.0120670]
  5. 5.WHO Immediate KMC Study Group. (2021). Immediate “Kangaroo Mother Care” and Survival of Infants with Low Birth Weight. New England Journal of Medicine, 384(21), 2028–2038[10.1056/nejmoa2026486]
  6. 6.Hongfang Zhou, Xiqian Wang, Rourou Zhu. (2021). Feature selection based on mutual information with correlation coefficient. Applied Intelligence, 52(5), 5457–5474[10.1007/s10489-021-02524-x]
  7. 7.Zhangyu Cheng, Chengming Zou, Jianwei Dong. (2019). Outlier detection using isolation forest and local outlier factor. ACM Conference on Research in Adaptive and Convergent Systems (RACS), 161–168[10.1145/3338840.3355641]
  8. 8.Anil K. C., Prem Lal Basel, Sarswoti Singh. (2020). Low birth weight and its associated risk factors: Health facility-based case-control study. PLoS ONE, 15(6), e0234907[10.1371/journal.pone.0234907]
  9. 9.S. M. Ashikul Islam Pollob, Md. Menhazul Abedin, Md. Touhidul Islam, Md. Merajul Islam, Md. Maniruzzaman. (2022). Predicting risks of low birth weight in Bangladesh with machine learning. PLoS ONE, 17(5), e0267190[10.1371/journal.pone.0267190]
  10. 10.Wondesen Teshome Bekele. (2022). Machine learning algorithms for predicting low birth weight in Ethiopia. BMC Medical Informatics and Decision Making, 22(1), 232[10.1186/s12911-022-01981-9]
  11. 11.Jing Tao, Zhenming Yuan, Li Sun, Kai Yu, Zhifen Zhang. (2021). Fetal birthweight prediction with measured data by a temporal machine learning method. BMC Medical Informatics and Decision Making, 21(1), 26[10.1186/s12911-021-01388-y]
  12. 12.Janne Ørskou, Tine Brink Henriksen, Ulrik Kesmodel, Niels Jørgen Secher. (2003). Maternal Characteristics and Lifestyle Factors and the Risk of Delivering High Birth Weight Infants. Obstetrics and Gynecology, 102(1), 115–120[10.1016/s0029-7844(03)00402-2]
  13. 13.Namasivayam Ambalavanan, Waldemar A. Carlo, Georgiy Bobashev, Erin Mathias, Bing Liu, Kenneth Poole, Avroy A. Fanaroff, Barbara J. Stoll, Richard Ehrenkranz, Linda L. Wright. (2005). Prediction of Death for Extremely Low Birth Weight Neonates. PEDIATRICS, 116(6), 1367–1373[10.1542/peds.2004-2099]
  14. 14.Alfensi Faruk, Endro Setyo Cahyono. (2018). Prediction and Classification of Low Birth Weight Data Using Machine Learning Techniques. Indonesian Journal of Science and Technology, 3(1), 18[10.17509/ijost.v3i1.10799]
  15. 15.Najmus Sakib Borson, Md. Riftabin Kabir, Zaisha Zamal, Rashedur M. Rahman. (2020). Correlation analysis of demographic factors on low birth weight and prediction modeling using machine learning techniques. World Conference on Smart Trends in Systems, Security and Sustainability (WorldS4), 169–173[10.1109/worlds450073.2020.9210338]
  16. 16.Zakir Hussain, Malaya Dutta Borah. (2020). Birth weight prediction of new born baby with application of machine learning techniques on features of mother. Journal of Statistics and Management Systems, 23(6), 1079–1091[10.1080/09720510.2020.1814499]
  17. 17.Anita Desiani, Rifkie Primartha, Muhammad Arhami, Osvari Orsalan. (2019). Naive Bayes classifier for infant weight prediction of hypertension mother. Journal of Physics Conference Series, 1282(1), 012005[10.1088/1742-6596/1282/1/012005]
  18. 18.Chalachew Tiruneh, Daniel Teshome. (2021). Prediction of Birth Weight by Using Neonatal Anthropometric Parameters at Birth in Finote Selam Hospital, Ethiopia. Pediatric Health Medicine and Therapeutics, Volume 12, 259–267[10.2147/phmt.s309573]
  19. 19.Osama Salah El-Din Al-Madhoun, Afnan Omar Abu Hasira, Soha Ahmed Hegazy, Samy S. Abu-Naser. (2020). Low Birth Weight Prediction Using JNN. PhilPapers (PhilPapers Foundation), 4(11)[link]
  20. 20.Nishita Shettian, Hema Venkata Naga Sai Sudha Pavuluri. (2021). Correlation between maternal thyroid stimulating hormone and birth weight of neonate. International Journal of Reproduction Contraception Obstetrics and Gynecology, 11(1), 55[10.18203/2320-1770.ijrcog20215074]
  21. 21.Liu, Ziming, Wang, Yixuan, Vaidya, Sachin, Ruehle, Fabian, Halverson, James, Soljačić, Marin, Hou, Thomas Y., Tegmark, Max. (2024). KAN: Kolmogorov-Arnold Networks. arXiv (Cornell University)[10.48550/arxiv.2404.19756]
← All publications