arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2412.06045eess.IVcs.CVcs.LG

用于医学图像分割的扩张平衡交叉熵损失

Dilated Balanced Cross Entropy Loss for Medical Image Segmentation

University of Tehran(德黑兰大学);Sharif University of Technology(谢里夫理工大学)

Seyed Mohsen Hosseini, Mahdieh Soleymani Baghshah

AI总结:

针对医学图像分割的类别不平衡问题,提出基于类别掩码扩张区域加权的扩张平衡交叉熵损失,大幅提升平衡交叉熵性能,效果可媲美Dice与交叉熵的组合损失。

摘要:

本研究提出了一种解决医学图像分割中数据不平衡问题的新方法。平衡交叉熵(CE)损失是加权交叉熵损失的一种,其为每个类别分配的权重是类别频率的倒数。这些平衡权重原本旨在均衡每个类别对总损失的影响,防止模型偏向多数类。但正如既往研究所示,该方法会导致性能大幅下降,因此平衡交叉熵并非医学分割任务中常用的损失函数,这类任务通常采用Dice损失等基于区域的损失来解决类别不平衡问题。在本文提出的方法中,每个类别的交叉熵损失权重基于各类别掩码的扩张区域计算,平衡权重会分配给每个类别及其周围像素。本研究的目标是证明,通过修改加权策略,平衡交叉熵损失的性能可得到大幅提升。在不同数据集上的实验表明,所提出的扩张平衡交叉熵(DBCE)损失大幅优于平衡交叉熵损失,且效果优于标准交叉熵损失,其性能与Dice损失和交叉熵损失的组合效果相近。这意味着采用合理加权策略的加权交叉熵损失,在处理医学分割任务中的类别不平衡问题时,能达到与基于区域的损失相当的效果。

英文摘要:

A novel method for tackling the problem of imbalanced data in medical image segmentation is proposed in this work. In balanced cross entropy (CE) loss, which is a type of weighted CE loss, the weight assigned to each class is the in-verse of the class frequency. These balancing weights are expected to equalize the effect of each class on the overall loss and prevent the model from being biased towards the majority class. But, as it has been shown in previous studies, this method degrades the performance by a large margin. Therefore, balanced CE is not a popular loss in medical segmentation tasks, and usually a region-based loss, like the Dice loss, is used to address the class imbalance problem. In the pro-posed method, the weighting of cross entropy loss for each class is based on a dilated area of each class mask, and balancing weights are assigned to each class together with its surrounding pixels. The goal of this study is to show that the performance of balanced CE loss can be greatly improved my modifying its weighting strategy. Experiments on different datasets show that the proposed dilated balanced CE (DBCE) loss outperforms the balanced CE loss by a large margin and produces superior results compared to CE loss, and its performance is similar to the performance of the combination of Dice and CE loss. This means that a weighted cross entropy loss with the right weighing strategy can be as effective as a region-based loss in handling the problem of class imbalance in medical segmentation tasks.

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

↑