Adaptive composite loss for volumetric whole heart segmentation

用于全心容积分割的自适应复合损失

阅读:3

Abstract

Accurate segmentation in medical imaging requires loss functions that capture both regional overlap and boundary alignment. This study evaluates composite losses combining binary cross-entropy (BCE) and a boundary-based term under fixed and adaptive weighting schemes, using U-Net and SwinUNETR on the MM-WHS dataset. For U-Net, a small boundary contribution with adaptive weighting yielded the best results: Standard SoftAdapt (90/10 BCE + BoundaryDoU) achieved the highest Dice score ([Formula: see text]), surpassing both the baseline ([Formula: see text]) and fixed ratios. In contrast, SwinUNETR achieved its strongest performance with a fixed 70% BCE + 10% boundary ratio (0.919 ± 0.02). The result showed that combining a boundary-based loss term helps improve the segmentation accuracy. However, the performance gain is dependent on the architecture of the segmentation model; convolution-based U-Net benefited from the adaptive loss weighting scheme, whereas Transformer-based SwinUNETR without strong inductive bias did not benefit from increased influence of the boundary loss term.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。