A Novel Active Learning Technique for Fetal Health Classification Based on XGBoost Classifier
Ensuring safe pregnancy and reducing maternal and infant mortality rates require early prediction of fetal health. The application of machine learning algorithms in monitoring fetal health helps to improve the chances of timely intervention and better outcomes in the event of any possible health iss...
Saved in:
| Main Authors: | , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
IEEE
2025-01-01
|
| Series: | IEEE Access |
| Subjects: | |
| Online Access: | https://ieeexplore.ieee.org/document/10833651/ |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1832592836474699776 |
|---|---|
| author | Kaushal Bhardwaj Niyati Goyal Bhavika Mittal Vandna Sharma Shiv Naresh Shivhare |
| author_facet | Kaushal Bhardwaj Niyati Goyal Bhavika Mittal Vandna Sharma Shiv Naresh Shivhare |
| author_sort | Kaushal Bhardwaj |
| collection | DOAJ |
| description | Ensuring safe pregnancy and reducing maternal and infant mortality rates require early prediction of fetal health. The application of machine learning algorithms in monitoring fetal health helps to improve the chances of timely intervention and better outcomes in the event of any possible health issues in fetuses. Existing studies offered to help this issue, typically by training models using a significant portion of the dataset, ranging mainly above 70%. The only existing active learning method in this field employs around 41% training samples to achieve 98% accuracy. This work presents a novel active learning technique to identify the most informative data samples to train a model, leading to high accuracy with a limited number of training samples. It employs a novel query function built upon uncertainty and diversity criteria which are derived based on properties of XGBoost classifier and distance from each other. For deriving uncertainty criterion the soft probabilities obtained for the unlabeled samples are used, while the distance among the uncertain samples in feature space is utilized for deriving diversity criterion. The proposed approach shows superior performance compared to all state-of-the-art methods. Through analysis and experimentation, the proposed solution achieves an accuracy greater than 99% using less than 20% of the dataset for training. This shows its efficacy and potential in the monitoring of fetal health. The code and dataset are available on the GitHub repository <uri>https://github.com/niyg7/fetal-health-dataset</uri>. |
| format | Article |
| id | doaj-art-11a2268014cf46b4a267e1f98fef076d |
| institution | Kabale University |
| issn | 2169-3536 |
| language | English |
| publishDate | 2025-01-01 |
| publisher | IEEE |
| record_format | Article |
| series | IEEE Access |
| spelling | doaj-art-11a2268014cf46b4a267e1f98fef076d2025-01-21T00:02:09ZengIEEEIEEE Access2169-35362025-01-01139485949710.1109/ACCESS.2025.352715110833651A Novel Active Learning Technique for Fetal Health Classification Based on XGBoost ClassifierKaushal Bhardwaj0https://orcid.org/0000-0003-0484-5206Niyati Goyal1https://orcid.org/0009-0001-9417-8584Bhavika Mittal2Vandna Sharma3https://orcid.org/0009-0005-2501-4745Shiv Naresh Shivhare4https://orcid.org/0000-0002-6306-1113School of Computer Science Engineering and Technology, Bennett University, Greater Noida, Uttar Pradesh, IndiaSchool of Computer Science Engineering and Technology, Bennett University, Greater Noida, Uttar Pradesh, IndiaSchool of Computer Science Engineering and Technology, Bennett University, Greater Noida, Uttar Pradesh, IndiaSchool of Computer Science Engineering and Technology, Bennett University, Greater Noida, Uttar Pradesh, IndiaSchool of Computer Science Engineering and Technology, Bennett University, Greater Noida, Uttar Pradesh, IndiaEnsuring safe pregnancy and reducing maternal and infant mortality rates require early prediction of fetal health. The application of machine learning algorithms in monitoring fetal health helps to improve the chances of timely intervention and better outcomes in the event of any possible health issues in fetuses. Existing studies offered to help this issue, typically by training models using a significant portion of the dataset, ranging mainly above 70%. The only existing active learning method in this field employs around 41% training samples to achieve 98% accuracy. This work presents a novel active learning technique to identify the most informative data samples to train a model, leading to high accuracy with a limited number of training samples. It employs a novel query function built upon uncertainty and diversity criteria which are derived based on properties of XGBoost classifier and distance from each other. For deriving uncertainty criterion the soft probabilities obtained for the unlabeled samples are used, while the distance among the uncertain samples in feature space is utilized for deriving diversity criterion. The proposed approach shows superior performance compared to all state-of-the-art methods. Through analysis and experimentation, the proposed solution achieves an accuracy greater than 99% using less than 20% of the dataset for training. This shows its efficacy and potential in the monitoring of fetal health. The code and dataset are available on the GitHub repository <uri>https://github.com/niyg7/fetal-health-dataset</uri>.https://ieeexplore.ieee.org/document/10833651/Fetal healthactive learningXGBoostquery functionuncertaintydiversity |
| spellingShingle | Kaushal Bhardwaj Niyati Goyal Bhavika Mittal Vandna Sharma Shiv Naresh Shivhare A Novel Active Learning Technique for Fetal Health Classification Based on XGBoost Classifier IEEE Access Fetal health active learning XGBoost query function uncertainty diversity |
| title | A Novel Active Learning Technique for Fetal Health Classification Based on XGBoost Classifier |
| title_full | A Novel Active Learning Technique for Fetal Health Classification Based on XGBoost Classifier |
| title_fullStr | A Novel Active Learning Technique for Fetal Health Classification Based on XGBoost Classifier |
| title_full_unstemmed | A Novel Active Learning Technique for Fetal Health Classification Based on XGBoost Classifier |
| title_short | A Novel Active Learning Technique for Fetal Health Classification Based on XGBoost Classifier |
| title_sort | novel active learning technique for fetal health classification based on xgboost classifier |
| topic | Fetal health active learning XGBoost query function uncertainty diversity |
| url | https://ieeexplore.ieee.org/document/10833651/ |
| work_keys_str_mv | AT kaushalbhardwaj anovelactivelearningtechniqueforfetalhealthclassificationbasedonxgboostclassifier AT niyatigoyal anovelactivelearningtechniqueforfetalhealthclassificationbasedonxgboostclassifier AT bhavikamittal anovelactivelearningtechniqueforfetalhealthclassificationbasedonxgboostclassifier AT vandnasharma anovelactivelearningtechniqueforfetalhealthclassificationbasedonxgboostclassifier AT shivnareshshivhare anovelactivelearningtechniqueforfetalhealthclassificationbasedonxgboostclassifier AT kaushalbhardwaj novelactivelearningtechniqueforfetalhealthclassificationbasedonxgboostclassifier AT niyatigoyal novelactivelearningtechniqueforfetalhealthclassificationbasedonxgboostclassifier AT bhavikamittal novelactivelearningtechniqueforfetalhealthclassificationbasedonxgboostclassifier AT vandnasharma novelactivelearningtechniqueforfetalhealthclassificationbasedonxgboostclassifier AT shivnareshshivhare novelactivelearningtechniqueforfetalhealthclassificationbasedonxgboostclassifier |