arXiv:2410.12584eess.IVcs.CV2024-10被引 6

用自监督与集成学习提升肺结节分类准确率

Self-DenseMobileNet: A Robust Framework for Lung Nodule Classification using Self-ONN and Stacking-based Meta-Classifier

  • 结合自监督特征提取与多模型堆叠,提升分类鲁棒性
  • 内部数据达99.28%准确率,外部测试仍保持89.40%
  • 通过注意力热力图增强模型决策可解释性,适合医疗影像分析

本研究提出一种新型鲁棒框架Self-DenseMobileNet,用于胸部X光片(CXRs)中肺结节与非结节的分类。通过先进图像标准化与增强技术优化输入质量,提升分类精度。将Self-DenseMobileNet的预测概率转换为表格数据,训练八种经典机器学习模型,选取前三名表现者通过堆叠算法构建元分类器,融合多方见解以实现更优性能。为提升结果可解释性,采用类别激活映射(CAM)可视化最优模型的决策过程。在内部验证数据上,该框架使用元-随机森林分类器达到99.28%准确率;在外部数据集上仍保持89.40%的高准确率,显著提升肺结节分类效果。

原文摘要 · Abstract (English)

In this study, we propose a novel and robust framework, Self-DenseMobileNet, designed to enhance the classification of nodules and non-nodules in chest radiographs (CXRs). Our approach integrates advanced image standardization and enhancement techniques to optimize the input quality, thereby improving classification accuracy. To enhance predictive accuracy and leverage the strengths of multiple models, the prediction probabilities from Self-DenseMobileNet were transformed into tabular data and used to train eight classical machine learning (ML) models; the top three performers were then combined via a stacking algorithm, creating a robust meta-classifier that integrates their collective insights for superior classification performance. To enhance the interpretability of our results, we employed class activation mapping (CAM) to visualize the decision-making process of the best-performing model. Our proposed framework demonstrated remarkable performance on internal validation data, achieving an accuracy of 99.28\% using a Meta-Random Forest Classifier. When tested on an external dataset, the framework maintained strong generalizability with an accuracy of 89.40\%. These results highlight a significant improvement in the classification of CXRs with lung nodules.

肺结节分类集成学习可解释性医学影像

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。