This project focuses on segmenting low-grade gliomas from brain MRI scans using deep learning models.
- Models:
- ResNet18 β tumor presence classifier.
- U-Net β main segmentation model (31M parameters).
- U-Net Mini β lightweight version (1.8M parameters).
- Loss Functions:
BCEWithLogitsLoss(with class weighting).- Dice Loss for evaluation.
- ResNet18: 98% accuracy on tumor presence classification.
- Segmentation Performance (Dice coefficient):
- ResNet18 + U-Net β 93.1% (all images), 79.1% (glioma only).
- ResNet18 + U-Net Mini β 89.9% (all images), 68.9% (glioma only).
- Competitive with Kaggle solutions (best reported Dice β 96%).
- Kaggle Dataset
- Siddique et al., U-Net and Its Variants for Medical Image Segmentation, IEEE Access, 2021.