A novel enhanced softmax loss function for brain tumour detection using deep learning

Sunil Maharjan, Abeer Alsadoon, P. W.C. Prasad, Thair Al-Dalain, Omar Hisham Alsadoon

Research output: Contribution to journalArticlepeer-review

6 Citations (Scopus)

Abstract

Background and Aim: In deep learning, the sigmoid function is unsuccessfully used for the multiclass classification of the brain tumour due to its limit of binary classification. This study aims to increase the classification accuracy by reducing the risk of overfitting problem and supports multi-class classification. The proposed system consists of a convolutional neural network with modified softmax loss function and regularization. Results: Classification accuracy for the different types of tumours and the processing time were calculated based on the probability score of the labeled data and their execution time. Different accuracy values and processing time were obtained when testing the proposed system using different samples of MRI images. The result shows that the proposed solution is better compared to the other systems. Besides, the proposed solution has higher accuracy by almost 2 % and less processing time of 40∼50 ms compared to other current solutions. Conclusion: The proposed system focused on classification accuracy of the different types of tumours from the 3D MRI images. This paper solves the issues of binary classification, the processing time, and the issues of overfitting of the data.

Original languageEnglish
Article number108520
Pages (from-to)1-11
Number of pages11
JournalJournal of Neuroscience Methods
Volume330
Early online date11 Nov 2019
DOIs
Publication statusPublished - 15 Jan 2020

Fingerprint Dive into the research topics of 'A novel enhanced softmax loss function for brain tumour detection using deep learning'. Together they form a unique fingerprint.

Cite this