Aug, 2023

利用轻量级注意力机制增强乳腺癌分类的迁移 ResNet

TL;DR深度学习模型已经通过学习原始像素数据中的复杂特征层次结构,革新了图像分类。本文介绍了一种基于 ResNet 模型的图像分类方法,并引入了一个轻量级的注意力机制框架来提高性能。该框架优化了特征表示,增强了分类能力,并提高了特征的判别性。我们在 Breakhis 数据集上验证了该算法的有效性,在许多方面显示出其优越性。我们的方法不仅在传统模型方面具有优势,还在当代视觉转换器等最先进方法上显示出优势。在精确度、准确度、召回率、F1 分数和 G-means 等指标方面实现了显著的改进,同时在收敛时间方面表现良好。这些结果加强了算法的性能,并巩固了其在实际图像分类任务中的应用前景。