---
library_name: neurovlm
tags:
- neuroimaging
- contrastive-learning
- projection-heads
---

# ProjectionHeads

Projection heads used by NeuroVLM to align brain-map autoencoder latents and SPECTER text latents.

## Files

- `proj_head_image_infonce.safetensors`
- `proj_head_text_infonce.safetensors`
- `proj_head_text_mse.safetensors`
- `config.json`

Load through `neurovlm.models.load_model("proj_head_image_infonce")`,
`load_model("proj_head_text_infonce")`, or `load_model("proj_head_text_mse")`.