CM-UNet: A Self-Supervised Learning-Based Model for Coronary Artery Segmentation in X-Ray Angiography

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Challier, Camille, Sun, Xiaowu, Mahendiran, Thabo, Senouf, Ortal, De Bruyne, Bernard, Auberson, Denise, Müller, Olivier, Fournier, Stephane, Frossard, Pascal, Abbé, Emmanuel, Thanou, Dorina
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866911073883389952
author Challier, Camille
Sun, Xiaowu
Mahendiran, Thabo
Senouf, Ortal
De Bruyne, Bernard
Auberson, Denise
Müller, Olivier
Fournier, Stephane
Frossard, Pascal
Abbé, Emmanuel
Thanou, Dorina
author_facet Challier, Camille
Sun, Xiaowu
Mahendiran, Thabo
Senouf, Ortal
De Bruyne, Bernard
Auberson, Denise
Müller, Olivier
Fournier, Stephane
Frossard, Pascal
Abbé, Emmanuel
Thanou, Dorina
contents Accurate segmentation of coronary arteries remains a significant challenge in clinical practice, hindering the ability to effectively diagnose and manage coronary artery disease. The lack of large, annotated datasets for model training exacerbates this issue, limiting the development of automated tools that could assist radiologists. To address this, we introduce CM-UNet, which leverages self-supervised pre-training on unannotated datasets and transfer learning on limited annotated data, enabling accurate disease detection while minimizing the need for extensive manual annotations. Fine-tuning CM-UNet with only 18 annotated images instead of 500 resulted in a 15.2% decrease in Dice score, compared to a 46.5% drop in baseline models without pre-training. This demonstrates that self-supervised learning can enhance segmentation performance and reduce dependence on large datasets. This is one of the first studies to highlight the importance of self-supervised learning in improving coronary artery segmentation from X-ray angiography, with potential implications for advancing diagnostic accuracy in clinical practice. By enhancing segmentation accuracy in X-ray angiography images, the proposed approach aims to improve clinical workflows, reduce radiologists' workload, and accelerate disease detection, ultimately contributing to better patient outcomes. The source code is publicly available at https://github.com/CamilleChallier/Contrastive-Masked-UNet.
format Preprint
id arxiv_https___arxiv_org_abs_2507_17779
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle CM-UNet: A Self-Supervised Learning-Based Model for Coronary Artery Segmentation in X-Ray Angiography
Challier, Camille
Sun, Xiaowu
Mahendiran, Thabo
Senouf, Ortal
De Bruyne, Bernard
Auberson, Denise
Müller, Olivier
Fournier, Stephane
Frossard, Pascal
Abbé, Emmanuel
Thanou, Dorina
Quantitative Methods
Machine Learning
I.2; I.4; I.5; J.3
Accurate segmentation of coronary arteries remains a significant challenge in clinical practice, hindering the ability to effectively diagnose and manage coronary artery disease. The lack of large, annotated datasets for model training exacerbates this issue, limiting the development of automated tools that could assist radiologists. To address this, we introduce CM-UNet, which leverages self-supervised pre-training on unannotated datasets and transfer learning on limited annotated data, enabling accurate disease detection while minimizing the need for extensive manual annotations. Fine-tuning CM-UNet with only 18 annotated images instead of 500 resulted in a 15.2% decrease in Dice score, compared to a 46.5% drop in baseline models without pre-training. This demonstrates that self-supervised learning can enhance segmentation performance and reduce dependence on large datasets. This is one of the first studies to highlight the importance of self-supervised learning in improving coronary artery segmentation from X-ray angiography, with potential implications for advancing diagnostic accuracy in clinical practice. By enhancing segmentation accuracy in X-ray angiography images, the proposed approach aims to improve clinical workflows, reduce radiologists' workload, and accelerate disease detection, ultimately contributing to better patient outcomes. The source code is publicly available at https://github.com/CamilleChallier/Contrastive-Masked-UNet.
title CM-UNet: A Self-Supervised Learning-Based Model for Coronary Artery Segmentation in X-Ray Angiography
topic Quantitative Methods
Machine Learning
I.2; I.4; I.5; J.3
url https://arxiv.org/abs/2507.17779