Topology aware multitask cascaded U-Net for cerebrovascular segmentation.
Cerebrovascular segmentation is a crucial preliminary task for many computer-aided diagnosis tools dealing with cerebrovascular pathologies. Over the last years, deep learning based methods have been widely applied to this task. However, classic deep learning approaches struggle to capture the compl...
Saved in:
| Main Authors: | , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Public Library of Science (PLoS)
2024-01-01
|
| Series: | PLoS ONE |
| Online Access: | https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0311439&type=printable |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1850163848425766912 |
|---|---|
| author | Pierre Rougé Nicolas Passat Odyssée Merveille |
| author_facet | Pierre Rougé Nicolas Passat Odyssée Merveille |
| author_sort | Pierre Rougé |
| collection | DOAJ |
| description | Cerebrovascular segmentation is a crucial preliminary task for many computer-aided diagnosis tools dealing with cerebrovascular pathologies. Over the last years, deep learning based methods have been widely applied to this task. However, classic deep learning approaches struggle to capture the complex geometry and specific topology of cerebrovascular networks, which is of the utmost importance in many applications. To overcome these limitations, the clDice loss, a topological loss that focuses on the vessel centerlines, has been recently proposed. This loss requires computing the skeletons of both the manual annotation and the predicted segmentation in a differentiable way. Currently, differentiable skeletonization algorithms are either inaccurate or computationally demanding. In this article, it is proposed that a U-Net be used to compute the vascular skeleton directly from the segmentation and the magnetic resonance angiography image. This method is naturally differentiable and provides a good trade-off between accuracy and computation time. The resulting cascaded multitask U-Net is trained with the clDice loss to embed topological constraints during the segmentation. In addition to this topological guidance, this cascaded U-Net also benefits from the inductive bias generated by the skeletonization during the multitask training. This model is able to predict the cerebrovascular segmentation with a more accurate topology than current state-of-the-art methods and with a low training time. This method is evaluated on two publicly available time-of-flight magnetic resonance angiography (TOF-MRA) images datasets, also the codes of the proposed method and the reimplementation of state-of-the-art methods are made available at: https://github.com/PierreRouge/Cascaded-U-Net-for-vessel-segmentation. |
| format | Article |
| id | doaj-art-8a642b6bcbd84c9b9cfbb95ed7fa6601 |
| institution | OA Journals |
| issn | 1932-6203 |
| language | English |
| publishDate | 2024-01-01 |
| publisher | Public Library of Science (PLoS) |
| record_format | Article |
| series | PLoS ONE |
| spelling | doaj-art-8a642b6bcbd84c9b9cfbb95ed7fa66012025-08-20T02:22:06ZengPublic Library of Science (PLoS)PLoS ONE1932-62032024-01-011912e031143910.1371/journal.pone.0311439Topology aware multitask cascaded U-Net for cerebrovascular segmentation.Pierre RougéNicolas PassatOdyssée MerveilleCerebrovascular segmentation is a crucial preliminary task for many computer-aided diagnosis tools dealing with cerebrovascular pathologies. Over the last years, deep learning based methods have been widely applied to this task. However, classic deep learning approaches struggle to capture the complex geometry and specific topology of cerebrovascular networks, which is of the utmost importance in many applications. To overcome these limitations, the clDice loss, a topological loss that focuses on the vessel centerlines, has been recently proposed. This loss requires computing the skeletons of both the manual annotation and the predicted segmentation in a differentiable way. Currently, differentiable skeletonization algorithms are either inaccurate or computationally demanding. In this article, it is proposed that a U-Net be used to compute the vascular skeleton directly from the segmentation and the magnetic resonance angiography image. This method is naturally differentiable and provides a good trade-off between accuracy and computation time. The resulting cascaded multitask U-Net is trained with the clDice loss to embed topological constraints during the segmentation. In addition to this topological guidance, this cascaded U-Net also benefits from the inductive bias generated by the skeletonization during the multitask training. This model is able to predict the cerebrovascular segmentation with a more accurate topology than current state-of-the-art methods and with a low training time. This method is evaluated on two publicly available time-of-flight magnetic resonance angiography (TOF-MRA) images datasets, also the codes of the proposed method and the reimplementation of state-of-the-art methods are made available at: https://github.com/PierreRouge/Cascaded-U-Net-for-vessel-segmentation.https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0311439&type=printable |
| spellingShingle | Pierre Rougé Nicolas Passat Odyssée Merveille Topology aware multitask cascaded U-Net for cerebrovascular segmentation. PLoS ONE |
| title | Topology aware multitask cascaded U-Net for cerebrovascular segmentation. |
| title_full | Topology aware multitask cascaded U-Net for cerebrovascular segmentation. |
| title_fullStr | Topology aware multitask cascaded U-Net for cerebrovascular segmentation. |
| title_full_unstemmed | Topology aware multitask cascaded U-Net for cerebrovascular segmentation. |
| title_short | Topology aware multitask cascaded U-Net for cerebrovascular segmentation. |
| title_sort | topology aware multitask cascaded u net for cerebrovascular segmentation |
| url | https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0311439&type=printable |
| work_keys_str_mv | AT pierrerouge topologyawaremultitaskcascadedunetforcerebrovascularsegmentation AT nicolaspassat topologyawaremultitaskcascadedunetforcerebrovascularsegmentation AT odysseemerveille topologyawaremultitaskcascadedunetforcerebrovascularsegmentation |