Dec, 2023

利用 MobileNetV2 进行增强的乳腺癌瘤分类:对图像强度、错误修正和基于 Streamlit 的实时部署的详细探索

TL;DR基于 Google 的 MobileNetV2 模型,利用来自埃及开罗的 Baheya 医院的数据集,该研究引入了一种复杂的转移学习模型,用于对乳腺癌肿瘤进行分类,分为正常、良性和恶性三类,采用 1576 张超声图像(265 张正常、891 张良性、420 张恶性)。该模型实现了 0.82 的准确率,0.83 的精确率,0.81 的召回率,0.94 的 ROC-AUC,0.88 的 PR-AUC 和 0.74 的 MCC。它研究了图像强度分布和错误分类,为未来的应用提供了改进。通过解决数据集不平衡问题,该研究确保了一个具有普适性的模型。此外,本论文探讨了基于 Streamlit 的实时肿瘤分类部署,展示了 MobileNetV2 在医学成像中的适用性,并为肿瘤诊断领域的未来研究设定了一个基准。