Topologically faithful multi-class segmentation in medical images

التفاصيل البيبلوغرافية
العنوان: Topologically faithful multi-class segmentation in medical images
المؤلفون: Berger, Alexander H., Stucki, Nico, Lux, Laurin, Buergin, Vincent, Shit, Suprosanna, Banaszak, Anna, Rueckert, Daniel, Bauer, Ulrich, Paetzold, Johannes C.
سنة النشر: 2024
المجموعة: Computer Science
مصطلحات موضوعية: Electrical Engineering and Systems Science - Image and Video Processing, Computer Science - Computer Vision and Pattern Recognition, Computer Science - Machine Learning
الوصف: Topological accuracy in medical image segmentation is a highly important property for downstream applications such as network analysis and flow modeling in vessels or cell counting. Recently, significant methodological advancements have brought well-founded concepts from algebraic topology to binary segmentation. However, these approaches have been underexplored in multi-class segmentation scenarios, where topological errors are common. We propose a general loss function for topologically faithful multi-class segmentation extending the recent Betti matching concept, which is based on induced matchings of persistence barcodes. We project the N-class segmentation problem to N single-class segmentation tasks, which allows us to use 1-parameter persistent homology making training of neural networks computationally feasible. We validate our method on a comprehensive set of four medical datasets with highly variant topological characteristics. Our loss formulation significantly enhances topological correctness in cardiac, cell, artery-vein, and Circle of Willis segmentation.
نوع الوثيقة: Working Paper
URL الوصول: http://arxiv.org/abs/2403.11001
رقم الأكسشن: edsarx.2403.11001
قاعدة البيانات: arXiv