arXiv:2606.22400cs.CVcs.LG2026-06被引 1

轻量CNN+迁移学习,三类癌症影像识别准确率超90%

Multi-cancer detection using a computationally efficient CNN with transfer learning

  • 设计轻量CNN并用迁移学习,仅20轮微调即达高精度
  • 脑/肺/肾癌分类准确率分别为90.85%、98.64%、99.92%
  • 每图每轮仅需0.014秒,适合低算力环境部署

本研究提出一种计算高效的卷积神经网络(CNN)架构,结合迁移学习实现多癌症检测。该轻量级CNN在保持高分类性能的同时降低计算复杂度,适用于资源受限环境。在包含脑部磁共振成像(MRI)及肺、肾计算机断层扫描(CT)的三个肿瘤数据集上进行评估,采用5折分层交叉验证,脑、肺、肾癌分类测试准确率分别为90.85 ± 2.22%、98.64 ± 2.43%和99.92 ± 0.08%。通过在一种癌症上预训练后,在其他癌症上微调,仅需20个额外训练轮次即可达到与从头训练相当的性能。微调过程仅更新CNN分类部分,使用NVIDIA GeForce GTX 960时每图像每轮耗时约0.014秒。对比实验表明,该模型优于Xception、VGG16、VGG19、MobileNetV2和DenseNet121等多个先进预训练模型。模型在具有不同形态特征的三类癌症上表现稳健,涵盖MRI与CT两种成像模态,验证了其跨任务、跨数据类型的泛化能力。结果表明,精简深度学习框架可在不牺牲准确性前提下加速癌症诊断,尤其适用于算力有限场景。

原文摘要 · Abstract (English)

This study introduces a computationally efficient convolutional neural network (CNN) architecture enhanced with transfer learning for multi-cancer detection using biomedical images. The proposed lightweight CNN model is designed to reduce computational complexity while maintaining high classification performance, making it suitable for deployment in resource-constrained environments. We evaluate this approach on three distinct tumor datasets comprising brain magnetic resonance imaging (MRI) and lung and kidney computed tomography (CT) scans. The model achieves test accuracy of 90.85 +- 2.22%, 98.64 +- 2.43% and 99.92 +- 0.08% for brain, lung, and kidney cancer classification, respectively, using 5-fold stratified cross-validation (CV). Transfer learning is employed by pretraining the model on one cancer type and fine-tuning it on the others, requiring only 20 additional epochs to achieve performance comparable to models trained from scratch. The fine-tuning process involves updating the classification part of the CNN and requires approximately 0.014 seconds per image per epoch using an NVIDIA GeForce GTX 960. Comparative evaluations show that the proposed model outperforms several state-of-the-art pretrained architectures, such as Xception, VGG16, VGG19, MobileNetV2 and DenseNet121. Overall, the model's effectiveness is evaluated across three types of cancer with distinct morphological characteristics, assessing its performance on both MRI and CT imaging modalities and demonstrating robust performance across diverse tasks and data types. These findings underscore the potential of streamlined deep learning (DL) frameworks in accelerating cancer diagnosis without sacrificing accuracy, especially in settings with limited computational resources.

癌症检测轻量模型迁移学习医学影像

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