Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

GRAFT: Gain-Recalibrated Adapters for Transformer-Based Neural Population Activity Modeling

About

Neural population activity models can recover rich temporal structure from binned spikes, but their read-in and readout layers often remain tied to a fixed set of recorded neurons. This coupling limits reuse in long-term brain-computer interfaces, where recorded neuron identities, counts, and response statistics can change across days. We introduce GRAFT, a Transformer-based neural population activity model that separates reusable temporal dynamics from a recalibratable neuron interface. The neuron interface controls how recorded neurons enter and leave the shared backbone, and auxiliary gain and positional mechanisms support neural activity modeling inside the Transformer. On MC Maze under the standard NLB'21 protocol, GRAFT reaches 0.3866 co-bps as an ensemble, setting a new state of the art on the primary co-bps metric among public and reported NLB'21 results. In a cross-day protocol constructed from the NLB'21 MC Maze dataset series, GRAFT recalibrates from MC Maze to the scaled MC Maze datasets (Large/Medium/Small) by updating only 9.21% of parameters, reaching 0.3749, 0.3112, and 0.3152 co-bps with restricted target-day support sets. These results show that the same interface-backbone separation supports both strong Transformer-based neural population activity modeling and data-efficient cross-day recalibration.

Xiangsheng Ge, Yang Xie• 2026

Related benchmarks

TaskDatasetResultRank
neural population activity modelingMC Maze NLB'21 (test)
co-bps0.3866
11
neural population activity modelingscaled MC Maze Large 2009-10-06
Co-bps0.3749
9
neural population activity modelingscaled MC Maze Medium 2009-09-29
Co-BPS0.3112
9
neural population activity modelingscaled MC Maze Small 2009-09-28
co-bps0.3152
9
Showing 4 of 4 rows

Other info

Follow for update