medical

You need to agree to share your contact information to access this model

This repository is publicly accessible, but you have to accept the conditions to access its files and content.

Please confirm that you have read and agree to the following disclaimer.
The model in this repository is provided for research use only (Research-only RAIL-M license). The model(s) and/or software are not intended for use in clinical decision-making or for any other clinical use, and performance for clinical use has not been established.

Log in or Sign Up to review the conditions and access this model content.

Raidium

🌐 Raidium | 📄 Paper Link | 🤗 Curia-2 | 🤗 Curia

Curia-MAE: Multi-Modal Multi-Anatomy MAE Pre-Training for 3D Medical Image Segmentation

We introduce Curia-MAE, a multi-modal, multi-anatomy Masked Autoencoder (MAE) pre-trained on roughly 300,000 CT and MRI volumes covering a large number of anatomical sites. Curia-MAE targets dense 3D medical image segmentation: it provides a strong, reusable frozen encoder that can be adapted to a new segmentation task by training only a lightweight decoder on top, while remaining competitive under full fine-tuning and particularly strong on lesion-focused tasks where labeled data is scarce. A single frozen encoder can thus be reused across diverse segmentation tasks, reducing the cost of adapting and deploying such models in clinical workflows.

Check the research paper: https://arxiv.org/abs/2608.05844

Finetuning with nnU-Net

Curia-MAE is a ResEnc-L (residual-encoder U-Net) backbone in the nnU-Net self-supervised pre-training format. Finetune it on a labeled 3D segmentation dataset with nnU-Net's finetuning-from-nnssl workflow:

# 1) dataset fingerprint + base plan
nnUNetv2_extract_fingerprint -d <DATASET_ID>
nnUNetv2_plan_experiment    -d <DATASET_ID>

# 2) preprocess to the pre-training spacing/normalization and register this checkpoint
nnUNetv2_preprocess_like_nnssl -d <DATASET_ID> -n Curia-MAE \
    -pc checkpoint_final.pth -am like_pretrained

# 3) finetune (the generated plan already embeds the checkpoint path)
nnUNetv2_train_pretrained <DATASET_ID> 3d_fullres <FOLD> \
    -p ptPlans__Curia-MAE____Spacing__1.00_1.00_1.00___Norm__Z -tr PretrainedTrainer

Only the encoder and stem are transferred; the segmentation decoder is trained from scratch. The architecture and preprocessing plan are also provided in adaptation_plan.json.

License

The model is released under the RESEARCH-ONLY RAIL-M license. https://huggingface.co/raidium/curia/blob/main/LICENSE

Cite our paper

@article{danielou2026curiamae,
      title={Curia-MAE: Multi-Modal Multi-Anatomy MAE Pre-Training for 3D Medical Image Segmentation}, 
      author={Théo Danielou and Antoine Saporta and Léo Alberge and Corentin Dancette},
      year={2026},
      eprint={2608.05844},
      archivePrefix={arXiv},
      primaryClass={cs.CV},
      url={https://arxiv.org/abs/2608.05844}, 
}

Acknowledgements

Curia-MAE uses the ResEnc-L residual-encoder U-Net architecture from nnU-Net v2 (Division of Medical Image Computing, German Cancer Research Center / DKFZ), and the released checkpoint is packaged to be fine-tunable within the nnU-Net framework. We gratefully acknowledge the nnU-Net project.

Downloads last month
-
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Paper for raidium/Curia-MAE