论文标题

基于弹性的相互作用损失函数用于医学图像分割

An Elastic Interaction-Based Loss Function for Medical Image Segmentation

论文作者

Lan, Yuan, Xiang, Yang, Zhang, Luchan

论文摘要

深度学习技术显示了它们在医学图像细分方面的成功,因为它们易于操纵和强大的数据集。深度分割任务中常用的损失函数是像素损失函数。这导致了这些模型的瓶颈,以实现生物医学图像中复杂结构的高精度。例如,在视网膜图像中预测的小血管通常在像素损失的监督下断开甚至错过。本文通过引入基于远程弹性相互作用的培训策略来解决此问题。在此策略中,卷积神经网络(CNN)在预测区域和实际物体的边界之间的弹性相互作用能量的指导下学习了目标区域。在拟议损失的监督下,预测区域的边界被物体边界强烈吸引,并倾向于保持联系。实验结果表明,与常用的像素损失函数(横向熵和骰子损失)以及在三个视网膜血管分割数据集,驱动器,凝视和Chascasb1上,我们的方法能够实现相当大的改进。

Deep learning techniques have shown their success in medical image segmentation since they are easy to manipulate and robust to various types of datasets. The commonly used loss functions in the deep segmentation task are pixel-wise loss functions. This results in a bottleneck for these models to achieve high precision for complicated structures in biomedical images. For example, the predicted small blood vessels in retinal images are often disconnected or even missed under the supervision of the pixel-wise losses. This paper addresses this problem by introducing a long-range elastic interaction-based training strategy. In this strategy, convolutional neural network (CNN) learns the target region under the guidance of the elastic interaction energy between the boundary of the predicted region and that of the actual object. Under the supervision of the proposed loss, the boundary of the predicted region is attracted strongly by the object boundary and tends to stay connected. Experimental results show that our method is able to achieve considerable improvements compared to commonly used pixel-wise loss functions (cross entropy and dice Loss) and other recent loss functions on three retinal vessel segmentation datasets, DRIVE, STARE and CHASEDB1.

扫码加入交流群

加入微信交流群

微信交流群二维码

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