English

Reproducible Evaluation of Data Augmentation and Loss Functions for Brain Tumor Segmentation

Computer Vision and Pattern Recognition 2025-10-13 v1 Machine Learning

Abstract

Brain tumor segmentation is crucial for diagnosis and treatment planning, yet challenges such as class imbalance and limited model generalization continue to hinder progress. This work presents a reproducible evaluation of U-Net segmentation performance on brain tumor MRI using focal loss and basic data augmentation strategies. Experiments were conducted on a publicly available MRI dataset, focusing on focal loss parameter tuning and assessing the impact of three data augmentation techniques: horizontal flip, rotation, and scaling. The U-Net with focal loss achieved a precision of 90%, comparable to state-of-the-art results. By making all code and results publicly available, this study establishes a transparent, reproducible baseline to guide future research on augmentation strategies and loss function design in brain tumor segmentation.

Keywords

Cite

@article{arxiv.2510.08617,
  title  = {Reproducible Evaluation of Data Augmentation and Loss Functions for Brain Tumor Segmentation},
  author = {Saumya B},
  journal= {arXiv preprint arXiv:2510.08617},
  year   = {2025}
}

Comments

Code and results available at https://github.com/Saumya4321/2d-brain-tumor-segmentation