File size: 2,661 Bytes
078ad56
 
 
 
 
 
 
 
 
 
 
 
 
 
 
659f502
078ad56
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
---
tags:
- neuroscience,
- spike-sorting
- electrophysiology
- mouse
- neuropixels
- scikit-learn
- spikeinterface
---

# 🧠 UnitRefine Monkey SUA Classifier

## πŸ“Œ Model Summary

This model is part of the **UnitRefine** pipeline and is trained to classify **single-unit activity (SUA)** in **monkey Utah array recordings**. It uses supervised machine learning to distinguish well-isolated units from multi-unit activity (MUA) and noise based on unit-level spike metrics.

The classifier is designed for **fast, automated unit curation**, and generalizes across **multiple recordings and brain regions**, achieving high accuracy even with limited training data.

The training data includes recordings from **Dr.Sonja Gruen's Lab**(in FZJ).

---

## πŸ” Use Cases

- Automated post-processing of spike sorting output  
- Removing low-quality or noisy units prior to analysis  
- Reducing manual curation effort in large-scale neural recordings  
- Benchmarking unit quality metrics against expert annotations  

---

## 🧬 Metric Selection

For information on which spike metrics were used to train this classifier, please refer to the `model_info.json` file included in the repository.

---

## πŸ’‘ How to Use

This model can be used to **automatically identify SUA units** from spike-sorted data. If you are working with a `SortingAnalyzer` object, you can run the following:

```python
from spikeinterface.curation import auto_label_units

labels = auto_label_units(
    sorting_analyzer=sorting_analyzer,
    repo_id="AnoushkaJain3/UnitRefine-monkey-sua-classifier",
    trusted=["numpy.dtype"]
)
```
This returns a dictionary of predicted labels per unit (1 = SUA, 0 = MUA/Noise).


## πŸ“œ Citation  

If you find [UnitRefine](https://github.com/anoushkajain/UnitRefine) models useful in your research, please cite: **[biorxiv paper](https://www.biorxiv.org/content/10.1101/2025.03.30.645770v1.full.pdf)**.  


## πŸ”— Resources  

- **GitHub Repository:** [UnitRefine](https://github.com/anoushkajain/UnitRefine)  
- πŸ“– **SpikeInterface Tutorial – Automated Curation:**  
  [View Here](https://spikeinterface.readthedocs.io/en/latest/tutorials_custom_index.html#automated-curation-tutorials)  

UnitRefine is **fully integrated with SpikeInterface**, making it easy to incorporate into existing workflows. πŸš€  


## πŸ™ Acknowledgments

Special thanks to **Sonja Gruen**, and **Aitor Morales-Gregorio** for generously providing the datasets used to train and evaluate this model.

---

## πŸ‘©β€πŸ”¬ Authors

**Anoushka Jain**  
PhD Researcher, Musall Lab, Forschungszentrum JΓΌlich  

**Chris Halcrow**  
Lead Developer, SpikeInterface