cs.CVAug 12, 2026

A Neighborhood Attention Transformer Network for Enhanced 3D Segmentation of the Left Anterior Descending Artery

Authors: Rafi Ibn SultanChengyin LiYiannos DemetriouAhmed I. GhanemJoshua P. KimJustine CunninghamHassan Bagher-EbadianDongxiao Zhu+1 more

Organizations: Department of Computer Science, Wayne State University, Detroit, MI, 48202, USA · Department of Radiation Oncology, Henry Ford Health, Detroit, MI, 48202, USA · Department of Clinical Oncology, Alexandria University, Alexandria, 21526, Egypt · Department of Radiology, Michigan State University, E. Lansing, MI, 48824, USA · College of Osteopathic Medicine, Michigan State University, E. Lansing, MI, 48824, USA · Department of Physics, Oakland University, Rochester, MI, 48309, USA · Institute for AI and Data Science, Wayne State University, Detroit, MI, 48202, USA

Abstract

Background: Accurate segmentation of the Left Anterior Descending (LAD) artery in 3D free-breathing, non-contrast CT is critical for cardiac dose sparing in thoracic radiotherapy. The LAD is extremely small, has poor soft-tissue contrast, and varies substantially across patients; even manual contours show limited inter-observer agreement, underscoring the ambiguity of the vessel boundaries. Purpose: To develop a transformer-based framework that improves LAD delineation in low-contrast, imbalanced CT through local-global context modeling and uncertainty-guided optimization. Methods: We propose NA-UNETR, a 3D transformer-based segmentation model whose Neighborhood Attention (NA) and Dilated NA (DiNA) blocks jointly capture fine structural detail and long-range context. Given the scarcity of annotated LAD data, the model is pretrained on 1,000 CTA volumes of general coronary anatomy and fine-tuned with LoRA-based parameter-efficient adaptation on 20 free-breathing institutional CT scans. A composite Dice-Focal and Hausdorff loss, dynamically balanced via homoscedastic uncertainty, improves overlap and boundary accuracy. Results: NA-UNETR reached 45.64% Dice, 38.16 mm HD95, and 10.01 mm ASD, improving Dice by 3.10 percentage points over nnU-Net and reducing HD95 by 2.96 mm relative to Swin UNETR, with the strongest boundary accuracy among all models and improved centerline stability. On ImageCAS it achieved 79.49% Dice, 8.89 mm HD95, and 1.02 mm ASD. Ablations confirmed that residual blocks, variable kernels, and uncertainty-weighted loss each contributed. Conclusions: NA-UNETR balances local precision and global context for thin, low-contrast LAD structures, offering a computationally efficient framework for substructure-level cardiac segmentation in radiotherapy planning.

Explore similar work

CardsList