Using Pupil Diameter for Psychological Resilience Assessment in Medical Students Based on SVM and SHAP Model

IEEE J Biomed Health Inform. 2024 Apr 22:PP. doi: 10.1109/JBHI.2024.3390390. Online ahead of print.

Abstract

Effectively assessing psychological resilience for medical students is vital for identifying at-risk individuals and developing tailored interventions. At present, few studies have combined physiological indexes of the human body and machine learning for psychological resilience assessment. This study presents a novel approach that employs pupil diameter features and machine learning to predict psychological resilience risk objectively. Firstly, we designed a stimulus paradigm (via auditory and visual stimuli) and collected pupil diameter data from participants using eye-tracking technology. Secondly, the pupil data was preprocessed, including linear interpolation, blink detection, and subtractive baseline correction. Thirdly, statistical metrics were extracted and optimal feature subsets were obtained by Recursive Feature Elimination with Cross-Validation (RFECV). Subsequently, the classification models, including Logistic Regression (LR), Random Forest (RF), Support Vector Machine (SVM), and eXtreme Gradient Boosting (XGBoost), were trained. The experimental results show that the SVM model has the best performance, and its balance accuracy, recall, and AUC reach 0.906, 0.89, and 0.932, respectively. Finally, we leveraged the Shapley additive explanation (SHAP) model for interpretability analysis. It revealed auditory stimuli have a more significant effect than visual stimuli in psychological resilience assessment. These findings suggested that pupil diameter could be a vital metric for assessing psychological resilience.