Feature reconstruction-based self-supervised learning model for vessel segmentation
- VernacularTitle:基于特征重建的血管分割自监督学习模型
- Author:
Bowen ZHOU
1
;
Hui SUN
2
;
Kaiyue DIAO
1
;
Qing XIA
2
;
Kang LI
1
Author Information
1. West China Biomedical Big Data Center, West China Hospital, Sichuan University, Chengdu, 610041, P. R. China
2. West China Hospital-Sensetime Joint Laboratory, Chengdu, 610213, P. R. China
- Publication Type:Journal Article
- Keywords:
Vessel segmentation;
self-supervised learning;
histogram of oriented gradients
- From:
Chinese Journal of Clinical Thoracic and Cardiovascular Surgery
2025;32(06):779-784
- CountryChina
- Language:Chinese
-
Abstract:
Objective To propose an innovative self-supervised learning method for vascular segmentation in computed tomography angiography (CTA) images by integrating feature reconstruction with masked autoencoding. Methods A 3D masked autoencoder-based framework was developed, where in 3D histogram of oriented gradients (HOG) was utilized for multi-scale vascular feature extraction. During pre-training, random masking was applied to local patches of CTA images, and the model was trained to jointly reconstruct original voxels and HOG features of masked regions. The pre-trained model was further fine-tuned on two annotated datasets for clinical-level vessel segmentation. Results Evaluated on two independent datasets (30 labeled CTA images each), our method achieved superior segmentation accuracy to the supervised neural network U-Net (nnU-Net) baseline, with Dice similarity coefficients of 91.2% vs. 89.7% (aorta) and 84.8% vs. 83.2% (coronary arteries). Conclusion The proposed self-supervised model significantly reduces manual annotation costs without compromising segmentation precision, showing substantial potential for enhancing clinical workflows in vascular disease management.
