论文标题

基于机器学习的基于形状的新型损耗功能

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

论文作者

Karimzadeh, Reza, Fatemizadeh, Emad, Arabi, Hossein

论文摘要

自动化医学图像细分是帮助/加快临床实践中诊断和治疗程序的重要任务。深度卷积神经网络在准确和自动的精确分割方面表现出了有希望的表现。对于细分任务,这些方法通常依赖于最大程度地减少成本/损失功能,该功能旨在最大程度地提高专家划定的估计目标和地面掩码之间的重叠。基于重叠程度(即骰子指标)的简单损失函数将不考虑目标主体的基本形状和形态及其现实/自然变化;因此,将以横向分割结果的形式观察到次优的结果。从这个角度来看,已经进行了许多研究,以完善/后处理分割结果,并将初始猜测视为避免异常值和/或不切实际估计的先验知识。在这项研究中,提出了一种新颖的基于形状的成本函数,该功能鼓励/限制网络学习/捕获基本形状特征,以便对目标结构产生有效/现实的估计。为此,对矢量化训练数据集进行了主成分分析(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.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源