File size: 1,156 Bytes
98cff93
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
---
language: en
tags:
- medical-imaging
- mri
- self-supervised
- 3d
- neuroimaging
license: apache-2.0
library_name: pytorch
datasets:
- custom
---

# SimCLR-MRI Pre-trained Encoder (SeqAug)

This repository contains a pre-trained 3D CNN encoder for MRI analysis. The model was trained using contrastive learning (SimCLR) with Bloch equation simulations to generate multi-contrast views of the same anatomy.

## Model Description

The encoder is a 3D CNN with 5 convolutional blocks (64, 128, 256, 512, 768 channels), outputting 768-dimensional features. This SeqAug variant treats different simulated MRI sequences as strong augmentations during contrastive learning, encouraging sequence-robust representations.

### Training Procedure
- **Pre-training Data**: 51 qMRI datasets (22 healthy, 29 stroke subjects)
- **Augmentations**: Bloch simulation-based sequence augmentation + standard transformations
- **Input**: 3D MRI volumes (96×96×96)
- **Output**: 768-dimensional feature vectors

## Intended Uses

This encoder is particularly suited for:
- Cross-sequence transfer learning
- Multi-contrast MRI analysis
- Sequence-robust feature extraction