用ViT+预处理提升乳腺和卵巢癌分类效率,准确率超现有方法。
Efficient Breast and Ovarian Cancer Classification via ViT-Based Preprocessing and Transfer Learning
- 基于预训练ViT模型,结合图像标准化预处理流程。
- 在BreakHis和UBC-OCEAN数据集上达到更高分类准确率。
- 适合医学影像分析、深度学习辅助诊断研究者参考。
癌症是女性面临的重大健康挑战,尤其是乳腺癌和卵巢癌。早期发现可通过及时干预显著提高生存率。传统检测依赖病理学家手动分析乳腺钼靶、CT、超声等影像,耗时且需专业技能。本文提出一种基于视觉变换器(ViT)的新型癌症检测与分类方法。采用预训练的ViT-Base-Patch16-224模型,对公开的组织病理图像数据集进行微调,完成二分类与五分类任务。构建了将原始病理图像转换为标准PyTorch张量的预处理流程,适配ViT架构并提升性能。在两个基准数据集上评估:BreakHis用于二分类,UBC-OCEAN用于五分类,均未使用数据增强。结果表明,该模型在二分类中优于现有CNN、ViT及拓扑数据分析方法;在多分类任务中也显著超越近期拓扑方法。研究表明,结合视觉变压器与高效预处理的迁移学习在肿瘤诊断中具有显著优势。
原文摘要 · Abstract (English)
Cancer is one of the leading health challenges for women, specifically breast and ovarian cancer. Early detection can help improve the survival rate through timely intervention and treatment. Traditional methods of detecting cancer involve manually examining mammograms, CT scans, ultrasounds, and other imaging types. However, this makes the process labor-intensive and requires the expertise of trained pathologists. Hence, making it both time-consuming and resource-intensive. In this paper, we introduce a novel vision transformer (ViT)-based method for detecting and classifying breast and ovarian cancer. We use a pre-trained ViT-Base-Patch16-224 model, which is fine-tuned for both binary and multi-class classification tasks using publicly available histopathological image datasets. Further, we use a preprocessing pipeline that converts raw histophological images into standardized PyTorch tensors, which are compatible with the ViT architecture and also help improve the model performance. We evaluated the performance of our model on two benchmark datasets: the BreakHis dataset for binary classification and the UBC-OCEAN dataset for five-class classification without any data augmentation. Our model surpasses existing CNN, ViT, and topological data analysis-based approaches in binary classification. For multi-class classification, it is evaluated against recent topological methods and demonstrates superior performance. Our study highlights the effectiveness of Vision Transformer-based transfer learning combined with efficient preprocessing in oncological diagnostics.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。