A Class Imbalance Monitoring Model for Fetal Heart Contractions Based on Gradient Boosting Decision Tree Ensemble Learning

Aiming at the imbalance and cost-sensitive problem of sample categories in actual fetal monitoring, as well as actual needs, we proposed a category imbalance fetal contraction monitoring model based on GBDT (Gradient Boosting Decision Tree) combined learning. Subsets with balanced category were generated by random under-sampling and applied to train several GBDT base classifiers using the method of feature selection. We integrated the base classifiers by the simple average method and calculated the final prediction probability. In this study, AUC and cost-sensitive error rate were used as evaluation indicators to compare with the commonly used single learning models such as Decision Tree, Logistic Regression and combined learning models like Random Forest to verify the effectiveness of the model.