TY - GEN
T1 - AN L2-Normalized Spatial Attention Network for Accurate and Fast Classification of Brain Tumors in 2D T1-Weighted CE-MRI Images
AU - Billingsley, Grace
AU - Dietlmeier, Julia
AU - Narayanaswamy, Vivek
AU - Spanias, Andreas
AU - O'Connor, Noel E.
N1 - Publisher Copyright:
© 2023 IEEE.
PY - 2023
Y1 - 2023
N2 - We propose an accurate and fast classification network for classification of brain tumors in MRI images that outperforms all lightweight methods investigated in terms of accuracy. We test our model on a challenging 2D T1-weighted CE-MRI dataset containing three types of brain tumors: Meningioma, Glioma and Pituitary. We introduce an l2-normalized spatial attention mechanism that acts as a regularizer against overfitting during training. We compare our results against the state-of-the-art on this dataset and show that by integrating l2-normalized spatial attention into a baseline network we achieve a performance gain of 1.79 percentage points. Even better accuracy can be attained by combining our model in an ensemble with the pretrained VGG16 at the expense of execution speed. Our code is publicly available at https://github.com/juliadietlmeier/MRI-image-classification.
AB - We propose an accurate and fast classification network for classification of brain tumors in MRI images that outperforms all lightweight methods investigated in terms of accuracy. We test our model on a challenging 2D T1-weighted CE-MRI dataset containing three types of brain tumors: Meningioma, Glioma and Pituitary. We introduce an l2-normalized spatial attention mechanism that acts as a regularizer against overfitting during training. We compare our results against the state-of-the-art on this dataset and show that by integrating l2-normalized spatial attention into a baseline network we achieve a performance gain of 1.79 percentage points. Even better accuracy can be attained by combining our model in an ensemble with the pretrained VGG16 at the expense of execution speed. Our code is publicly available at https://github.com/juliadietlmeier/MRI-image-classification.
KW - attention mechanisms
KW - brain tumor classification
KW - Deep learning
KW - MRI image analysis
UR - http://www.scopus.com/inward/record.url?scp=85180750194&partnerID=8YFLogxK
UR - http://www.scopus.com/inward/citedby.url?scp=85180750194&partnerID=8YFLogxK
U2 - 10.1109/ICIP49359.2023.10222887
DO - 10.1109/ICIP49359.2023.10222887
M3 - Conference contribution
AN - SCOPUS:85180750194
T3 - Proceedings - International Conference on Image Processing, ICIP
SP - 1895
EP - 1899
BT - 2023 IEEE International Conference on Image Processing, ICIP 2023 - Proceedings
PB - IEEE Computer Society
T2 - 30th IEEE International Conference on Image Processing, ICIP 2023
Y2 - 8 October 2023 through 11 October 2023
ER -