Cross-Architectural Feature Fusion for Coronary Artery Stenosis Detection in X-ray Angiography
Authors: Haobo Zhu, Mojtaba Lashgari, Jason Chai, Giovanni Luigi De Maria, Vicente Grau, Abhirup Banerjee
Organizations: Institute of Biomedical Engineering, Department of Engineering Science, University of Oxford, Oxford, United Kingdom · Division of Cardiovascular Medicine, Radcliffe Department of Medicine, University of Oxford, Oxford, United Kingdom · Oxford Heart Centre, Oxford University Hospitals NHS Trust, Oxford, United Kingdom
Coronary artery disease remains a leading cause of mortality, with stenosis detection from invasive X-ray Coronary Angiography (XCA) critical for diagnosis. Manual interpretation is time-consuming, subjective, and variable, especially under low contrast and vessel overlap. Existing deep learning methods often rely solely on detection or segmentation, limiting performance. We propose a hybrid framework, Fusion-YOLOv11-L, combining a YOLOv11-L detector with a pretrained UNet++ via a channel attention gating module. This enables fusion of semantic and vessel-aware structural features, improving localisation of stenotic regions. The UNet++ is pretrained on ARCADE vessel segmentation dataset to provide anatomically informed representations. Experiments on the relabeled ARCADE (ARCADE*) dataset show improved performance over YOLOv11-L, achieving AP@50 of 0.585 and F1-score of 0.596. Cross-domain evaluation on a dataset from the Oxford Acute Myocardial Infarction (OxAMI) Study reveals performance degradation due to domain shift, mitigated using test-time adaptation and augmentation. Test-time augmentation improves AP@50 from 0.434 to 0.480, indicating better generalisability. These results show that cross-architecture feature fusion and lightweight adaptation improve robustness and generalisability of automated stenosis detection in XCA.
Figures & tables
Dataset
ARCADE*
OxAMI
Split
Train
Validation
Test
Test
# Image/ # Stenosis
1000/998
200/376
300/412
48/85
Table 1: Dataset summary.
Figure 1: A) Overall model architecture; B) Channel attention gating (CAG) module for cross-architecture feature fusion between the UNet++ decoder (UNet++ branch) and the YOLOv11-L encoder (YOLO branch).
Hyperparameter
#Epochs
LR
BBLG
CLG
Search Space
{50, 100}
{2e-4, 1e-4, 5e-5}
{10, 7.5, 5}
{0.5, 0.3}
Table 2: Hyperparameter search space.
Optimal Hyperparameters from Optuna
Model
LR
#Epochs
BBLG
CLG
AP@50
YOLOv8-L
1e-4
100
7.5
0.5
0.579
YOLOv11-L
1e-4
100
7.5
0.3
0.574
Fusion-YOLOv11-L (Ours)
1e-4
100
5
0.3
0.585
Table 3: Comparison of AP@50 for different models on ARCADE* test set.
Confidence
Precision
Recall
F1-score
0.250
0.665
0.529
0.589
0.200
0.630
0.566
0.596
0.150
0.564
0.592
0.578
0.100
0.480
0.621
0.542
0.050
0.383
0.682
0.490
Table 4: Evaluation of the performance of Fusion-YOLO-v11 across confidence thresholds on the ARCADE* test set.
Model
Base
TTAd Only
TTAg Only
TTAd + TTAg
YOLOv11-L
0.339
0.392
0.383
0.413
Fusion-YOLOv11-L
0.434
0.432
0.480
0.470
Table 5: Generalisability of YOLOv11-L and Fusion-YOLOv11-L across inference variants on the OxAMI dataset (AP@50).
Original
TTAd Only
TTAg Only
TTAd + TTAg
Confidence
P
R
F1
P
R
F1
P
R
F1
P
R
F1
0.250
0.625
0.235
0.342
0.658
0.294
0.407
0.638
0.353
0.455
0.612
0.353
0.448
0.200
0.650
0.306
0.416
0.628
0.318
0.422
0.620
0.365
0.459
0.561
0.376
0.451
0.150
0.622
0.329
0.431
0.588
0.353
0.441
0.574
0.412
0.479
0.530
0.412
0.464
0.100
0.586
0.400
0.476
0.585
0.447
0.507
0.551
0.506
0.528
0.500
0.541
0.520
0.050
0.460
0.471
0.465
0.467
0.506
0.486
0.409
0.553
0.470
0.384
0.565
0.457
Table 6: Generalisability of detection performance across confidence thresholds and inference variants on the OxAMI dataset.
Figure 2: Comparison of ground truth (green) and predictions (red) on example XCA images from ARCADE* dataset (in-domain test dataset).
Figure 3: Comparison of ground truth (green) and predictions (red) on example XCA images from OxAMI dataset (OOD test) using Fusion-YOLOv11-L with base model without TTAd or TTAg, model with TTAd only, model with TTAg only, and model with TTAd + TTAg.
Coronary angiography uses medical imaging to visualize heart vessels and assess disease, supporting automated diagnosis, measurement, and procedural guidance.
Multi-view reasoning in coronary X-ray angiography is inherently a cross-projection geometric problem, yet automated report generation in this setting remains largely unexplored. The 3D vascular topology leads to projection-dependent branch overlap and foreshortening, rendering single-view modeling fundamentally incomplete and unstable for lesion localization and stenosis grading. Although multi-view fusion appears promising, learning anatomically consistent fusion from real angiograms is impeded by a critical limitation: cross-view alignment is unobservable and cannot be explicitly supervised. Consequently, conventional fusion relies on implicit correlations rather than verified anatomical correspondence. We address this by reformulating multi-view stenosis reporting as an alignment-constrained aggregation problem. A controllable synthetic angiography generation strategy is introduced to expose geometry-derived patch-level correspondence supervision unavailable in real data. An anatomy-correspondence module learns cross-view correspondence matrices that explicitly align auxiliary features within the main-view coordinate space prior to fusion, thereby constraining evidence aggregation to anatomically consistent regions. Experiments on synthetic data and zero-shot transfer to real angiograms show that this alignment-constrained design improves correspondence consistency and structured stenosis reporting compared to single-view modeling and conventional multi-view fusion methods. The code will be publicly available upon publication.
Chen Jia, Baochang Zhang, Fatia Kusuma Dewi +4
Computer Aided Medical Procedures, Technical University of Munich, Munich, Germany · relAI – Konrad Zuse School of Excellence in Reliable AI, Munich, Germany · German Heart Center Munich, Munich, Germany +3
Accurate segmentation of coronary arteries in X-ray angiography videos is essential for quantitative coronary analysis and image-guided interventions. However, accurate segmentation remains challenging because coronary vessels are thin and exhibit low contrast, while the presence of catheters, guidewires, and complex anatomical background structures can further interfere with vessel delineation. Existing U-Net- and Transformer-based models provide strong baselines, but their shared feature-adaptation pathways may be insufficient for heterogeneous angiographic appearances. In this paper, we propose a prompt-free mixture-of-experts (MoE) feature adapter for binary coronary artery segmentation. Built upon parameter-efficient Vision Transformer adapters, the proposed method uses multiple lightweight experts with input-dependent top-k routing to adaptively refine vessel-related features while limiting active computational cost. Experiments on MOSXAV and external evaluation on XACV show that the proposed method outperforms representative baselines and improves cross-dataset generalisation. These results suggest that MoE-based adapter learning is effective for robust coronary artery segmentation in X-ray angiography videos.
Lin Xi, Yingliang Ma
University College London, United Kingdom · University of East Anglia, United Kingdom · King’s College London, United Kingdom
Invasive Coronary Angiography (ICA) is the clinical gold standard for the assessment of coronary artery disease. However, its interpretation remains subjective and prone to intra- and inter-operator variability. In this work, we introduce ODySSeI: an Open-source end-to-end framework for automated Detection, Segmentation, and Severity estimation of lesions in ICA images. ODySSeI integrates deep learning-based lesion detection and lesion segmentation models trained using a novel Pyramidal Augmentation Scheme (PAS) to enhance robustness and real-time performance across diverse patient cohorts (2149 patients from Europe, North America, and Asia). Furthermore, we propose a quantitative coronary angiography-free Lesion Severity Estimation (LSE) technique that directly computes the Minimum Lumen Diameter (MLD) and diameter stenosis from the predicted lesion geometry. Extensive evaluation on both in-distribution and out-of-distribution clinical datasets demonstrates ODySSeI's strong generalizability. Our PAS yields large performance gains in highly complex tasks as compared to relatively simpler ones, notably, a 2.5-fold increase in lesion detection performance versus a 1-3% increase in lesion segmentation performance over their respective baselines. Our LSE technique achieves high accuracy, with predicted MLD values differing by only ± 2-3 pixels from the corresponding ground truths. On average, ODySSeI processes a raw ICA image within only a few seconds on a CPU and in a fraction of a second on a GPU and is available as a plug-and-play web interface at swisscardia.epfl.ch. Overall, this work establishes ODySSeI as a comprehensive and open-source framework which supports automated, reproducible, and scalable ICA analysis for real-time clinical decision-making.
Anand Choudhary, Xiaowu Sun, Thabo Mahendiran +8
LTS4 Laboratory, EPFL, Lausanne, Switzerland · Chair of Mathematical Data Science, EPFL, Lausanne, Switzerland · School of AIAC, Xi’an Jiaotong-Liverpool University, Suzhou, China +3