轻量模型+可解释性,提升肺癌病理图像诊断准确率与可信度
DCSNet: A Lightweight Knowledge Distillation-Based Model with Explainable AI for Lung Cancer Diagnosis from Histopathological Images
- 用教师模型知识蒸馏训练轻量学生模型,降低计算开销
- 在多个数据集上实现95%以上准确率,媲美大模型性能
- 结合可解释AI技术,让诊断过程透明可追溯,适合临床应用
肺癌是全球癌症致死的主要原因,早期检测与准确诊断对提高生存率至关重要。深度学习,尤其是卷积神经网络(CNN),通过识别早期肺癌的细微模式,已显著推动医学图像分析发展,但其广泛应用面临挑战:模型计算成本高,资源受限环境难以部署;且缺乏透明性,影响医疗领域信任度。知识蒸馏通过将大型复杂模型(教师)的知识迁移至小型轻量模型(学生),有效缓解上述问题。本文提出一种基于知识蒸馏的肺癌检测方法,并融合可解释AI(XAI)技术以增强模型透明性。评估了八种CNN模型(包括ResNet50、EfficientNetB0、EfficientNetB3和VGG16)作为教师模型,构建并训练了一个轻量级学生模型——蒸馏定制学生网络(DCSNet),以ResNet50为教师。该方法不仅在资源受限环境中保持高诊断性能,还解决了透明性问题,促进AI诊断工具在医疗场景中的实际应用。
原文摘要 · Abstract (English)
Lung cancer is a leading cause of cancer-related deaths globally, where early detection and accurate diagnosis are critical for improving survival rates. While deep learning, particularly convolutional neural networks (CNNs), has revolutionized medical image analysis by detecting subtle patterns indicative of early-stage lung cancer, its adoption faces challenges. These models are often computationally expensive and require significant resources, making them unsuitable for resource constrained environments. Additionally, their lack of transparency hinders trust and broader adoption in sensitive fields like healthcare. Knowledge distillation addresses these challenges by transferring knowledge from large, complex models (teachers) to smaller, lightweight models (students). We propose a knowledge distillation-based approach for lung cancer detection, incorporating explainable AI (XAI) techniques to enhance model transparency. Eight CNNs, including ResNet50, EfficientNetB0, EfficientNetB3, and VGG16, are evaluated as teacher models. We developed and trained a lightweight student model, Distilled Custom Student Network (DCSNet) using ResNet50 as the teacher. This approach not only ensures high diagnostic performance in resource-constrained settings but also addresses transparency concerns, facilitating the adoption of AI-driven diagnostic tools in healthcare.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。