| name | graft-neural-population-adapter |
| description | GRAFT methodology for Transformer-based neural population activity modeling with gain-recalibrated adapters enabling cross-day BCI recalibration |
| tags | ["neural population dynamics","transformer","BCI","cross-day adaptation","neural decoding","NLB benchmark"] |
| version | 1 |
| arxiv_id | 2606.11066v1 |
| created | 2026-06-10T00:00:00.000Z |
GRAFT: Gain-Recalibrated Adapters for Transformer-Based Neural Population Activity Modeling
Paper Information
Summary
GRAFT introduces a novel architecture that separates reusable temporal dynamics from a recalibratable neuron interface in Transformer-based neural population activity models. This separation enables efficient cross-day recalibration for long-term brain-computer interfaces where recorded neuron identities, counts, and response statistics change across days.
Key Contributions
1. Interface-Backbone Separation Architecture
- Backbone: Transformer-based temporal dynamics model that captures reusable population-level patterns
- Neuron Interface: Recalibratable adapter layers controlling how recorded neurons enter/leave the shared backbone
- Auxiliary Mechanisms: Gain and positional mechanisms supporting neural activity modeling inside Transformer
2. Cross-Day Recalibration
- Update only 9.21% of parameters when adapting to new recording sessions
- Handles changes in:
- Neuron identities (different neurons recorded)
- Neuron counts (variable population sizes)
- Response statistics (gain/scale changes)
3. State-of-the-Art Performance
- MC Maze NLB'21: 0.3866 co-bps (ensemble) - new state of the art on primary metric
- Cross-day protocol: 0.3749 (Large), 0.3112 (Medium), 0.3152 (Small) co-bps with restricted target-day support sets
Methodology Details
Neuron Interface Design
- Read-in Layer: Adapter mapping variable neuron populations to fixed backbone dimensionality
- Readout Layer: Recalibratable projection from backbone to specific neuron outputs
- Gain Mechanism: Auxiliary scaling to handle response magnitude changes
- Positional Mechanism: Neuron identity encoding supporting variable neuron sets
Training Protocol
- Standard NLB'21: Train on MC Maze dataset
- Cross-Day: Recalibrate from MC Maze to scaled datasets (Large/Medium/Small)
- Restricted Support: Use limited target-day samples for efficient adaptation
Applications
Primary Use Cases
- Long-term BCI systems: Neural prosthetics requiring months/years of operation
- Clinical monitoring: Patient populations with changing neural recording conditions
- Research reproducibility: Sharing models across different recording setups
Neural Decoding Tasks
- Motor cortex decoding (movement trajectory prediction)
- Behavioral state inference from population activity
- Cross-session neural data integration
Technical Details
Model Architecture
Input: Binned spike counts [T, N_var]
↓
Neuron Interface (Adapter): [T, N_var] → [T, D_fixed]
↓
Transformer Backbone: Temporal dynamics processing
↓
Neuron Interface (Readout): [T, D_fixed] → [T, N_var]
↓
Output: Decoded behavioral variables
Gain-Recalibrated Mechanism
- Per-neuron gain parameters: Learnable scaling factors
- Positional encoding: Neuron identity embeddings
- Adapter layers: Bottleneck transformations with low parameter count
Implementation Notes
Key Innovations vs. Previous Approaches
- LFADS: Fixed neuron set, no cross-day support
- NDT: Fixed architecture tied to specific recordings
- VAE-based models: Latent dynamics but neuron-specific readout
Advantages of Separation
- Reusable temporal backbone across recording sessions
- Minimal parameter updates for adaptation (9.21% of total)
- Handles varying neuron counts without retraining entire model
Experimental Validation
Benchmarks
- NLB'21 Challenge: Neural Latents Benchmark standard protocol
- MC Maze Dataset: Motor cortex recordings from maze navigation task
- Cross-Day Protocol: Novel evaluation measuring adaptation efficiency
Metrics
- co-bps: Co-smoothing bits-per-spike (primary NLB metric)
- Recalibration efficiency: Parameter update percentage
- Target-day performance: Decoding accuracy on new recording session
Future Directions
- Extension to multi-region neural populations
- Integration with real-time BCI deployment
- Neuromorphic hardware implementation
- Clinical translation for prosthetic control
Related Skills
latent-neural-dynamics-ml-survey: broader context of neural dynamics modelingneural-population-decoding: decoding methods overviewbci-rehabilitation-protocols: clinical applications
References
Activation: Use when modeling neural population activity with changing neuron sets, designing cross-day BCI systems, adapting pretrained neural decoders, or handling variable neural recording configurations. Keywords: GRAFT, neural population, transformer, BCI, cross-day, recalibration, gain adapter, NLB benchmark.
