Towards robust and generalizable representations of extracellular data using contrastive learning
Vishnubhotla, A.; Loh, C.; Paninski, L.; Srivastava, A.; Hurwitz, C. L.
Show abstract
Contrastive learning is quickly becoming an essential tool in neuroscience for extracting robust and meaningful representations of neural activity. Despite numerous applications to neuronal population data, there has been little exploration of how these methods can be adapted to key primary data analysis tasks such as spike sorting or cell-type classification. In this work, we propose a novel contrastive learning framework, CEED (Contrastive Embeddings for Extracellular Data), for high-density extracellular recordings. We demonstrate that through careful design of the network architecture and data augmentations, it is possible to generically extract representations that far outperform current specialized approaches. We validate our method across multiple high-density extracellular recordings. All code used to run CEED can be found at https://github.com/ankitvishnu23/CEED.
Matching journals
The top 5 journals account for 50% of the predicted probability mass.
Similar papers in this journal
Similar papers in this journal
Similar papers in this journal
Similar papers in this journal
- How do spike collisions affect spike sortingperformance? 96%
- MEDiCINe: Motion Correction for Neural Electrophysiology Recordings 95%
- A Stochastic Dynamic Operator framework that improves the precision of analysis and prediction relative to the classical spike-triggered average method, extending the toolkit. 93%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.