Table 6 Comparison of the proposed loss function and standard loss function for class imbalance handling on test set with mean LA volume 149.80 ± 4.3 mL and EAT volume 14.63 ± 1.9 mL.
Method | Dice score | LA volume | EAT volume |
|---|---|---|---|
Cross-entropy loss | 81.15 ± 3.5 | 150.45 ± 4.7 | 15.55 ± 1.7 |
Dice loss | 83.00 ± 3.1 | 148.07 ± 4.5 | 14.37 ± 2.5 |
Dice loss + Cross-entropy loss | 84.95 ± 2.8 | 149.97 ± 5.7 | 15.56 ± 2.6 |
The proposed loss | 87.25 ± 2.8 | 149.84 ± 4.2 | 14.70 ± 1.3 |
