论文标题

基于先前的损失对细分性能的影响:基准测试

Effect of Prior-based Losses on Segmentation Performance: A Benchmark

论文作者

Jurdi, Rosana El, Petitjean, Caroline, Cheplygina, Veronika, Honeine, Paul, Abdallah, Fahed

论文摘要

如今,深层卷积神经网络(CNN)已在各种成像方式和任务上证明了医学图像分割的最新性能。尽管取得了早期成功,但分割网络仍可能会产生解剖上异常的分割,孔或不准确的对象边界附近。为了实施解剖学的合理性,最近的研究重点是将诸如对象形状或边界等先验知识纳入损失函数中的约束。先前的集成可以是低级的,指从地面实际分段中提取的重新反复表示,或代表外部医学信息(例如器官的形状或大小)的高级别。在过去的几年中,基于先前的损失对研究领域表现出了不断上升的兴趣,因为它们允许在仍然是建筑 - 敏锐的同时融合专家知识。但是,鉴于在不同的医学成像挑战和任务上有多种基于先前的损失,因此很难确定哪种损失最适合哪个数据集。在本文中,我们建立了最近基于医疗图像分割的基于先前的基于先前的损失的基准。主要目的是提供直觉,以选择给定特定任务或数据集选择哪些损失。为此,选择了四个低级和高级先前的损失。从多种医疗图像分割挑战中,包括十项全能,小岛和WMH挑战(WMH挑战),在8个不同的数据集中验证了所考虑的损失。结果表明,尽管低水平的基于先前的损失可以保证无论数据集特征如何,但基于骰子损失基线的性能会提高,但根据数据特征,高级基于先验的损失可以提高解剖学的合理性。

Today, deep convolutional neural networks (CNNs) have demonstrated state-of-the-art performance for medical image segmentation, on various imaging modalities and tasks. Despite early success, segmentation networks may still generate anatomically aberrant segmentations, with holes or inaccuracies near the object boundaries. To enforce anatomical plausibility, recent research studies have focused on incorporating prior knowledge such as object shape or boundary, as constraints in the loss function. Prior integrated could be low-level referring to reformulated representations extracted from the ground-truth segmentations, or high-level representing external medical information such as the organ's shape or size. Over the past few years, prior-based losses exhibited a rising interest in the research field since they allow integration of expert knowledge while still being architecture-agnostic. However, given the diversity of prior-based losses on different medical imaging challenges and tasks, it has become hard to identify what loss works best for which dataset. In this paper, we establish a benchmark of recent prior-based losses for medical image segmentation. The main objective is to provide intuition onto which losses to choose given a particular task or dataset. To this end, four low-level and high-level prior-based losses are selected. The considered losses are validated on 8 different datasets from a variety of medical image segmentation challenges including the Decathlon, the ISLES and the WMH challenge. Results show that whereas low-level prior-based losses can guarantee an increase in performance over the Dice loss baseline regardless of the dataset characteristics, high-level prior-based losses can increase anatomical plausibility as per data characteristics.

扫码加入交流群

加入微信交流群

微信交流群二维码

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