Explainable attention based breast tumor segmentation using a combination of UNet, ResNet, DenseNet, and EfficientNet models
Islamic Azad University South Tehran Branch · University of Tehran · +3 more institutions
Abstract
This study utilizes the Breast Ultrasound Image (BUSI) dataset to present a deep learning technique for breast tumor segmentation based on a modified UNet architecture. To improve segmentation accuracy, the model integrates attention mechanisms, such as the Convolutional Block Attention Module (CBAM) and Non-Local Attention, with advanced encoder architectures, including ResNet, DenseNet, and EfficientNet. These attention mechanisms enable the model to focus more effectively on relevant tumor areas, resulting in significant performance improvements. Models incorporating attention mechanisms outperformed those without, as reflected in superior evaluation metrics. The effects of Dice Loss and Binary…
Citation impact
- FWCI
- 96.34
- Percentile
- 100%
- References
- 71
Authors
5Topics & keywords
- Computer science
- Segmentation
- Artificial intelligence
- Cross entropy
- Interpretability
- Deep learning
- Dice
- Encoder