Deep Learning-Based Key Frame Recognition Algorithm for Adrenal Vascular in X-Ray Imaging

Abstract

Adrenal vein sampling is required for the staging diagnosis of primary aldosteronism, and the frames in which the adrenal veins are presented are called key frames. Currently, the selection of key frames relies on the doctor's visual judgement which is time-consuming and laborious. This study proposes a key frame recognition algorithm based on deep learning. Firstly, wavelet denoising and multi-scale vessel-enhanced filtering are used to preserve the morphological features of the adrenal veins. Furthermore, by incorporating the self-attention mechanism, an improved recognition model called ResNet50-SA is obtained. Compared with commonly used transfer learning, the new model achieves 97.11% in accuracy, precision, recall, F1, and AUC, which is superior to other models and can help clinicians quickly identify key frames in adrenal veins.

Keywords

• transfer learning
• self-attention mechanism
• wavelet transform
• key frame recognition
• adrenal angiography