Analysis of Loss Functions for Colorectal Polyp Segmentation Under Class Imbalance

2025

Full text Open access: Gold Metadata (CSL-JSON)
DOI 10.3390/engproc2025104017
Data confidence 70 medium
How this score is calculated

It is built from checkable properties of the record itself: whether the article has a DOI, how many independent sources confirmed its data, whether the required fields are filled in, and how recent the latest confirmation is. Disagreement between sources and a missing identifier lower it. A value of 80 or above means the record was published without further checking; 60 to 79 means it is published at reduced confidence, with a manual check advisable. A record verified by an editor by hand is never pushed below 90 by a recalculation. This scores our data about the article, not the article itself.

Abstract

Class imbalance is a persistent limitation in polyp segmentation, commonly resulting in biased predictions and reduced accuracy in identifying clinically relevant structures. This study systematically evaluated 12 loss functions, including standard, weighted, and compound formulas, applied to colon polyp segmentation using the UNet-VGG16 fixed architecture on the Kvasir-SEG dataset. The encoder was frozen to isolate the effect of loss functions under the same training conditions. A fixed random seed was used in all experiments to ensure reproducibility and control variance during training. The results reveal that the combined loss functions, namely WBCE combined with Dice and Tversky combined with Focal, achieved the top Dice scores of 0.8916 and 0.8917, respectively. Tversky plus Focal also provided the highest sensitivity of 0.8885, and WBCE obtained the best average IoU of 0.8120. Tversky loss showed the lowest error rate of 4.99, indicating stable optimization. These results clarify the influence of loss function selection on segmentation performance in scenarios characterized by considerable class imbalance.