Attention adaptive instance normalization style transfer for vascular segmentation using deep learning

doi:10.1007/s10489-023-05033-1

2023-11-02

PRE

Supriti Mulay *

Keerthi Ram

Mohanasankar Sivaprakasam

DOI：10.1007/s10489-023-05033-1

原文链接

原文求助

摘要

En 中文

Deep learning models have demonstrated substantial progress in medical image segmentation. However, these models require large datasets for training, which can prove to be clinically difficult. Medical imaging datasets exhibit domain shift problems due to different imaging techniques, scanners, and data privacy issues and, conventional deep neural networks lack generalization capabilities, as their effectiveness decreases with different data distributions. This paper presents a deep learning-based attention-adaptive instance normalization style transfer technique to address the challenges encountered when segmenting blood vessels. The proposed methodology combines adaptive instance normalization style transfer with a dense extreme inception network and convolution block attention module to achieve the best observed vessel segmentation performance. A simple yet effective method is proposed, and it improves the generalization performance of deep neural networks in vascular segmentation. The network is trained on natural images and tested on medical images, thereby overcoming the need for a large dataset or labelled ground truth to train for vessel segmentation. The proposed technique uses experimental results from five distinct medical datasets to demonstrate higher cross-domain generalization capabilities than the state-of-the-art baselines available in the current literature, and the segmentation performance is compared qualitatively and quantitatively with other models. The results demonstrate the feasibility of generalizing our approach to various datasets. This approach overcomes the constraints of traditional deep learning algorithms, which require enormous volumes of medical data along with manually-labelled ground truth. The predictions by the proposed approach are based on natural image training and can be reliably used to detect and identify cardiac and retinal abnormalities without prior medical imaging information.

Keyword:

Domain generalization

Style transfer

Vessel segmentation

Deep learning