SPINT: Spatial Permutation-Invariant Neural Transformer for Consistent Intracortical Motor Decoding

TL;DR

SPINT is a novel neural transformer framework that enables robust, session-independent intracortical motor decoding by operating on unordered neural data and supporting few-shot adaptation, improving long-term BCI stability.

Contribution

We introduce SPINT, a permutation-invariant neural transformer with dynamic positional embeddings and dropout regularization, enhancing cross-session generalization and adaptation in intracortical BCI decoding.

Findings

Outperforms existing zero-shot and few-shot baselines in cross-session tasks.

Supports inference on variable neural population sizes without test-time alignment.

Eliminates need for test-time fine-tuning or label-dependent alignment.

Abstract

Intracortical Brain-Computer Interfaces (iBCI) aim to decode behavior from neural population activity, enabling individuals with motor impairments to regain motor functions and communication abilities. A key challenge in long-term iBCI is the nonstationarity of neural recordings, where the composition and tuning profiles of the recorded populations are unstable across recording sessions. Existing methods attempt to address this issue by explicit alignment techniques; however, they rely on fixed neural identities and require test-time labels or parameter updates, limiting their generalization across sessions and imposing additional computational burden during deployment. In this work, we introduce SPINT - a Spatial Permutation-Invariant Neural Transformer framework for behavioral decoding that operates directly on unordered sets of neural units. Central to our approach is a novel context-dependent positional embedding scheme that dynamically infers unit-specific identities, enabling flexible generalization across recording sessions. SPINT supports inference on variable-size populations and allows few-shot, gradient-free adaptation using a small amount of unlabeled data from the test session. To further promote model robustness to population variability, we introduce dynamic channel dropout, a regularization method for iBCI that simulates shifts in population composition during training. We evaluate SPINT on three multi-session datasets from the FALCON Benchmark, covering continuous motor decoding tasks in human and non-human primates. SPINT demonstrates robust cross-session generalization, outperforming existing zero-shot and few-shot unsupervised baselines while eliminating the need for test-time alignment and fine-tuning. Our work contributes an initial step toward a robust and scalable neural decoding framework for long-term iBCI applications.