用咳嗽声自动分析辅助肺癌早期筛查,准确率达83%
Automatic Cough Analysis for Non-Small Cell Lung Cancer Detection
- 通过深度学习分析咳嗽音频,识别肺癌患者与健康人
- 卷积神经网络在测试集上达到83%准确率,表现最优
- 模型解释性强且对不同年龄组公平性略低于性别组
早期发现非小细胞肺癌(NSCLC)对改善患者预后至关重要,亟需新型筛查方法。本研究探索了自动咳嗽分析作为区分NSCLC患者与健康对照的初步筛查工具。前瞻性采集227名受试者的咳嗽音频,采用支持向量机(SVM)、XGBoost、卷积神经网络(CNN)及基于VGG16的迁移学习等机器学习方法进行分析。为提升模型可解释性,引入SHAP方法。通过测试集上的等效机会差异评估模型在不同年龄组(≤58岁与>58岁)和性别间的公平性。结果表明,CNN在测试集上表现最佳,准确率为0.83;SVM在验证集和测试集上分别达到0.76和0.78,适合计算资源有限场景。使用SHAP解释SVM增强了临床可信度。公平性分析显示,年龄组间差异(0.15)略高于性别组(0.09)。因此,为增强结论可靠性,仍需更大、更多样、无偏倚的数据集,尤其包含高风险人群及早期患者。
原文摘要 · Abstract (English)
Early detection of non-small cell lung cancer (NSCLC) is critical for improving patient outcomes, and novel approaches are needed to facilitate early diagnosis. In this study, we explore the use of automatic cough analysis as a pre-screening tool for distinguishing between NSCLC patients and healthy controls. Cough audio recordings were prospectively acquired from a total of 227 subjects, divided into NSCLC patients and healthy controls. The recordings were analyzed using machine learning techniques, such as support vector machine (SVM) and XGBoost, as well as deep learning approaches, specifically convolutional neural networks (CNN) and transfer learning with VGG16. To enhance the interpretability of the machine learning model, we utilized Shapley Additive Explanations (SHAP). The fairness of the models across demographic groups was assessed by comparing the performance of the best model across different age groups (less than or equal to 58y and higher than 58y) and gender using the equalized odds difference on the test set. The results demonstrate that CNN achieves the best performance, with an accuracy of 0.83 on the test set. Nevertheless, SVM achieves slightly lower performances (accuracy of 0.76 in validation and 0.78 in the test set), making it suitable in contexts with low computational power. The use of SHAP for SVM interpretation further enhances model transparency, making it more trustworthy for clinical applications. Fairness analysis shows slightly higher disparity across age (0.15) than gender (0.09) on the test set. Therefore, to strengthen our findings' reliability, a larger, more diverse, and unbiased dataset is needed -- particularly including individuals at risk of NSCLC and those in early disease stages.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。