Adaptive composite loss for volumetric whole heart segmentation.
Authors
Affiliations (4)
Affiliations (4)
- Faculty of Information and Communication Technology, Mahidol University, Nakhon Pathom, 73170, Thailand.
- Faculty of Information and Communication Technology, Mahidol University, Nakhon Pathom, 73170, Thailand. [email protected].
- School of Computing, Macquarie University, Macquarie Park, NSW, 2113, Australia.
- Department of Radiology, Faculty of Medicine Siriraj Hospital, Mahidol University, Bangkok, 10700, Thailand.
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.