Papers › A Novel Focal Tversky loss function with improved Attention U-Net for lesion segmentation
A Novel Focal Tversky loss function with improved Attention U-Net for lesion segmentation
Nabila Abraham, Naimul Mefraz Khan
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.
In Syntology View this paper on Syntology: its repositories, every harvested function with whether it ran, its licence and the call to fetch it.
Open this paper in Syntology's Atlas, the map of the papers in Syntology's graph and their citations.
Code
Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.
Code Syntology ran Syntology
Not run by Syntology. Nothing on this page verifies that the listed code works.
Tasks
Results from the paper archive 2025-07-28
| Task | Dataset | Model | Metric | Value | Rank at snapshot | Leaderboard | Report |
|---|---|---|---|---|---|---|---|
| Lesion Segmentation | BUS 2017 Dataset B | Attn U-Net + Multi-Input + FTL | Dice Score | 0.804 | #1 of 4 | Archive leaderboard | report |
| Lesion Segmentation | BUS 2017 Dataset B | U-Net + FTL | Dice Score | 0.669 | #3 of 4 | Archive leaderboard | report |
| Lesion Segmentation | BUS 2017 Dataset B | Attn U-Net + DL | Dice Score | 0.615 | #4 of 4 | Archive leaderboard | report |
| Lesion Segmentation | ISIC 2018 | Attn U-Net + Multi-Input + FTL | mean Dice | 0.856 | #12 of 17 | Archive leaderboard | report |
| Lesion Segmentation | ISIC 2018 | U-Net + FTL | mean Dice | 0.829 | #15 of 17 | Archive leaderboard | report |
| Lesion Segmentation | ISIC 2018 | Attn U-Net + DL | mean Dice | 0.806 | #16 of 17 | Archive leaderboard | report |
Ranks are positions in the archive's leaderboards as they stood at the 2025-07-28 snapshot. Results published since then are not among these rows, so a rank here is not a current standing.
Methods
Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections