Uncertainty-Guided Active Learning for Access Route Segmentation and Planning in Transcatheter Aortic Valve Implantation
Abstract
1. Introduction
- An active learning pipeline for aortic structure segmentation is introduced, designed to achieve high accuracy while significantly reducing the manual annotation effort compared with traditional fully supervised methods. Unlike prior segmentation works that focused on CTA or abdominal aorta only, this represents one of the first applications for CMR-based TAVI planning.
- A robust uncertainty estimation strategy is incorporated to guide sample selection, whereby the most informative cases are queried for expert review and high-confidence predictions are used for pseudo-labelling. This hybrid strategy optimizes the use of limited annotation budgets, addressing the annotation bottleneck that has not been previously explored in the TAVI context.
- A complete workflow is developed that extends beyond segmentation to include automated centerline extraction and a hybrid diameter calculation strategy. In contrast to existing studies that report only landmark-based or discrete diameter measurements, this study enables continuous profiling of vessel diameters along the aorto-iliac path, which has not yet been addressed in prior literature.
- Using a dataset of paired pre- and post-contrast scans, a modality ablation study is performed to systematically evaluate the distinct and combined contributions of each imaging modality. To the best of authors’ knowledge, this is the first such evaluation in the context of CMR-based TAVI planning.
2. Dataset
3. Methods
3.1. Vessel Diameter Quantification
3.1.1. Centerline Extraction
- Morphological Skeletonization: The 3D binary mask is iteratively thinned by successively removing boundary voxels while preserving the overall topology, resulting in a 1-voxel-wide skeleton that represents the medial axis of the structure.
- Largest Connected Component (LCC): Only the LCC of , , is retained to remove noise/irrelevant branches.
- Graph Representation and Path Finding: is converted to a graph using ‘skan.skeleton_to_csgraph’. Endpoints are identified. The principal centerline is defined as the longest shortest path between any two endpoints in G, obtained with Dijkstra’s algorithm [24].
- Centerline Ordering: The sequence of ZYX coordinates represents the points along the extracted longest path of the skeleton. To enforce a consistent orientation, the path is ordered such that the Z-coordinate of the first point () is less than or equal to the Z-coordinate of the last point (), i.e., . This ensures a robust and reproducible inferior-to-superior progression (or an equivalent direction depending on the Z-axis definition) when traversing the centerline.
3.1.2. Centerline Refinement
- 1.
- Smoothing: To mitigate voxel discretization artifacts, Y and X coordinates of the raw centerline C are smoothed using a 1D uniform filter as Equation (1).Sudden coordinate jumps exceeding a threshold are corrected by local averaging. The smoothed centerline is .
- 2.
- Endpoint Correction: For an initial fraction of points, 5%, their Y and X coordinates are adjusted to the Center of Mass (CoM) of the vessel mask in the corresponding Z-slice . This ensures the centerline robustly starts within the vessel lumen at its ends, yielding .
3.1.3. Diameter Calculation
- 1.
- Distance Transform (DT) Diameter: A 3D Euclidean Distance Transform (EDT) is computed on considering voxel spacing S. gives the initial diameter estimate.
- 2.
- Ray-Casting Diameter:
- Local Tangent: A local tangent vector at is computed using finite differences between neighboring centerline points.
- Orthogonal Plane Definition: A plane orthogonal to is defined.
- –
- If is nearly parallel to the Z-axis (i.e., ), the YX-plane at is used.
- –
- Otherwise, two orthogonal vectors and are computed that span the plane perpendicular to (e.g., using the Frisvad method), ensuring numerical stability and orthogonality.
- Ray Casting: A set of (e.g., 8) rays are cast from within this plane. Each direction of ray is defined as a linear combination of and , or lies within the YX plane if the Z-parallel condition holds.
- Boundary Detection: For each ray originating at and extending in directions , the exit points where the ray leaves the vessel mask are detected, denoted as and .
- Ray Diameter: The diameter along each ray is computed as Equation (2):
- 3.
- Combined Diameter: The “raw” diameter is defined as Equation (3)If missing values in are encountered, they are linearly interpolated. Finally, a Savitzky–Golay filter (polynomial order 3, window length up to 51 points) is applied to obtain the final smoothed diameter profile .
3.2. Active Learning Implementation
Uncertainty Estimation
3.3. Multimodal Ablation Study
4. Results
4.1. Vessel Diameter Profile
4.2. Active Learning Performance Evaluation
4.3. Multimodal Ablation
5. Discussion
- Diameter profile validation: The automated pipeline generated physiologically consistent diameter curves (Figure 4), confirming its ability to provide clinically interpretable vessel measurements beyond segmentation accuracy. This is particularly relevant given the 5 mm iliac threshold that often determines transfemoral access eligibility.
- Effectiveness of active learning: Performance improved across iterations (Figure 5), with Dice scores rising from ≈0.79 to and iliac MAPE decreasing by more than 50%. The most substantial gains occurred within the first few cycles, showing that expert annotation efforts concentrated on uncertain cases yield rapid improvements, especially for anatomically challenging iliac arteries. The rapid plateau after two to three iterations suggests annotation efficiency can be maximized by focusing expert effort on early cycles, with diminishing returns beyond 20–30 manually labelled cases.
- Impact of modality and pseudo-labels: The multimodal ablation study (Table 2) revealed post-contrast CMR as the most reliable input, delivering the highest Dice and lowest MAPE. By contrast, the inclusion of inferred cases slightly degraded performance, underscoring the sensitivity of segmentation models to label noise when pseudo-label confidence is not tightly controlled. The success of uncertainty-guided pseudo-labelling during active learning contrasted with blind inference performance, emphasizing that pseudo-labelling strategies require carefully designed confidence thresholds and quality control mechanisms.
- Model architecture vs. data quality: Differences between 3D U-Net and UMamba were modest compared to the impact of input modality. This suggests that improving data quality and curation may be more beneficial than pursuing marginal network refinements for this application. Surprisingly, combined pre- and post-contrast data did not consistently improve performance, suggesting current architectures may struggle with effective multimodal feature fusion or that alignment issues introduce artifacts.
- Clinical implications: The combination of high Dice (≈0.97 for aorta, ≈0.85 for iliacs) and low diameter error (≈5% MAPE) supports the potential of this framework for integration into preoperative workflows. Accurate iliac quantification is particularly valuable, as even small diameter misestimations can alter access decisions in borderline cases. While accuracy approaches clinical requirements, results suggest stronger potential for semi-automated workflows where the system provides initial measurements and uncertainty estimates to guide radiologist review. Qualitative analysis (Figure 6) revealed continued improvements for challenging cases even after aggregate metrics plateaued, justifying the full five-iteration protocol.
- Limitations: The study was based on single-center data, which may limit generalizability across populations, scanner vendors, or imaging protocols. Performance on pre-contrast CMR remained substantially weaker, even when combined with post-contrast data, indicating that multimodal integration strategies require further refinement. Finally, as the pipeline is sequential, segmentation inaccuracies propagate to downstream diameter estimation, representing an area for future methodological improvement.
6. Conclusions
Author Contributions
Funding
Institutional Review Board Statement
Informed Consent Statement
Data Availability Statement
Conflicts of Interest
References
- Osnabrugge, R.L.; Mylotte, D.; Head, S.J.; Van Mieghem, N.M.; Nkomo, V.T.; LeReun, C.M.; Bogers, A.J.; Piazza, N.; Kappetein, A.P. Aortic Stenosis in the Elderly: Disease Prevalence and Number of Candidates for Transcatheter Aortic Valve Replacement: A Meta-Analysis and Modeling Study. J. Am. Coll. Cardiol. 2013, 62, 1002–1012. [Google Scholar] [CrossRef]
- Díez. Transcatheter aortic valve implantation (TAVI): The hype and the hope. Tex. Heart Inst. J. 2013, 40, 298–301. [Google Scholar]
- Mach, M.; Okutucu, S.; Kerbel, T.; Arjomand, A.; Fatihoglu, S.G.; Werner, P.; Simon, P.; Andreas, M. Vascular Complications in TAVR: Incidence, Clinical Impact, and Management. J. Clin. Med. 2021, 10, 5046. [Google Scholar] [CrossRef] [PubMed]
- Al-Najafi, S.; Sanchez, F.; Lerakis, S. The crucial role of cardiac imaging in transcatheter aortic valve replacement (TAVR): Pre- and post-procedural assessment. Curr. Treat. Options Cardiovasc. Med. 2016, 18, 70. [Google Scholar] [CrossRef]
- Ram, P.; Mezue, K.; Pressman, G.; Rangaswami, J. Acute kidney injury post–transcatheter aortic valve replacement. Clin. Cardiol. 2017, 40, 1357–1362. [Google Scholar] [CrossRef] [PubMed]
- Mahon, C.; Mohiaddin, R. The emerging applications of cardiovascular magnetic resonance imaging in transcatheter aortic valve implantation. Clin. Radiol. 2021, 76, 73.e21–73.e37. [Google Scholar] [CrossRef]
- Mayr, A.; Klug, G.; Reinstadler, S.J.; Feistritzer, H.J.; Reindl, M.; Kremser, C.; Kranewitter, C.; Bonaros, N.; Friedrich, G.; Feuchtner, G.; et al. Is MRI equivalent to CT in the guidance of TAVR? A pilot study. Eur. Radiol. 2018, 28, 4625–4634. [Google Scholar] [CrossRef]
- Wang, Y.; Bai, L.; Yang, J.; Lu, Y.; Fan, W.; Nie, Z.; Yu, J.; Wen, K.; Wang, R.; He, L.; et al. Artificial intelligence measuring the aortic diameter assists in identifying adverse blood pressure status including masked hypertension. Postgrad. Med. 2022, 134, 111–121. [Google Scholar] [CrossRef]
- Postiglione, T.J.; Guillo, E.; Heraud, A.; Rossillon, A.; Bartoli, M.; Herpe, G.; Adam, C.; Fabre, D.; Ardon, R.; Azarine, A.; et al. Multicentric clinical evaluation of a computed tomography-based fully automated deep neural network for aortic maximum diameter and volumetric measurements. J. Vasc. Surg. 2024, 79, 1390–1400. [Google Scholar] [CrossRef]
- Ferrari, I.; Shehu, N.; Mkrtchyan, N.; Latus, H.; Stern, H.; Martinoff, S.; Ewert, P.; Meierhofer, C. Aortic diameter assessment by cardiovascular magnetic resonance: Do we really need contrast enhanced images? Cardiovasc. Diagn. Ther. 2021, 11, 1389–1394. [Google Scholar] [CrossRef]
- Rong, L.Q.; Chun, S.; Villar-Calle, P.; Reza, M.; Leshem, E.; Falco, G.; Kim, J.; Devereux, R.B.; Weinsaft, J.W. Thoracic Aorta Measurements Using Intraoperative Transesophageal Echocardiography: Validation via Cardiac Magnetic Resonance. J. Cardiothorac. Vasc. Anesth. 2025, 39, 1135–1145. [Google Scholar] [CrossRef] [PubMed]
- Saitta, S.; Sturla, F.; Gorla, R.; Oliva, O.A.; Votta, E.; Bedogni, F.; Redaelli, A. A CT-based deep learning system for automatic assessment of aortic root morphology for TAVI planning. Comput. Biol. Med. 2023, 163, 107147. [Google Scholar] [CrossRef]
- Wang, M.; Niu, G.; Chen, Y.; Zhou, Z.; Feng, D.; Zhang, Y.; Wu, Y. Development and validation of a deep learning-based fully automated algorithm for pre-TAVR CT assessment of the aortic valvular complex and detection of anatomical risk factors: A retrospective, multicentre study. eBioMedicine 2023, 96, 104794. [Google Scholar] [CrossRef]
- Santaló-Corcoy, M.; Corbin, D.; Tastet, O.; Lesage, F.; Modine, T.; Asgar, A.; Ben Ali, W. TAVI-PREP: A Deep Learning-Based Tool for Automated Measurements Extraction in TAVI Planning. Diagnostics 2023, 13, 3181. [Google Scholar]
- Blankenberg, S.; Stenzinger, A.; Kiefer, P.; Braun, D.; Wehrum, T.; Michel, C.; Gigliotti, M.J.; Kloeckner, R.; Schnabel, J.A.; Maintz, D.; et al. A fully automated artificial intelligence-driven software for planning of transcatheter aortic valve replacement. JACC Cardiovasc. Imaging 2024, 17, 8. [Google Scholar] [CrossRef]
- Inomata, S.; Yoshimura, T.; Tang, M.; Ichikawa, S.; Sugimori, H. Automatic Aortic Valve Extraction Using Deep Learning with Contrast-Enhanced Cardiac CT Images. JCDD 2025, 12, 3. [Google Scholar]
- Imran, M.; Krebs, J.R.; Gopu, V.R.R.; Fazzone, B.; Sivaramand, V.B.; Kumar, A.; Viscardi, C.; Heithaus, R.E.; Shickel, B.; Zhou, Y.; et al. CIS-UNet: Multi-Class Segmentation of the Aorta in Computed Tomography Angiography via Context-Aware Shifted Window Self-Attention. arXiv 2024, arXiv:2401.13049. [Google Scholar] [CrossRef]
- Ruijsink, B.; Puyol-Anton, E.; Li, Y.; Bai, W.; Kerfoot, E.; Razavi, R.; King, A.P. Quality-aware semi-supervised learning for CMR segmentation. arXiv 2020, arXiv:2009.00584. [Google Scholar]
- Ma, J.; Yang, Z.; Kim, S.; Chen, B.; Baharoon, M.; Fallahpour, A.; Asakereh, R.; Lyu, H.; Wang, B. MedSAM2: Segment Anything in 3D Medical Images and Videos. arXiv 2025, arXiv:2504.03600. [Google Scholar] [CrossRef]
- Shi, J.; Ruan, S.; Zhu, Z.; Zhao, M.; An, H.; Xue, X.; Yan, B. Predictive Accuracy-Based Active Learning for Medical Image Segmentation. arXiv 2024, arXiv:2405.00452. [Google Scholar] [CrossRef]
- Gaillochet, M.; Desrosiers, C.; Lombaert, H. TAAL: Test-Time Augmentation for Active Learning in Medical Image Segmentation. arXiv 2023, arXiv:2301.06624. [Google Scholar] [CrossRef]
- Ma, S.; Wu, H.; Lawlor, A.; Dong, R. Breaking the Barrier: Selective Uncertainty-Based Active Learning for Medical Image Segmentation. arXiv 2024, arXiv:2401.16298. [Google Scholar] [CrossRef]
- Reindl, M.; Lechner, I.; Holzknecht, M.; Tiller, C.; Fink, P.; Oberhollenzer, F.; von der Emde, S.; Pamminger, M.; Troger, F.; Kremser, C.; et al. Cardiac Magnetic Resonance Imaging Versus Computed Tomography to Guide Transcatheter Aortic Valve Replacement: A Randomized, Open-Label, Noninferiority Trial. Circulation 2023, 148, 1220–1230. [Google Scholar] [CrossRef] [PubMed]
- Dijkstra, E.W. A note on two problems in connexion with graphs. Numer. Math. 1959, 1, 269–271. [Google Scholar] [CrossRef]
- Maurer, C.R.; Qi, R.; Raghavan, V. A linear time algorithm for computing exact Euclidean distance transforms of binary images in arbitrary dimensions. IEEE Trans. Pattern Anal. Mach. Intell. 2003, 25, 265–270. [Google Scholar] [CrossRef]
- Frangi, A.F.; Niessen, W.J.; Vincken, K.L.; Viergever, M.A. Multiscale vessel enhancement filtering. In Proceedings of the International Conference on Medical Image Computing and Computer-Assisted Intervention (MICCAI), Cambridge, MA, USA, 11–13 October 1998; Springer: Berlin/Heidelberg, Germany, 1998; pp. 130–137. [Google Scholar]
- Sato, Y.; Nakajima, S.; Shiraga, N.; Atsumi, H.; Yoshida, S.; Koller, T.; Gerig, G.; Kikinis, R. 3D multi-scale line filter for segmentation and visualization of curvilinear structures in medical images. In Proceedings of the International Conference on Computer Vision, Virtual Reality and Robotics in Medicine and Medical Robotics and Computer-Assisted Surgery, Nice, France, 3–6 April 1997; Springer: Berlin/Heidelberg, Germany, 1997; pp. 213–222. [Google Scholar]
- Lorenz, C.; Carlsen, I.C.; Buzug, T.M.; Fassnacht, C.; Weese, J. Multi-scale line segmentation with automatic estimation of width, contrast and tangential direction in 2D and 3D medical images. Comput. Vis. Image Underst. 2002, 86, 171–190. [Google Scholar]
- van der Laan, W.J.; van der Graaf, Y.; Eikelboom, B.C.; Mali, W.P.T.M. Quantitative assessment of aortic aneurysm geometry by three-dimensional reconstruction from computed tomography. Br. J. Surg. 1993, 80, 57–63. [Google Scholar]
- Li, D.; Wang, Z.; Chen, Y.; Jiang, R.; Ding, W.; Okumura, M. A survey on deep active learning: Recent advances and new frontiers. IEEE Trans. Neural Netw. Learn. Syst. 2024, 36, 5879–5899. [Google Scholar]
- Isensee, F.; Jaeger, P.F.; Kohl, S.A.A.; Petersen, J.; Maier-Hein, K.H. nnU-Net: A self-configuring method for deep learning-based biomedical image segmentation. Nat. Methods 2021, 18, 203–211. [Google Scholar] [CrossRef]
- Ma, J.; Li, F.; Wang, B. U-Mamba: Enhancing Long-range Dependency for Biomedical Image Segmentation. arXiv 2024, arXiv:2401.04722. [Google Scholar]
Trainer | ResEnc M |
---|---|
Model config | 3D fullres U-Net |
Total epochs | 300 |
Optimizer | SGD |
Initial learning rate (lr) | 1 × 10−2 |
Momentum | 0.99 |
Weight decay | 3 × 10−5 |
Lr decay schedule | PolyLR |
Loss function | Dice + Cross-Entropy |
Deep supervision | True |
Oversample foreground (%) | 0.33 |
Probabilistic oversampling | False |
Gradient clipping norm | 12 |
Configuration | Model | Aorta | Left Iliac | Right Iliac | Average |
---|---|---|---|---|---|
Post | 3D-UNet | 0.9664 | 0.8250 | 0.8627 | 0.8871 |
UMamba | 0.9678 | 0.8111 | 0.8321 | 0.8732 | |
Post + Inferred | 3D-UNet | 0.9626 | 0.8107 | 0.8507 | 0.8773 |
UMamba | 0.9617 | 0.8272 | 0.8260 | 0.8743 | |
Pre&Post | 3D-UNet | 0.9667 | 0.8167 | 0.8132 | 0.8655 |
UMamba | 0.9684 | 0.8222 | 0.8404 | 0.8770 | |
Pre&Post + Inferred | 3D-UNet | 0.9677 | 0.7991 | 0.7954 | 0.8541 |
UMamba | 0.9660 | 0.8152 | 0.7944 | 0.8585 |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2025 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (https://creativecommons.org/licenses/by/4.0/).
Share and Cite
Islam, M.; Tabassum, M.; Mayr, A.; Kremser, C.; Haltmeier, M.; Almar-Munoz, E. Uncertainty-Guided Active Learning for Access Route Segmentation and Planning in Transcatheter Aortic Valve Implantation. J. Imaging 2025, 11, 318. https://doi.org/10.3390/jimaging11090318
Islam M, Tabassum M, Mayr A, Kremser C, Haltmeier M, Almar-Munoz E. Uncertainty-Guided Active Learning for Access Route Segmentation and Planning in Transcatheter Aortic Valve Implantation. Journal of Imaging. 2025; 11(9):318. https://doi.org/10.3390/jimaging11090318
Chicago/Turabian StyleIslam, Mahdi, Musarrat Tabassum, Agnes Mayr, Christian Kremser, Markus Haltmeier, and Enrique Almar-Munoz. 2025. "Uncertainty-Guided Active Learning for Access Route Segmentation and Planning in Transcatheter Aortic Valve Implantation" Journal of Imaging 11, no. 9: 318. https://doi.org/10.3390/jimaging11090318
APA StyleIslam, M., Tabassum, M., Mayr, A., Kremser, C., Haltmeier, M., & Almar-Munoz, E. (2025). Uncertainty-Guided Active Learning for Access Route Segmentation and Planning in Transcatheter Aortic Valve Implantation. Journal of Imaging, 11(9), 318. https://doi.org/10.3390/jimaging11090318