arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2305.09660eess.IVcs.CV
收藏

使用迁移学习模型的骨肉瘤肿瘤检测

Osteosarcoma Tumor Detection using Transfer Learning Models

American International University Bangladesh(美国国际大学孟加拉分校)

Raisa Fairooz Meem, Khandaker Tabin Hasan

AI总结:

本文针对骨肉瘤肿瘤检测任务,比较了EfficientNetB7、InceptionResNetV2、NasNetLarge和ResNet50四种在ImageNet上预训练的迁移学习模型,发现InceptionResNetV2在准确率、精确率和召回率上均取得最佳表现。

摘要:

由于计算复杂度较低、准确率更高等优势,临床图像分析领域正越来越多地应用迁移学习模型。这些预训练模型无需从头开始训练,从而消除了对大型数据集的需求。迁移学习模型主要用于大脑、乳腺或肺部图像的分析,但其他领域如骨髓细胞检测或骨癌检测同样能从中受益,特别是考虑到这些任务缺乏可用的大型数据集。本文研究了几种迁移学习模型在骨肉瘤肿瘤检测中的性能。骨肉瘤是一种骨癌,主要发现于身体长骨的细胞中。数据集由H&E染色图像组成,分为4个类别:Viable Tumor(存活肿瘤)、Non-viable Tumor(非存活肿瘤)、Non-Tumor(非肿瘤)和Viable Non-viable(存活非存活)。数据集被随机划分为训练集和测试集,遵循80-20的比例。80%用于训练,20%用于测试。比较了4个模型:EfficientNetB7、InceptionResNetV2、NasNetLarge和ResNet50。所有这些模型均在ImageNet上进行了预训练。结果显示,InceptionResNetV2实现了最高的准确率(93.29%),其次是NasNetLarge(90.91%)、ResNet50(89.83%)和EfficientNetB7(62.77%)。它还在4个模型中拥有最高的精确率(0.8658)和召回率(0.8658)值。

英文摘要:

The field of clinical image analysis has been applying transfer learning models increasingly due to their less computational complexity, better accuracy etc. These are pre-trained models that don't require to be trained from scratch which eliminates the necessity of large datasets. Transfer learning models are mostly used for the analysis of brain, breast, or lung images but other sectors such as bone marrow cell detection or bone cancer detection can also benefit from using transfer learning models, especially considering the lack of available large datasets for these tasks. This paper studies the performance of several transfer learning models for osteosarcoma tumour detection. Osteosarcoma is a type of bone cancer mostly found in the cells of the long bones of the body. The dataset consists of H&E stained images divided into 4 categories- Viable Tumor, Non-viable Tumor, Non-Tumor and Viable Non-viable. Both datasets were randomly divided into train and test sets following an 80-20 ratio. 80% was used for training and 20\% for test. 4 models are considered for comparison- EfficientNetB7, InceptionResNetV2, NasNetLarge and ResNet50. All these models are pre-trained on ImageNet. According to the result, InceptionResNetV2 achieved the highest accuracy (93.29%), followed by NasNetLarge (90.91%), ResNet50 (89.83%) and EfficientNetB7 (62.77%). It also had the highest precision (0.8658) and recall (0.8658) values among the 4 models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑