no code implementations • 24 Apr 2023 • Andac Demir, Elie Massaad, Bulent Kiziltan
To tackle this problem, we introduce a novel loss function, namely Topology-Aware Focal Loss (TAFL), that incorporates the conventional Focal Loss with a topological constraint term based on the Wasserstein distance between the ground truth and predicted segmentation masks' persistence diagrams.