ClinicalFMamba

PyTorch Implementation of ClinicalFMamba: Advancing Clinical Assessment using Mamba-based Multimodal Neuroimaging Fusion, MICCAI MLMI 2025

Arxiv Link: here

Setup (Run on Google Colab)

Install Mamba

cd causal-conv1d
python setup.py install

cd ../mamba
python setup.py install

Other useful dependencies

pip install fvcore
pip install einops
pip install torchmetrics
pip install pyfftw
pip install phasepack

Test if Mamba have installed successfully

python ./causal-conv1d/tests/test_causal_conv1d.py

Sample Training pipeline

python ./my_train.py --dataset "MRI-SPECT" --lr 0.0005 --epoch 200 --exp_folder_name "dep5_epo200_exp"

Sample Inference pipeline

python validation.py --model_pt ./model/dep5_epo200_exp.pt --test_folder ./my_data --folder_name "dep5_epo200_exp" --exp 0