Single level UNet3D with multipath residual attention block for brain tumor segmentation

Atrous convolution and attention have improved the performance of the UNet architecture for segmentation purposes. However, a perfect combination of atrous convolution and attention to improve brain tumor segmentation performance is still an interesting challenge. In this paper, we propose UNet arch...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Veröffentlicht in:Journal of King Saud University. Computer and information sciences 2022-06, Vol.34 (6), p.3247-3258
Hauptverfasser: Akbar, Agus Subhan, Fatichah, Chastine, Suciati, Nanik
Format: Artikel
Sprache:eng
Schlagworte:
Online-Zugang:Volltext
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
Beschreibung
Zusammenfassung:Atrous convolution and attention have improved the performance of the UNet architecture for segmentation purposes. However, a perfect combination of atrous convolution and attention to improve brain tumor segmentation performance is still an interesting challenge. In this paper, we propose UNet architecture with the addition of attention in the skip connection and the replacement of the processing block with two atrous convolution sequences connected to the attention unit combined with one residual path called the Multipath Residual Attention Block (MRAB). The architecture was trained using the Brain Tumor Segmentation(BraTS) 2018, 2019, 2020, and 2021 challenge datasets. The ensembled model was validated online and obtained dice scores of 77.71%, 79.77%, 89.59% for BraTS2018, 74.91%, 80.98%, 88.48% for BraTS2019, 72.91%, 80.19%, 88.57% for BraTS2020, and 77.73%, 82.19%, 89.33% for BraTS2021 validation datasets for Enhanced Tumor(ET), Tumor Core(TC), and Whole Tumor(WT) areas, respectively. These dice score performances outperformed state-of-the-art brain tumor segmentation architectures and promised to be developed for clinical application.
ISSN:1319-1578
2213-1248
DOI:10.1016/j.jksuci.2022.03.022