AUTOMATED SEGMENTATION OF ULTRASOUND MEDICAL IMAGES USING THE ATTENTION U-NET MODEL

Abstract

Abstract The article deals with the method of automated semantic segmentation of ultrasound medical images using the Attention U-Net deep learning model. The advantages of using Attention blocks in neural network architectures for segmentation tasks are analyzed. To test the described algorithms, the Breast Ultrasound Images training dataset was chosen. The method described in the article allows for automating the process of detecting and preliminary classification of breast tumors based on the analysis of ultrasound images. As a result of training the Attention U-Net model, the Mean IOU value of 49.2% was obtained on the test set. The network can automatically classify the detected neoplasm as benign or malignant with an F1 Score of 0.87. The results indicate the prospects of using the Attention U-Net model in the tasks of analyzing ultrasound medical images. Ways to further improve the considered method are proposed.