Table 3: Multi-unit brain region classification for the IBL dataset.
The accuracy and F1-scores for brain region classification are reported for three conditions: (i) a linear layer and (ii) MLP on top of the frozen pretrained representations (for VAE and NEMO), and (iii) after MLP finetuning.
| Model | Linear | MLP | MLP fine-tuned | |||
|---|---|---|---|---|---|---|
| Acc | F1 | Acc | F1 | Acc | F1 | |
|
| ||||||
| Supervised | N/A | N/A | N/A | N/A | 0.50 ± 0.00 | 0.48 ± 0.01 |
| VAE | 0.36 ± 0.00 | 0.32 ± 0.00 | 0.45 ± 0.00 | 0.42 ± 0.00 | 0.48 ± 0.00 | 0.46 ± 0.00 |
| NEMO | 0.48 ± 0.00 | 0.45 ± 0.00 | 0.50 ± 0.00 | 0.48 ± 0.00 | 0.51 ± 0.00 | 0.50 ± 0.00 |