lxaw / mamba-tumor-seg

Segmentation of cancerous tumors using Mamba. Code, resources, and paper provided. We manage to make a small (42k param) model that can segment pretty well.

Overview:

This project uses Mamba blocks and some other modules to segment tumors. We also make a small (42,000 parameter) model that can segment decently, while showcasing other Mamba-based segmentation architectures. Code and description of methods are provided.

Dataset

We used the following dataset: https://www.kaggle.com/datasets/mateuszbuda/lgg-mri-segmentation

Example segmentations:

More can be found in our paper.

Model comparisons: