Automatic Brain Tumour Classification Based on Transfer Learning Models

Jinhong Zhu

Chapter 75 in Economic Management and Big Data Application:Proceedings of the 3rd International Conference, 2024, pp 850-857 from World Scientific Publishing Co. Pte. Ltd.

Abstract: It is time-consuming and error-prone to manually determine whether there is a brain tumour in an image. However, traditional automatic classification algorithms have certain limitations, which makes the automation of brain tumour classification still a challenging problem. In this article, a new method for the automatic classification of brain tumours is proposed, combining neural network models with transfer learning methods to improve or solve the problem of slow iteration and long time-consuming model generation, improve accuracy, and reduce parameters. In short, the convolutional neural network model (CNN) is combined with the method of transfer learning to achieve automatic image classification on the Brain Tumour Detection 2020 dataset provided by Model Whale. More specifically, during the experiment, Tensorflow was selected as the deep learning framework. First, the transfer learning method was used, and imagenet weights were used. Then, the Comparing model performance by changing the choice of the backbone network of the CNN. Select the accuracy rate as the evaluation index, compare the performance of the model, use binary cross-entropy as the loss function, and the optimizer uses adam. In this paper, three backbone networks, VGG, MobileNet and ResNet, are compared. Experimental results indicate that the automatic classification of brain tumours with the combination of the CNN model and transfer learning method has better performance and the VGG model has the best performance.

Keywords: Big Data; Information Management; Economic; Data Applications; Blockchain; E-commerce (search for similar items in EconPapers)
JEL-codes: C63 C8 O14 (search for similar items in EconPapers)
Date: 2024
References: Add references at CitEc
Citations:

Downloads: (external link)
https://www.worldscientific.com/doi/pdf/10.1142/9789811270277_0075 (application/pdf)
https://www.worldscientific.com/doi/abs/10.1142/9789811270277_0075 (text/html)
Ebook Access is available upon purchase.

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:wsi:wschap:9789811270277_0075

Ordering information: This item can be ordered from

Access Statistics for this chapter

More chapters in World Scientific Book Chapters from World Scientific Publishing Co. Pte. Ltd.
Bibliographic data for series maintained by Tai Tone Lim ().