HuiduRep: A Self-Supervised Learning Framework for More Robust Neural Representations from Extracellular Recordings

Kohaku Kirisame, Zishuo Feng, Wei Shi, Jicong Zhang

Keywords: Neuroscience, Spike Sorting, Electrophysiology, Contrastive Learning

TL;DR: We propose HuiduRep, a self-supervised framework that learns robust spike representations for unsupervised spike sorting, outperforming state-of-the-art tools under noise and drift.

Abstract: Extracellular recordings are transient voltage fluctuations in the vicinity of neurons, serving as a fundamental modality in neuroscience for decoding brain activity at single-neuron resolution. Spike sorting, the process of attributing each detected spike to its corresponding neuron, is a pivotal step in brain sensing pipelines. However, it remains challenging under low signal-to-noise ratio (SNR), electrode drift, and cross-session variability. In this paper, we propose HuiduRep, a robust self-supervised representation learning framework that extracts discriminative and generalizable features from extracellular recordings. By integrating contrastive learning with a denoising autoencoder, HuiduRep learns latent representations robust to noise and drift. With HuiduRep, we develop a spike sorting pipeline that clusters spike representations without ground truth labels. Experiments on hybrid and real-world datasets demonstrate that HuiduRep achieves strong robustness. Furthermore, the pipeline outperforms state-of-the-art tools such as KiloSort4 and MountainSort5.

Submission Number: 4