3 个月前

利用图像块大小与分辨率进行多尺度深度学习在乳腺X线图像分类中的应用

利用图像块大小与分辨率进行多尺度深度学习在乳腺X线图像分类中的应用

摘要

近年来,深度学习(Deep Learning, DL)的进展重新激发了基于深度学习的计算机辅助检测或诊断(CAD)系统在乳腺癌筛查领域的研究兴趣。基于图像块(patch-based)的方法是当前二维乳腺X线摄影图像分类的主要先进技术之一,但其性能在很大程度上受限于图像块尺寸的选择,因为并不存在一种适用于所有病灶尺寸的最优块大小。此外,输入图像分辨率对分类性能的影响尚未得到充分理解。本文系统研究了图像块尺寸与输入图像分辨率对二维乳腺X线图像分类器性能的影响。为充分利用不同块尺寸与分辨率的优势,本文提出了一种多块尺寸分类器(multi patch-size classifier)和一种多分辨率分类器(multi-resolution classifier)。这两种新架构通过融合多种图像块尺寸与不同输入分辨率,实现多尺度分类。在公开的CBIS-DDSM数据集上,模型的AUC提升了3%;在内部数据集上,AUC提升达5%。相较于基准的单块尺寸、单分辨率分类器,本文提出的多尺度分类器在两个数据集上分别达到了0.809和0.722的AUC值。

基准测试

基准方法指标
cancer-no-cancer-per-image-classification-onPatch-based DenseNet-121
AUC: 0.784
cancer-no-cancer-per-image-classification-onMulti-patch size DenseNet-121
AUC: 0.809
cancer-no-cancer-per-image-classification-onFeature Pyramid Network DenseNet-121
AUC: 0.788
cancer-no-cancer-per-image-classification-onMulti-resolution DenseNet-121
AUC: 0.789

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
利用图像块大小与分辨率进行多尺度深度学习在乳腺X线图像分类中的应用 | 论文 | HyperAI超神经