Unified Focal loss: Generalising Dice and cross entropy-based losses to handle class imbalanced medical image segmentation

🤖 Plain-English Summary

Automatic segmentation methods are an important advancement in medical image analysis. We compare our loss function performance against six Dice or cross entropy-based loss functions, across 2D binary, 3D binary and 3D multiclass segmentation tasks, demonstrating that our proposed loss function is robust to class imbalance and consistently outperforms the other loss functions.

🔑 Key Findings

• Machine learning techniques, and deep neural networks in particular, are the advanced for most medical image segmentation tasks.
• Issues with class imbalance pose a significant challenge in medical datasets, with lesions often occupying a considerably smaller volume relative to the background.
• Loss functions used in the training of deep learning algorithms differ in their robustness to class imbalance, with direct consequences for model convergence.

🔗 Related Resources