3 个月前

注意力增强的深度学习模型用于超声图像中的乳腺肿瘤分割

注意力增强的深度学习模型用于超声图像中的乳腺肿瘤分割

摘要

将人类领域知识融入乳腺肿瘤诊断面临挑战,因为形状、边界、曲率、灰度强度等常见医学先验信息在不同患者之间差异显著,难以统一应用。本文提出一种新方法,将视觉显著性引入深度学习模型,用于超声图像中的乳腺肿瘤分割。视觉显著性指图像中更可能吸引放射科医生视觉注意的区域所构成的图像映射。所提出的方法在U-Net架构中引入注意力模块,学习优先关注显著性水平较高的空间区域的特征表示。验证结果表明,相较于未引入显著性注意力层的模型,该方法在肿瘤分割任务中显著提升了精度。在包含510张图像的数据集上,该方法取得了90.5%的Dice相似系数。该显著性注意力模型具有潜力,可通过将任务特定知识融入深度学习架构,提升对其他器官医学图像处理的准确性与鲁棒性。

基准测试

基准方法指标
lesion-segmentation-on-bus-2017-dataset-bSalient Attention U-Net
Dice Score: 0.7341
tumor-segmentation-on-bus-2017-dataset-bSalient Attention U-Net
Dice Score: 0.7341

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
注意力增强的深度学习模型用于超声图像中的乳腺肿瘤分割 | 论文 | HyperAI超神经