Resencl OpenMind MAE

🚀 OpenMind基準3D自監督學習（SSL）模型

本項目提供了一系列用於3D醫學影像的自監督學習（SSL）預訓練模型。這些模型基於OpenMind基準，在大規模標準化的公共腦MRI數據集上進行預訓練，為3D醫學視覺的自監督學習研究和應用提供了有力支持。

🚀 快速開始

本倉庫託管了來自 OpenMind 基準的預訓練模型檢查點：
📄 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) —— 這是首個針對 3D醫學影像 數據進行 自監督學習（SSL） 的全面基準研究。

每個模型都使用特定的SSL方法在 OpenMind數據集 上進行了預訓練，該數據集是一個大規模、標準化的公共腦MRI數據集集合。

不建議直接使用這些模型進行特徵提取。相反，建議使用 微調倉庫 中的下游微調框架進行 分割 和 分類 任務的適配。 雖然可以手動下載模型，但建議在微調倉庫中使用自動下載功能，只需提供Hugging Face上的倉庫URL，而不是本地檢查點路徑。

✨ 主要特性

模型來源信息

• 論文模型：An OpenMind for 3D medical vision self-supervised learning
• 用於創建檢查點的預訓練代碼庫：MIC-DKFZ/nnssl
• 數據集：AnonRes/OpenMind
• 下游（分割）微調：TaWald/nnUNet

模型變體

我們發佈了兩種骨幹架構的SSL檢查點：

• ResEnc-L：基於卷積神經網絡（CNN）的編碼器 [a, b]
• Primus-M：基於Transformer的編碼器 [Primus論文]

每個編碼器都使用以下SSL技術之一進行了預訓練：

📄 許可證

本項目採用CC BY 4.0許可證。