The effect of input size on the accuracy of a convolutional neural network performing brain tumor detection

Zirui Zhao

doi:10.1117/12.2672694

1 March 2023 The effect of input size on the accuracy of a convolutional neural network performing brain tumor detection

Zirui Zhao

Author Affiliations +

Proceedings Volume 12596, International Conference on Mechatronics Engineering and Artificial Intelligence (MEAI 2022); 1259617 (2023) https://doi.org/10.1117/12.2672694
Event: International Conference on Mechatronics Engineering and Artificial Intelligence (MEAI 2022), 2022, Changsha, China

Abstract

A brain tumor can negatively affect basic bodily functions and when malignant can result in low survival rates. Many studies were conducted to detect and classify brain tumors in MRI images using a convolutional neural network (CNN) and other techniques like image preprocessing and transfer learning. However, few studies have explored the effect of specific hyperparameters on the performance of such CNNs. This study aims to investigate how the input size affects the CNN’s accuracy in brain tumor detection. Brain MRI datasets were collected and split into training, validation, and test sets. Four models with identical architectures but different input sizes of 256px×256px×3, 224px×224px×3, 128px×128px×3, and 64px×64px×3 were built using TensorFlow Keras, trained on the training set with data augmentation, and evaluated using the test sets. Of these four models, the one with 64px as input size has the best performance, yielding the highest test accuracy, 99.16%, and lowest test loss, 0.0282, whereas the 224px model has the worst performance, with the lowest accuracy, 98.06%, and highest loss, 0.0976. Accordingly, it appears that larger input sizes do not necessarily result in higher accuracy of the CNN performing brain tumor detection. Future studies on this topic may consider using a smaller input size, not only maintaining high accuracy but also significantly reducing the required time to train and the space to save the model.

Citation Download Citation

Zirui Zhao "The effect of input size on the accuracy of a convolutional neural network performing brain tumor detection", Proc. SPIE 12596, International Conference on Mechatronics Engineering and Artificial Intelligence (MEAI 2022), 1259617 (1 March 2023); https://doi.org/10.1117/12.2672694

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
12 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Tumors

Brain

Data modeling

Performance modeling

Cancer detection

Neuroimaging

Image classification

Show All Keywords

Keywords/Phrases

Search In:

Publication Years