Multi-Class Brain Tumor Classification Using Advanced Deep Learning Models: A Comparative Study

· Source: Computer Vision and Pattern Recognition · Field: Science & Research — Health & Medical Research, Artificial Intelligence & Machine Learning · Depth: Advanced, quick

Summary

A comparative study evaluated five convolutional neural network (CNN) architectures for multi-class brain tumor classification using approximately 10,000 clinically-sourced MRI images. The research tested a customized baseline model, VGG16, VGG19, DenseNet121, and EfficientNetB0 within an identical experimental framework, measuring overall accuracy and tumor-wise recall. EfficientNetB0 demonstrated the highest overall classification accuracy at 95%, outperforming VGG16 (94.37%), VGG19 (92.29%), DenseNet121 (90.91%), and the customized CNN (78.00%). Crucially, EfficientNetB0 achieved an 89% recall rate for meningiomas, a significant improvement over simple CNNs' ~20% recall for these often subtle tumors. The study also noted that the deeper VGG19 performed worse than VGG16, suggesting architectural efficiency is more critical than depth for medical image analysis. EfficientNetB0 provides an optimal balance of accuracy, parameter count, and clinical performance.

Key takeaway

For Machine Learning Engineers developing medical imaging solutions, you should prioritize EfficientNetB0 for multi-class brain tumor classification from MRI scans. Its 95% overall accuracy and 89% meningioma recall significantly surpass other CNNs, offering an optimal balance of performance and efficiency. When evaluating models, focus on tumor-wise recall alongside overall accuracy to ensure clinical relevance, especially for subtle pathologies.

Key insights

EfficientNetB0 significantly improves multi-class brain tumor classification from MRI images, especially for subtle meningiomas, outperforming other CNNs.

Principles

Method

The study compared five CNNs (custom, VGG16, VGG19, DenseNet121, EfficientNetB0) on ~10,000 MRI images, measuring overall accuracy and tumor-wise recall within an identical framework.

In practice

Topics

Best for: Computer Vision Engineer, AI Scientist, Research Scientist, Machine Learning Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computer Vision and Pattern Recognition.