Skip to content
ModelListedExternal
Download

PrimusM OpenMind SimCLR

> Model from the paper: An OpenMind for 3D medical vision self-supervised learning > Pre-training codebase used to create checkpoint: MIC-DKFZ/nnssl > Dataset: AnonRes/OpenMind > Downstream (segmentation)...

KindOtherVersionRevision 7a8e02e

Model source

Kind
Other
File size
668 MB
Version
vRevision 7a8e02e
Updated
Sep 16, 2026
Downloads
26
Source
Hugging Face

Source description

OpenMind Benchmark 3D SSL Models

Model from the paper: An OpenMind for 3D medical vision self-supervised learning
Pre-training codebase used to create checkpoint: MIC-DKFZ/nnssl
Dataset: AnonRes/OpenMind
Downstream (segmentation) fine-tuning: TaWald/nnUNet


OpenMind

Overview

This repository hosts pre-trained checkpoints from the OpenMind benchmark:
📄 An OpenMind for 3D medical vision self-supervised learning (Wald, T., Ulrich, C., Suprijadi, J., Ziegler, S., Nohel, M., Peretzke, R., ... & Maier-Hein, K. H. (2024).)
(arXiv:2412.17041) — the first extensive benchmark study for self-supervised learning (SSL) on 3D medical imaging data.

Each model was pre-trained using a particular SSL method on the OpenMind Dataset, a large-scale, standardized collection of public brain MRI datasets.

These models are not recommended to be used as-is for feature extraction. Instead we recommend using the downstream fine-tuning frameworks for segmentation and classification adaptation, available in the adaptation repository. While manual download is possible, we recommend using the auto-download feature of the fine-tuning repository by providing the repository URL on Hugging Face instead of a local checkpoint path.


Model Variants

We release SSL checkpoints for two backbone architectures:

  • ResEnc-L: A CNN-based encoder [a, b]
  • Primus-M: A transformer-based encoder [Primus paper]

Each encoder has been pre-trained using one of the following SSL techniques:

MethodDescription
Volume Contrastive (VoCo)Contrastive pretraining method for 3D volumes
VolumeFusion (VF)Spatial volume fusion-based segmentation SSL method
Models Genesis (MG)Reconstruction and denoising based pretraining method
Masked Autoencoders (MAE)Default reconstruction based pretraining method
Spark 3D (S3D)Sparse reconstruction based pretraining mehtod (CNN only)
SimMIMSimple masked reconstruction based pretraining method (TR only)
SwinUNETR SSLRotation, Contrastive and Reconstruction based pre-training method.
SimCLRTransfer of 2D Contrastive learning baseline method to 3D

Sources

1 source
Hugging FacePrimaryVersion 7a8e02eab686f1d6cc2d2bc9761714d32da15105
huggingface.co/MIC-DKFZ/PrimusM-OpenMind-SimCLR/tree/7a8e02eab686f1d6cc2d2bc9761714d32da15105

Verified Sep 16

Model artifacts

1 artifact

checkpoint_final.pth

pth · 668 MB · SHA-256 39bb8cc0a5b8…ee70 · Hugging Face

Download

Source excerpts

2 excerpts

--- license: cc-by-4.0 datasets: - AnonRes/OpenMind pipeline_tag: image-feature-extraction tags: - medical --- # OpenMind Benchmark 3D SSL Models > **Model from the paper**: [An OpenMind for 3D medical vision self-supervised learning](https://arxiv.org/abs/2412.17041) > **Pre-training codebase used to create checkpoint**: [MIC-DKFZ/nnssl](https://github.com/MIC-DKFZ/nnssl) > **Dataset**: [AnonRes/OpenMind](https://huggingface.co/datasets/AnonRes/OpenMind) > **Downstream (segmentation) fine-tuning**: [TaWald/nnUNet](https://github.com/TaWald/nnUNet) --- ![OpenMind](https://huggingface.co/datasets/AnonRes/OpenMind/resolve/main/assets/OpenMindDataset.png) ## Overview This repository hosts pre-trained checkpoints from the **OpenMind** benchmark: 📄 **An OpenMind for 3D medical vision self-supervised learning** (Wald, T., Ulrich, C., Suprijadi, J., Ziegler, S., Nohel, M., Peretzke, R., ... & Maier-Hein, K. H. (2024).) ([arXiv:2412.17041](https://arxiv.org/abs/2412.17041)) — the first extensive benchmark study for **self-supervised learning (SSL)** on **3D medical imaging** data. Each model was pre-trained using a particular SSL method on the [OpenMind Dataset](https://huggingface.co/datasets/AnonRes/OpenMind), a large-scale, standardized collection of public brain MRI datasets. **These models are not recommended to be used as-is for feature extraction.** Instead we recommend using the downstream fine-tuning frameworks for **segmentation** and **classification** adaptation, available in the [adaptation repository](https://github.com/TaWald/nnUNet). *While manual download is possible, we recommend using the auto-download feature of the fine-tuning repository by providing the repository URL on Hugging Face instead of a local checkpoint path.* --- ## Model Variants We release SSL checkpoints for two backbone architectures: - **ResEnc-L**: A CNN-based encoder [[a](https://arxiv.org/abs/2410.23132), [b](https://arxiv.org/abs/2404.09556)] - **Primus-M**: A transformer-based encoder [[Primus paper](https://arxiv.org/abs/2503.01835)] Each encoder has been pre-trained using one of the following SSL techniques: | Method | Description | |---------------|-------------| | [Volume Contrastive (VoCo)](https://arxiv.org/abs/2402.17300) | Contrastive pretraining method for 3D volumes | | [VolumeFusion (VF)](https://arxiv.org/abs/2306.16925) | Spatial volume fusion...

Source context: 26 downloads · 0 likes · Pipeline image-feature-extraction · Repo MIC-DKFZ/PrimusM-OpenMind-SimCLR