Segmentation of Gray Matters and White Matters from Brain MRI data
Organizations: Department of Computer Science and Engineering, Waseda University, Shinjuku, Tokyo, Japan · Department of Neurosurgery, Teraoka Memorial Hospital, Fukuyama, Hiroshima, Japan · Department of Radiology, Nippon Medical School Hospital, Tokyo, Japan · Tokyo Rosai Hospital, Ota, Tokyo, Japan
Abstract
Accurate segmentation of brain tissues such as gray matter and white matter from magnetic resonance imaging is essential for studying brain anatomy, diagnosing neurological disorders, and monitoring disease progression. Traditional methods, such as FSL FAST, produce tissue probability maps but often require task-specific adjustments and face challenges with diverse imaging conditions. Recent foundation models, such as MedSAM, offer a prompt-based approach that leverages large-scale pretraining. In this paper, we propose a modified MedSAM model designed for multi-class brain tissue segmentation. Our preprocessing pipeline includes skull stripping with FSL BET, tissue probability mapping with FSL FAST, and converting these into 2D axial, sagittal, coronal slices with multi-class labels (background, gray matter, and white matter). We extend MedSAM's mask decoder to three classes, freezing the pre-trained image encoder and fine-tuning the prompt encoder and decoder. Experiments on the IXI dataset achieve Dice scores up to 0.8751. This work demonstrates that foundation models like MedSAM can be adapted for multi-class medical image segmentation with minimal architectural modifications. Our findings suggest that such models can be extended to more diverse medical imaging scenarios in future work.
Figures & tables
| Models | Best Overall Dice score |
|---|---|
| Axial model | 0.8717 |
| Sagittal model | 0.8604 |
| Coronal model | 0.8751 |
| Unified model (axial) | 0.8440 |
| Unified model (sagittal) | 0.8589 |
| Unified model (coronal) | 0.8742 |
| Models | Best Overall IoU score |
|---|---|
| Axial model | 0.7869 |
| Sagittal model | 0.7774 |
| Coronal model | 0.7935 |
| Unified model (axial) | 0.7601 |
| Unified model (sagittal) | 0.7750 |
| Unified model (coronal) | 0.7928 |
Medical Image Segmentation
Medical image segmentation delineates organs, lesions, and other structures in scans to support measurement, diagnosis, treatment planning, and research.
See all 788 papers