Machine learning models identify prognostic factors in systemic lupus erythematosus patients with epstein-barr virus infection
摘要
To identify poor prognostic factors in Epstein-Barr virus (EBV)-positive systemic lupus erythematosus (SLE) using interpretable machine-learning (ML) models.
MethodWe retrospectively analysed 217 EBV-positive SLE in-patients (2018–2022). Clinical and laboratory variables were compared between favorable and poor prognosis groups. Seven ML algorithms—logistic regression, support vector machine, naïve Bayes, random forest (RF), gradient boosting machine (GBM), artificial neural network, and AdaBoost—were trained with a 70/30 train–test split. Recursive feature elimination with cross-validation selected the optimal predictor set. SHAP (Shapley additive explanations) illustrated feature importance.
ResultsSix clinical features (SLEDAI-2 K, lupus nephritis, splenomegaly, arthritis, rash, fever) and six laboratory indicators (haemoglobin, 24-h urine protein, serum uric acid, EBNA-IgG, AST, CD3 + T-cell percentage) constituted the final model inputs. The RF model performed best for clinical variables (AUC 0.71; F1 0.55); GBM performed best for laboratory variables (AUC 0.56; F1 0.30). SHAP confirmed SLEDAI-2 K, lupus nephritis, and haemoglobin as the most influential predictors.
ConclusionInterpretable ML models highlight disease activity, renal involvement, haematological status, and EBV serology as important model-selected features associated with poor prognosis in EBV-positive SLE. These findings provide exploratory insights into potential prognostic factors.