Attention-enhanced segmentation network for automated cerebral microbleed detection and burden assessment
Kwon Hwi Cho1, Jonghyun Jeon2, Seonggyu Kim3
1Department of Artificial Intelligence, Hanyang University, Seoul, Republic of Korea.
Introduction:
Cerebral microbleeds (CMBs) are small hemorrhagic lesions visible as hypointense foci on susceptibility-sensitive MRI and are established biomarkers of stroke risk and amyloid-related imaging abnormalities (ARIA-H) in patients receiving anti-amyloid therapy. However, automated detection remains challenging because true CMBs closely resemble veins, calcifications, and susceptibility artifacts. This visual ambiguity results in a persistent precision-recall trade-off, where models optimized for high sensitivity tend to generate excessive false positives, while precision-focused models risk missing clinically relevant lesions. To address this limitation, we propose an attention-enhanced segmentation framework designed to suppress confounding activations while preserving lesion sensitivity.
Methods:
We developed RLK-UNet with Convolutional Block Attention Modules (CBAM), a single-stage encoder-decoder architecture that redefines skip connections as context-filtered pathways. The encoder incorporates large 13×13 residual local kernel (RLK) convolutions to capture broad contextual information for distinguishing spherical microbleeds from elongated vascular structures. CBAM modules are embedded in all skip connections to selectively enhance lesion-relevant features and suppress irrelevant background responses before feature fusion. The model was trained and evaluated on a multi-site dataset of 506 T2*-GRE and SWI scans, with lesion-level detection assessed using precision, recall, F1-score, and average false positives per scan. Subject-level burden estimation was further evaluated using ARIA-H severity intervals.
Results:
The proposed model achieved state-of-the-art lesion-level performance, with a precision of 0.891, recall of 0.887, F1-score of 0.887, and a markedly reduced false positive rate of 0.83 per subject. Five-fold cross-validation demonstrated stable performance with minimal variance across splits. In lesions ≤3 mm, the model maintained strong detection performance (F1-score 0.869) while effectively controlling false positives. Cross-modality evaluation between T2*-GRE and SWI confirmed robust generalization. Ablation studies verified that CBAM significantly improved precision while preserving sensitivity, and Grad-CAM visualizations demonstrated more spatially focused and clinically interpretable attention patterns. Subject-level CMB counts strongly correlated with ground truth (Spearman ρ = 0.93), and severity classification aligned with ARIA-H intervals.
Conclusion:
RLK-UNet with CBAM provides a robust and interpretable solution for automated CMB detection by directly addressing false-positive propagation through attention-guided skip connections. The framework achieves balanced precision and sensitivity within a single-stage architecture and demonstrates reliable subject-level burden estimation aligned with clinically meaningful ARIA-H categories. These findings support its potential application in vascular risk stratification and treatment monitoring in patients undergoing anti-amyloid therapy.
Insights
This study introduces RLK-UNet with attention mechanisms to accurately detect cerebral microbleeds (CMBs) on MRI scans. The model improves precision and recall, aiding stroke risk assessment and monitoring anti-amyloid therapy.
Area of Science:
- Neuroimaging
- Artificial Intelligence
- Medical Diagnostics
Background:
- Cerebral microbleeds (CMBs) are crucial biomarkers for stroke risk and amyloid-related imaging abnormalities (ARIA-H), particularly in patients undergoing anti-amyloid therapy.
- Automated CMB detection is challenging due to visual similarities with other brain structures and artifacts, leading to a precision-recall trade-off in existing models.
Purpose of the Study:
- To develop an attention-enhanced deep learning framework for robust and accurate automated detection of cerebral microbleeds (CMBs).
- To address the limitations of current methods by improving both sensitivity and precision in CMB identification, thereby reducing false positives.
Main Methods:
- Development of RLK-UNet, an encoder-decoder architecture incorporating Convolutional Block Attention Modules (CBAM) and residual local kernel (RLK) convolutions.
- Training and evaluation on a multi-site dataset of 506 T2*-GRE and SWI scans, assessing lesion-level and subject-level performance metrics.
- Utilizing attention mechanisms within skip connections to filter irrelevant information and enhance focus on relevant lesion features.
Main Results:
- The RLK-UNet model achieved state-of-the-art performance with a precision of 0.891, recall of 0.887, and F1-score of 0.887, alongside a significantly reduced false positive rate of 0.83 per subject.
- The model demonstrated robust generalization across different MRI modalities (T2*-GRE and SWI) and maintained strong performance for small lesions (≤3 mm).
- Ablation studies confirmed the efficacy of CBAM in improving precision and sensitivity, with Grad-CAM visualizations showing interpretable attention patterns.
Conclusions:
- RLK-UNet with CBAM offers a reliable and interpretable solution for automated CMB detection, effectively balancing precision and sensitivity.
- The framework's ability to provide accurate subject-level burden estimation supports its clinical utility in vascular risk stratification and treatment monitoring for patients on anti-amyloid therapy.


