Risk Prediction Based on Clinicopathologic Features in Korean Melanoma Patients by Machine Learning
Taeho Yuh1, Hyunwook Kim1,2, Eun Sil Baek3
1Department of Internal Medicine, Yonsei University College of Medicine, Seoul, Korea.
Purpose:
The prognosis of malignant melanoma differs across ethnic groups, partly due to variations in histologic subtypes. Accurate risk stratification at diagnosis is crucial but remains challenging in Korea due to its low incidence. This study aimed to develop and evaluate machine learning (ML) models for predicting 2-year disease free survival (DFS) using clinicopathologic features available at the time of diagnosis.
Materials And Methods:
A retrospective cohort of 1,657 Korean melanoma patients (2006-2023) was analyzed. Clinical and pathological variables-excluding TNM staging and treatment data-were used to train ML models, including decision tree, random forest, boosting methods, XGBoost, and deep neural networks (DNN). Performance was evaluated using accuracy, precision, recall, F1 score, and AUC.
Results:
The study included 1,657 patients with 12 clinicopathologic variables, showing median DFS of 63.9 months (95% CI 52.93-85.33) and a 2-year survival rate of 67.68% (95% CI: 65.16-70.00). Among the evaluated algorithms, XGBoost demonstrated the highest F1 score of 0.761. Feature importance analysis identified primary tumor site, mitotic rate, and body mass index (BMI) as the most influential predictors of survival. Kaplan-Meier curves and Cox proportional hazards analysis confirmed the clinical relevance of these features.
Conclusion:
The XGBoost model showed the best performance for predicting 2-year DFS probability in Korean melanoma patients. Primary cancer site, mitotic rate, and BMI were identified as significant features influencing survival.
