4 个月前

D-UNet:一种用于慢性脑卒中病灶分割的维度融合U形网络

D-UNet:一种用于慢性脑卒中病灶分割的维度融合U形网络

摘要

评估慢性中风引起的病灶位置和范围对于医学诊断、手术规划和预后至关重要。近年来,随着二维(2D)和三维(3D)卷积神经网络(CNN)的快速发展,编码器-解码器结构在医学图像分割领域展现出巨大潜力。然而,2D CNN忽略了医学图像的三维信息,而3D CNN则面临高计算资源需求的问题。本文提出了一种新的架构,称为维度融合UNet(D-UNet),该架构在编码阶段创新性地结合了2D和3D卷积。所提出的架构在分割性能上优于2D网络,同时相比3D网络显著减少了计算时间。此外,为了解决网络训练过程中正负样本之间的数据不平衡问题,我们提出了一种新的损失函数,称为增强混合损失函数(Enhance Mixing Loss, EML)。该函数引入了一个加权焦点系数,并结合了两种传统的损失函数。所提出的方法已在ATLAS数据集上进行了测试,并与三种最先进的方法进行了比较。结果表明,所提出的方法在Dice相似系数(DSC = 0.5349 ± 0.2763)和精度(precision = 0.6331 ± 0.295)方面均取得了最佳的质量性能。

代码仓库

SWKoreaBME/D-Unet_PyTorch
pytorch
GitHub 中提及

基准测试

基准方法指标
lesion-segmentation-on-anatomical-tracings-of-1D-UNet
Dice: 0.5349
Precision: 0.6331
Recall: 0.5243

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
D-UNet:一种用于慢性脑卒中病灶分割的维度融合U形网络 | 论文 | HyperAI超神经