Spike Sorting with VanillaSort

Abstract

Training spike detectors on real recordings is challenging because algorithmically generated labels can be noisy and incomplete. We propose VanillaSort, combining multichannel detection with spatially augmented, template-guided clustering. VanillaDet uses visibility-aware masking, truncated Gaussian targets and a temporally tolerant positive-bag loss, followed by conditional event-SNR gating. VanillaCluster combines HuiduRep embeddings with relative-amplitude features for Gaussian mixture clustering and refines assignments using cross-fitted waveform templates built from selected core events. VanillaDet improves detection accuracy over SimSort by two and three percentage points on the static and drift subsets of Hybrid Janelia, respectively. The complete pipeline also improves sorting performance over the corresponding HuiduRep baselines. These results support learning from imperfect real-data labels and incorporating waveform consistency into neuronal assignment.