Prediction of postoperative recovery in patients with acoustic neuroma using machine learning and SMOTE-ENN techniques

Math Biosci Eng. 2022 Jul 22;19(10):10407-10423. doi: 10.3934/mbe.2022487.

Abstract

Acoustic neuroma is a common benign tumor that is frequently associated with postoperative complications such as facial nerve dysfunction, which greatly affects the physical and mental health of patients. In this paper, clinical data of patients with acoustic neuroma treated with microsurgery by the same operator at Xiangya Hospital of Central South University from June 2018 to March 2020 are used as the study object. Machine learning and SMOTE-ENN techniques are used to accurately predict postoperative facial nerve function recovery, thus filling a gap in auxiliary diagnosis within the field of facial nerve treatment in acoustic neuroma. First, raw clinical data are processed and dependent variables are identified based on clinical context and data characteristics. Secondly, data balancing is corrected using the SMOTE-ENN technique. Finally, XGBoost is selected to construct a prediction model for patients' postoperative recovery, and is also compared with a total of four machine learning models, LR, SVM, CART, and RF. We find that XGBoost can most accurately predict the postoperative facial nerve function recovery, with a prediction accuracy of 90.0% and an AUC value of 0.90. CART, RF, and XGBoost can further select the more important preoperative indicators and provide therapeutic assistance to physicians, thereby improving the patient's postoperative recovery. The results show that machine learning and SMOTE-ENN techniques can handle complex clinical data and achieve accurate predictions.

Keywords: SMOTE-ENN; XGBoost; acoustic neuroma; machine learning; postoperative recovery.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Facial Nerve
  • Humans
  • Machine Learning
  • Neuroma, Acoustic*
  • Postoperative Complications
  • Postoperative Period