Search papers, labs, and topics across Lattice.
This paper introduces Polynomial Dice Loss, a novel extension of the traditional Dice Loss designed to address challenges in medical image segmentation, particularly data imbalance and small lesion detection. By employing a polynomial representation through Taylor expansion, the authors allow for adjustable contributions of higher-order components, enhancing the loss function's effectiveness. Experimental results demonstrate that this approach not only controls the loss shape but also achieves competitive performance compared to conventional Dice and Tversky coefficients across various segmentation tasks.
Polynomial Dice Loss offers a flexible way to fine-tune loss functions, significantly improving small lesion detection in medical image segmentation.
Medical image segmentation is a fundamental task for medical image processing and computer-assisted intervention, yet data imbalance and small lesion detection pose significant challenges. Dice Loss, which measures the overlap between predicted and ground truth regions, is widely used to mitigate these issues. To further emphasize its properties, we propose Polynomial Dice Loss, a polynomial extension of Dice Loss. Specifically, by leveraging the geometric characteristics of Dice Loss and formulating the loss function as a polynomial representation via Taylor expansion, we enable the adjustment of the contribution of higher-order components to the loss function. In our experiments, we evaluate the proposed method against loss functions derived from conventional Dice and Tversky coefficients. Experimental results and further analysis show that the polynomial formulation provides a simple way to control the loss shape and achieves competitive performance across multiple segmentation settings.
