arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2203.03336physics.med-phcs.CVeess.IV
收藏

一种新的基于形状的损失函数用于医学成像中基于机器学习的精液器官分割

A novel shape-based loss function for machine learning-based seminal organ segmentation in medical imaging

Sharif University of Technology(谢里夫理工大学);Geneva University Hospital(日内瓦大学医院)

Reza Karimzadeh, Emad Fatemizadeh, Hossein Arabi

摘要:

自动医学图像分割是临床实践中辅助/加速诊断和治疗过程的重要任务。深度卷积神经网络在准确和自动的精液分割方面表现出了良好的性能。对于分割任务,这些方法通常依赖于最小化成本/损失函数,该成本/损失函数被设计为最大化估计目标与专家描绘的地面实况掩模之间的重叠。基于重叠程度的简单损失函数(即,Dice度量)将不考虑目标对象的潜在形状和形态以及其现实/自然变化;因此,将以体素岛、孔和不现实的形状或变形的形式观察到次优分割结果。有鉴于此,已经进行了许多研究来细化/后处理分割结果,并将初始猜测视为先验知识,以避免离群值和/或不切实际的估计。在这项研究中,提出了一种新的基于形状的成本函数,它鼓励/约束网络学习/捕获潜在的形状特征,以生成目标结构的有效/现实的估计。为此,对向量化训练数据集进行主成分分析(PCA),以提取目标受试者的特征值和特征向量。关键思想是使用重建权重来区分有效结果与离群值/错误估计。

英文摘要:

Automated medical image segmentation is an essential task to aid/speed up diagnosis and treatment procedures in clinical practices. Deep convolutional neural networks have exhibited promising performance in accurate and automatic seminal segmentation. For segmentation tasks, these methods normally rely on minimizing a cost/loss function that is designed to maximize the overlap between the estimated target and the ground-truth mask delineated by the experts. A simple loss function based on the degrees of overlap (i.e., Dice metric) would not take into account the underlying shape and morphology of the target subject, as well as its realistic/natural variations; therefore, suboptimal segmentation results would be observed in the form of islands of voxels, holes, and unrealistic shapes or deformations. In this light, many studies have been conducted to refine/post-process the segmentation outcome and consider an initial guess as prior knowledge to avoid outliers and/or unrealistic estimations. In this study, a novel shape-based cost function is proposed which encourages/constrains the network to learn/capture the underlying shape features in order to generate a valid/realistic estimation of the target structure. To this end, the Principal Component Analysis (PCA) was performed on a vectorized training dataset to extract eigenvalues and eigenvectors of the target subjects. The key idea was to use the reconstruction weights to discriminate valid outcomes from outliers/erroneous estimations.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑