Brain tumor classification from magnetic resonance imaging (MRI) is a common direction in the application of deep learning to medical image analysis. However, this task often faces a significant challenge: class imbalance. Some tumor types appear frequently in datasets, while others are rare, which can lead to models that are biased-overfitting on the more prevalent classes and struggling with underrepresented ones. In this study, we introduce a method that modifies the standard CrossEntropy loss function by incorporating class weights based on the concept of effective sample size. Instead of simply using raw sample counts to adjust for imbalance, our approach attempts to better reflect the actual contribution of each class during training. This helps mitigate the model’s tendency to favor dominant classes. We tested this approach on two modern deep neural network architectures, aiming to evaluate its ability to enhance classification performance under label imbalance—without altering model structures or relying on resampling strategies. The goal is to offer a lightweight, adaptable solution that can be easily integrated into existing medical image analysis pipelines.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Improving Brain MRI Tumor Classification with Class-Balanced Loss: A Comparative Study on ConvNeXtTiny and InceptionV3

  • Nguyen Nhut Hung,
  • Pham Truong Hoang Duc

摘要

Brain tumor classification from magnetic resonance imaging (MRI) is a common direction in the application of deep learning to medical image analysis. However, this task often faces a significant challenge: class imbalance. Some tumor types appear frequently in datasets, while others are rare, which can lead to models that are biased-overfitting on the more prevalent classes and struggling with underrepresented ones. In this study, we introduce a method that modifies the standard CrossEntropy loss function by incorporating class weights based on the concept of effective sample size. Instead of simply using raw sample counts to adjust for imbalance, our approach attempts to better reflect the actual contribution of each class during training. This helps mitigate the model’s tendency to favor dominant classes. We tested this approach on two modern deep neural network architectures, aiming to evaluate its ability to enhance classification performance under label imbalance—without altering model structures or relying on resampling strategies. The goal is to offer a lightweight, adaptable solution that can be easily integrated into existing medical image analysis pipelines.