轻量级网络R-Net实现99.37%结直肠癌检测准确率,兼顾效率与可解释性。
R-Net: A Reliable and Resource-Efficient CNN for Colorectal Cancer Detection with XAI Integration
- 设计轻量级CNN R-Net,仅用少量计算资源完成结直肠癌分类
- 在EBHI数据集上达99.37%准确率,优于MobileNet和ResNet50
- 集成SHAP/LIME/Grad-CAM,可视化模型决策依据,适合医疗AI研究者
当前先进卷积神经网络(CNN)因计算开销大、训练时间长、依赖大规模数据而受批评。为此,本文提出轻量级网络R-Net,仅基于Enteroscope Biopsy Histopathological Hematoxylin and Eosin Image Dataset(EBHI)实现结直肠癌(CRC)检测与分类。同时,在相同数据集上测试了六种SOTA CNN模型,包括基于多路径的DenseNet121、ResNet50,基于深度的InceptionV3,基于宽度的Xception,基于深度分离卷积的MobileNetV2,基于空间利用的VGG16,以及两种集成模型:多路径-深度-宽度组合(DenseNet121-InceptionV3-Xception)和多路径-深度-空间组合(ResNet18-InceptionV3-VGG16)。R-Net以99.37%准确率超越MobileNet(95.83%)和ResNet50(96.94%)。更重要的是,引入SHAP、LIME和Grad-CAM等可解释AI方法,可视化图像中影响R-Net判断的关键区域。本研究主要创新在于构建了一个高效可靠、资源消耗少却表现强劲的轻量级模型,拓展了对结直肠癌分类的认知,并揭示像素强度对正确与错误分类的影响。
原文摘要 · Abstract (English)
State-of-the-art (SOTA) Convolutional Neural Networks (CNNs) are criticized for their extensive computational power, long training times, and large datasets. To overcome this limitation, we propose a reasonable network (R-Net), a lightweight CNN only to detect and classify colorectal cancer (CRC) using the Enteroscope Biopsy Histopathological Hematoxylin and Eosin Image Dataset (EBHI). Furthermore, six SOTA CNNs, including Multipath-based CNNs (DenseNet121, ResNet50), Depth-based CNNs (InceptionV3), width-based multi-connection CNNs (Xception), depth-wise separable convolutions (MobileNetV2), spatial exploitation-based CNNs (VGG16), Transfer learning, and two ensemble models are also tested on the same dataset. The ensemble models are a multipath-depth-width combination (DenseNet121-InceptionV3-Xception) and a multipath-depth-spatial combination (ResNet18-InceptionV3-VGG16). However, the proposed R-Net lightweight achieved 99.37% accuracy, outperforming MobileNet (95.83%) and ResNet50 (96.94%). Most importantly, to understand the decision-making of R-Net, Explainable AI such as SHAP, LIME, and Grad-CAM are integrated to visualize which parts of the EBHI image contribute to the detection and classification process of R-Net. The main novelty of this research lies in building a reliable, lightweight CNN R-Net that requires fewer computing resources yet maintains strong prediction results. SOTA CNNs, transfer learning, and ensemble models also extend our knowledge on CRC classification and detection. XAI functionality and the impact of pixel intensity on correct and incorrect classification images are also some novelties in CRC detection and classification.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。