用数据增强提升癌症患者症状预测准确率
An Oversampling-enhanced Multi-class Imbalanced Classification Framework for Patient Health Status Prediction Using Patient-reported Outcomes
- 通过插值法扩充少数类样本,缓解症状数据不平衡问题
- 随机森林和XGBoost在三类癌症数据上加权AUC表现最佳
- 适合临床决策支持系统开发人员参考使用
放射治疗期间患者自报症状(PROs)数据对预测毒性反应至关重要。然而,医院采集的原始PRO数据存在项目缺失和症状类别不均衡等挑战。本研究针对头颈、前列腺和乳腺癌三种常见癌症类型,利用六种先进机器学习分类器——随机森林(RF)、XGBoost、梯度提升(GB)、支持向量机(SVM)、带装袋的多层感知机(MLP-Bagging)和逻辑回归(LR)——解决多类别不平衡分类问题。为应对类别不平衡,采用过采样策略,通过类内邻近样本插值方式扩充训练集,增强少数类样本而不改变原有分布。实验结果表明,在多个PRO数据集上,RF与XGB方法在区分轻度、中度和重度症状方面表现出稳健的泛化性能,加权AUC与混淆矩阵分析验证了其有效性,展现了在临床决策支持中的应用潜力。
原文摘要 · Abstract (English)
Patient-reported outcomes (PROs) directly collected from cancer patients being treated with radiation therapy play a vital role in assisting clinicians in counseling patients regarding likely toxicities. Precise prediction and evaluation of symptoms or health status associated with PROs are fundamental to enhancing decision-making and planning for the required services and support as patients transition into survivorship. However, the raw PRO data collected from hospitals exhibits some intrinsic challenges such as incomplete item reports and imbalance patient toxicities. To the end, in this study, we explore various machine learning techniques to predict patient outcomes related to health status such as pain levels and sleep discomfort using PRO datasets from a cancer photon/proton therapy center. Specifically, we deploy six advanced machine learning classifiers -- Random Forest (RF), XGBoost, Gradient Boosting (GB), Support Vector Machine (SVM), Multi-Layer Perceptron with Bagging (MLP-Bagging), and Logistic Regression (LR) -- to tackle a multi-class imbalance classification problem across three prevalent cancer types: head and neck, prostate, and breast cancers. To address the class imbalance issue, we employ an oversampling strategy, adjusting the training set sample sizes through interpolations of in-class neighboring samples, thereby augmenting minority classes without deviating from the original skewed class distribution. Our experimental findings across multiple PRO datasets indicate that the RF and XGB methods achieve robust generalization performance, evidenced by weighted AUC and detailed confusion matrices, in categorizing outcomes as mild, intermediate, and severe post-radiation therapy. These results underscore the models' effectiveness and potential utility in clinical settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。