ALIGN: Adversarial Learning for Generalizable Speech Neuroprosthesis
Organizations: Kempner Institute for the Study of Natural and Artificial Intelligence, Harvard University, Boston, Massachusetts, USA · Halıcıo˘glu Data Science Institute, Department of Electrical and Computer Engineering, Department of Computer Science and Engineering, University of California San Diego, La Jolla, California, USA
Abstract
Intracortical brain-computer interfaces (BCIs) can decode speech from neural activity with high accuracy when trained on data pooled across recording sessions. In realistic deployment, however, models must generalize to new sessions without labeled data, and performance often degrades due to cross-session nonstationarities (e.g., electrode shifts, neural turnover, and changes in user strategy). In this paper, we propose ALIGN, a session-invariant learning framework based on multi-domain adversarial neural networks for semi-supervised cross-session adaptation. ALIGN trains a feature encoder jointly with a phoneme classifier and a domain classifier operating on the latent representation. Through adversarial optimization, the encoder is encouraged to preserve task-relevant information while suppressing session-specific cues. We evaluate ALIGN on intracortical speech decoding and find that it generalizes consistently better to previously unseen sessions, improving both phoneme error rate and word error rate relative to baselines. These results indicate that adversarial domain alignment is an effective approach for mitigating session-level distribution shift and enabling robust longitudinal BCI decoding.
Figures & tables
| WER (%) (mean std) | ||
|---|---|---|
| Days from last train | GRU Baseline | ALIGN |
| TTA from First Test Session | ||
| 71 | 51.28 0.95 | 45.40 0.00 |
| 84 | 86.71 2.29 | 65.65 0.19 |
| 86 | 99.11 0.11 | 71.79 0.68 |
| Ablation | Validation PER (%) |
|---|---|
| ALIGN | |
| w/o adversarial loss | |
| w/o temporal stretch augmentation |
Appendix figures & tables7 assets
Supplementary material from the paper’s appendix.
Appendix
| Parameter | Value |
|---|---|
| Optimizer | AdamW |
| Learning rate | 0.001 |
| Weight decay | |
| Batch size | 64 |
| Epochs | 500 |
| Scheduler | Multistep (milestone=300) |
| Parameter | T12 | T15 |
|---|---|---|
| Optimizer | Adam | AdamW |
| Learning rate scheduler | None | Cosine |
| Learning rate warmup batches | – | 1000 |
| Training | 73 epochs | 120000 batches |
| Batch size | 64 | 64 |
| (0.9, 0.999) | (0.9, 0.999) |
Brain-Computer Interfaces
Brain-computer interfaces translate neural activity into commands or feedback for communication, movement, rehabilitation, and sensory restoration.
See all 128 papers