HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Attention Enriched Deep Learning Model for Breast Tumor Segmentation in Ultrasound Images

Aleksandar Vakanski Min Xian Phoebe Freer

Attention Enriched Deep Learning Model for Breast Tumor Segmentation in Ultrasound Images

Abstract

Incorporating human domain knowledge for breast tumor diagnosis is challenging, since shape, boundary, curvature, intensity, or other common medical priors vary significantly across patients and cannot be employed. This work proposes a new approach for integrating visual saliency into a deep learning model for breast tumor segmentation in ultrasound images. Visual saliency refers to image maps containing regions that are more likely to attract radiologists visual attention. The proposed approach introduces attention blocks into a U-Net architecture, and learns feature representations that prioritize spatial regions with high saliency levels. The validation results demonstrate increased accuracy for tumor segmentation relative to models without salient attention layers. The approach achieved a Dice similarity coefficient of 90.5 percent on a dataset of 510 images. The salient attention model has potential to enhance accuracy and robustness in processing medical images of other organs, by providing a means to incorporate task-specific knowledge into deep learning architectures.

Code Repositories

Benchmarks

BenchmarkMethodologyMetrics
lesion-segmentation-on-bus-2017-dataset-bSalient Attention U-Net
Dice Score: 0.7341
tumor-segmentation-on-bus-2017-dataset-bSalient Attention U-Net
Dice Score: 0.7341

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Attention Enriched Deep Learning Model for Breast Tumor Segmentation in Ultrasound Images | Papers | HyperAI