A Transfer Learning Based Active Learning Framework for Brain Tumor Classification

TL;DR

This paper introduces a transfer learning and active learning framework for brain tumor classification that reduces annotation costs and maintains high accuracy, demonstrating improved AUC and robustness on MRI datasets.

Contribution

It presents a novel transfer learning-based active learning approach that effectively reduces labeling effort while achieving high classification performance in brain tumor grading.

Findings

Achieved 82.89% AUC on test data, 2.92% higher than baseline.

Reduced labeling cost by at least 40%.

Demonstrated robustness with balanced dataset results.

Abstract

Brain tumor is one of the leading causes of cancer-related death globally among children and adults. Precise classification of brain tumor grade (low-grade and high-grade glioma) at early stage plays a key role in successful prognosis and treatment planning. With recent advances in deep learning, Artificial Intelligence-enabled brain tumor grading systems can assist radiologists in the interpretation of medical images within seconds. The performance of deep learning techniques is, however, highly depended on the size of the annotated dataset. It is extremely challenging to label a large quantity of medical images given the complexity and volume of medical data. In this work, we propose a novel transfer learning based active learning framework to reduce the annotation cost while maintaining stability and robustness of the model performance for brain tumor classification. We employed a 2D slice-based approach to train and finetune our model on the Magnetic Resonance Imaging (MRI) training dataset of 203 patients and a validation dataset of 66 patients which was used as the baseline. With our proposed method, the model achieved Area Under Receiver Operating Characteristic (ROC) Curve (AUC) of 82.89% on a separate test dataset of 66 patients, which was 2.92% higher than the baseline AUC while saving at least 40% of labeling cost. In order to further examine the robustness of our method, we created a balanced dataset, which underwent the same procedure. The model achieved AUC of 82% compared with AUC of 78.48% for the baseline, which reassures the robustness and stability of our proposed transfer learning augmented with active learning framework while significantly reducing the size of training data.