arrow
Return

Clustering-enhanced active learning with dynamic sampling for brain tumor classification

delete2026-01-31
delete0
PRE
AI
Y
Yawen Fan *
X
Xiang Wang
Z
Zhen Yue
X
Xinchen Zhang
M
Mingkai Chen
J
Jianxin Chen
DOI:10.1016/j.bspc.2026.109715delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Automated classification of brain tumors is essential for reliable diagnosis and effective treatment planning. However, deep learning-based methods require large, well-labeled MRI datasets, which can be expensive, time-consuming, and challenging to obtain in clinical settings. Moreover, real-world datasets often exhibit severe class imbalance and inter-subject variability, both of which can compromise model robustness and limit generalization to unseen cases. In this paper, we introduce a novel dynamic active learning framework enhanced by clustering for brain tumor classification. First, the proposed framework extracts high-level features of MRI images by a self-supervised learning method, which are then clustered to form a multi-class data pool, providing a pre-classification of the samples. To reduce annotation effort while maintaining model performance, the framework dynamically selects the most informative samples from each cluster by jointly considering prediction uncertainty and cluster diversity. Additionally, we have constructed a high-quality brain tumor MRI dataset that includes three tumor types: glioma, metastatic tumor, and diffuse large B-cell lymphoma. Notably, the latter is scarce in existing public datasets. Extensive experiments on both public and private datasets show that the proposed method achieves competitive performance using only a small portion of labeled data. Also, on an external test set, the method obtained an average accuracy of 0.92. All these results suggest that our method offers a practical and efficient solution for MRI-based brain tumor classification in real-world clinical settings.

Journal

Biomedical Signal Processing and Control cover
Biomedical Signal Processing and Control
IF:
4.9
Papers:
9.8K
Citations:
2.4W

Organization

N
Nanjing Medical University
Scholars:
4.7W
Papers: 2.8W
Citations: 3.5K
N
Nanjing University of Posts and Telecommunications
Scholars:
2.4K
Papers: 969
Citations: 1.2W