基于Swin Transformer的模型实现多癌种病理图像高精度分类
DSVTLA: Deep Swin Vision Transformer-Based Transfer Learning Architecture for Multi-Type Cancer Histopathological Cancer Image Classification
- 融合Swin Transformer与ResNet50特征,捕捉长距离依赖与局部形态
- 在肺结肠癌、白血病等数据集上达100%准确率,乳腺癌99.23%
- 适合临床病理诊断辅助系统开发,兼具准确与可解释性
本研究提出一种基于深度Swin-Vision Transformer的迁移学习架构,用于稳健的多癌种病理图像分类。该框架结合分层Swin Transformer与基于ResNet50的卷积特征提取,使模型能够同时捕获病理图像中的长程上下文依赖关系和细粒度局部形态模式。为验证所提架构的效率,在包含乳腺癌、口腔癌、肺癌与结肠癌、肾癌及急性淋巴细胞白血病(ALL)的综合性多癌种数据集上进行了广泛实验,分析了原始图像与分割后图像,以评估模型在异质临床成像条件下的鲁棒性。对比了多种先进CNN与迁移学习模型,包括DenseNet121、DenseNet201、InceptionV3、ResNet50、EfficientNetB3、多个ViT变体及Swin Transformer模型。所有模型均采用统一训练与验证流程,包含平衡数据预处理、迁移学习与微调策略。实验结果表明,所提架构持续表现优异,肺癌-结肠癌、分割后白血病数据集达到100%测试准确率,乳腺癌分类最高达99.23%。模型还实现了近乎完美的精确率、F1分数与召回率,表明在各类癌症间具有高度稳定的表现。总体而言,该模型建立了一个高精度、可解释且鲁棒的多癌种分类体系,为未来研究提供强有力基准,并为设计可靠的AI辅助病理诊断与临床决策系统提供统一评估参考。
原文摘要 · Abstract (English)
In this study, we proposed a deep Swin-Vision Transformer-based transfer learning architecture for robust multi-cancer histopathological image classification. The proposed framework integrates a hierarchical Swin Transformer with ResNet50-based convolution features extraction, enabling the model to capture both long-range contextual dependencies and fine-grained local morphological patterns within histopathological images. To validate the efficiency of the proposed architecture, an extensive experiment was executed on a comprehensive multi-cancer dataset including Breast Cancer, Oral Cancer, Lung and Colon Cancer, Kidney Cancer, and Acute Lymphocytic Leukemia (ALL), including both original and segmented images were analyzed to assess model robustness across heterogeneous clinical imaging conditions. Our approach is benchmarked alongside several state-of-the-art CNN and transfer models, including DenseNet121, DenseNet201, InceptionV3, ResNet50, EfficientNetB3, multiple ViT variants, and Swin Transformer models. However, all models were trained and validated using a unified pipeline, incorporating balanced data preprocessing, transfer learning, and fine-tuning strategies. The experimental results demonstrated that our proposed architecture consistently gained superior performance, reaching 100% test accuracy for lung-colon cancer, segmented leukemia datasets, and up to 99.23% accuracy for breast cancer classification. The model also achieved near-perfect precision, f1 score, and recall, indicating highly stable scores across divers cancer types. Overall, the proposed model establishes a highly accurate, interpretable, and also robust multi-cancer classification system, demonstrating strong benchmark for future research and provides a unified comparative assessment useful for designing reliable AI-assisted histopathological diagnosis and clinical decision-making.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。