Vision Transformers for Brain Tumor Classification

Eliott Simon, Alexia Briassouli

2022

Abstract

With the increasing amount of data gathered by healthcare providers, interest has been growing in Machine Learning, and more specifically in Deep Learning. Medical applications of machine learning range from the prediction of medical events, to computer-aided detection, diagnosis, and classification. This paper will investigate the application of State-of-the-Art (SoA) Deep Neural Networks in classifying brain tumors. We distinguish between several types of brain tumors, which are typically diagnosed and classified by experts using Magnetic Resonance Imaging (MRI). The most common benign tumors are gliomas and meningiomas, however there exist many more which vary in size and location. Convolutional Neural Networks (CNN) are the SoA deep learning technique for image processing tasks such as image segmentation and classification. However, a recently developed architecture for image classification, namely Vision Transformers, have been shown to outperform classical CNNs in efficiency, while requiring fewer computational resources. This work introduces using only Transformer networks in brain tumor classification for the first time, and compares their performance with CNNs. A significant difference between the two models, tested in this manner, is the lack of translational equivariance in Transformers, which the CNNs already have. Experiments for brain tumor classification on benchmark real-world datasets show they can achieve comparable or better performance, despite using limited training data.

Paper Citation

in Harvard Style

Simon E. and Briassouli A. (2022). Vision Transformers for Brain Tumor Classification. In Proceedings of the 15th International Joint Conference on Biomedical Engineering Systems and Technologies (BIOSTEC 2022) - Volume 2: BIOIMAGING; ISBN 978-989-758-552-4, SciTePress, pages 123-130. DOI: 10.5220/0010834300003123

in Bibtex Style

@conference{bioimaging22,
author={Eliott Simon and Alexia Briassouli},
title={Vision Transformers for Brain Tumor Classification},
booktitle={Proceedings of the 15th International Joint Conference on Biomedical Engineering Systems and Technologies (BIOSTEC 2022) - Volume 2: BIOIMAGING},
year={2022},
pages={123-130},
publisher={SciTePress},
organization={INSTICC},
doi={10.5220/0010834300003123},
isbn={978-989-758-552-4},
}

in EndNote Style

TY - CONF

JO - Proceedings of the 15th International Joint Conference on Biomedical Engineering Systems and Technologies (BIOSTEC 2022) - Volume 2: BIOIMAGING
TI - Vision Transformers for Brain Tumor Classification
SN - 978-989-758-552-4
AU - Simon E.
AU - Briassouli A.
PY - 2022
SP - 123
EP - 130
DO - 10.5220/0010834300003123
PB - SciTePress