An L2-Normalized Spatial Attention Network For Accurate And Fast Classification Of Brain Tumors In 2D T1-Weighted CE-MRI Images

Grace Billingsley, Julia Dietlmeier, Vivek Narayanaswamy, Andreas Spanias, Noel E. OConnor
8/1/2023
eess.IVcs.CV

Official code: juliadietlmeier/MRI_image_classification (8 stars, last commit 2023-06-26).

Abstract

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

View on arXivView PDF

Code Implementations(1)

Classification of brain tumors (Meningioma, Glioma, Pituitary) in 2D T1-weighted CE-MRI images

80PythonJul 10, 20223 years ago

Similar Papers

Cite this paper

@article{billingsley2023lnormalized,
  title  = {An L2-Normalized Spatial Attention Network For Accurate And Fast Classification Of Brain Tumors In 2D T1-Weighted CE-MRI Images},
  author = {Grace Billingsley and Julia Dietlmeier and Vivek Narayanaswamy and Andreas Spanias and Noel E. OConnor},
  year   = {2023},
  eprint = {2308.00491},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2308.00491v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2308.00491.svg)](https://codeofpaper.com/paper/2308.00491)

Discussion