Command Palette
Search for a command to run...
A Novel Focal Tversky loss function with improved Attention U-Net for lesion segmentation
Nabila Abraham; Naimul Mefraz Khan
Abstract
We propose a generalized focal loss function based on the Tversky index to address the issue of data imbalance in medical image segmentation. Compared to the commonly used Dice loss, our loss function achieves a better trade off between precision and recall when training on small structures such as lesions. To evaluate our loss function, we improve the attention U-Net model by incorporating an image pyramid to preserve contextual features. We experiment on the BUS 2017 dataset and ISIC 2018 dataset where lesions occupy 4.84% and 21.4% of the images area and improve segmentation accuracy when compared to the standard U-Net by 25.7% and 3.6%, respectively.
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| lesion-segmentation-on-bus-2017-dataset-b | Attn U-Net + DL | Dice Score: 0.615 |
| lesion-segmentation-on-bus-2017-dataset-b | Attn U-Net + Multi-Input + FTL | Dice Score: 0.804 |
| lesion-segmentation-on-bus-2017-dataset-b | U-Net + FTL | Dice Score: 0.669 |
| lesion-segmentation-on-isic-2018 | U-Net + FTL | mean Dice: 0.829 |
| lesion-segmentation-on-isic-2018 | Attn U-Net + DL | mean Dice: 0.806 |
| lesion-segmentation-on-isic-2018 | Attn U-Net + Multi-Input + FTL | mean Dice: 0.856 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.