| Literature DB >> 35573306 |
Jordan L Vasko1, Laura Aume1, Sanjay Tamrakar1, Samuel C Iv Colachis1, Collin F Dunlap1,2, Adam Rich1, Eric C Meyers1, David Gabrieli1, David A Friedenberg1.
Abstract
For brain-computer interfaces (BCIs) to be viable for long-term daily usage, they must be able to quickly identify and adapt to signal disruptions. Furthermore, the detection and mitigation steps need to occur automatically and without the need for user intervention while also being computationally tractable for the low-power hardware that will be used in a deployed BCI system. Here, we focus on disruptions that are likely to occur during chronic use that cause some recording channels to fail but leave the remaining channels unaffected. In these cases, the algorithm that translates recorded neural activity into actions, the neural decoder, should seamlessly identify and adjust to the altered neural signals with minimal inconvenience to the user. First, we introduce an adapted statistical process control (SPC) method that automatically identifies disrupted channels so that both decoding algorithms can be adjusted, and technicians can be alerted. Next, after identifying corrupted channels, we demonstrate the automated and rapid removal of channels from a neural network decoder using a masking approach that does not change the decoding architecture, making it amenable for transfer learning. Finally, using transfer and unsupervised learning techniques, we update the model weights to adjust for the corrupted channels without requiring the user to collect additional calibration data. We demonstrate with both real and simulated neural data that our approach can maintain high-performance while simultaneously minimizing computation time and data storage requirements. This framework is invisible to the user but can dramatically increase BCI robustness and usability.Entities:
Keywords: brain–machine (computer interface); deep learning – artificial neural network; intracortical array; neuroprosthetic; statistical process control
Year: 2022 PMID: 35573306 PMCID: PMC9096265 DOI: 10.3389/fnins.2022.858377
Source DB: PubMed Journal: Front Neurosci ISSN: 1662-453X Impact factor: 5.152
FIGURE 1The proposed decoding framework for handling channel disruptions. (A) Raw voltage data and impedance measurements recorded from the implanted electrode array will be delivered to the statistical process control algorithm, where channels with disruptions will be identified. The discovery of disruptions will cause the input from the affected channels to be masked, or zeroed, before being sent to the decoder. This will also trigger an unsupervised update to readjust the weights to the missing input. (B) The statistical process control algorithm to detect disrupted channels is described. The four steps in this process are (1) transforming raw data into four array-level metrics useful for signal monitoring, (2) creating control charts for each of the metrics, (3) using the control charts to flag sessions with potential disruptions, and (4) performing Grubb’s test to determine outlying channels on the flagged sessions. (C) The decoder framework, including a masking layer, an LSTM with 80 hidden units, 25 convolutional filters, a fully connected layer, and finally a 5-unit layer corresponding to the output.
Dataset used for all experiments and training parameters common for all DNN models.
| Neural feature: | Mean wavelet power (MWP) |
| Number of blocks for test period: | 30 blocks |
| Number of blocks for initial training: | 60 blocks |
| Number of blocks for unsupervised updating: | 11 blocks (10 historical + 1 current) |
| Number of epochs: | Determined by early stopping (patience = 2, accuracy delta = 0.01, evaluation blocks = 20) |
| Data augmentation: | Mixup during update loop |
| Dropout: | 0.50 Forward Layers |
| Loss function: | Categorical Cross-Entropy Loss during initial training, BiTempered Loss during unsupervised updates |
| Channels corrupted: | The [0, 1, 5, 10, 15, 20, 30, 50] most important channels |
| Simulated corruptions: | Random linear combinations of floating channel data |
| Damage introduced: | First day of test period |
FIGURE 2The control charts produced for each of the four array-level metrics monitored by the statistical process control-based algorithm. Both and S-charts were produced for impedance and voltage range, while only -charts were produced for the correlation metrics. Dotted red lines represent control limits. The red points indicate metrics flagged for having been out-of-control for at least two consecutive sessions. The green regions represent periods of known damage. (A) Impedance -chart. (B) Impedance S-chart. (C) Vrange -chart. (D) Vrange S-chart. (E) Maximum absolute correlation -chart. (F) Minimum average absolute correlation -chart.
FIGURE 3The performance of the decoders with increasing numbers of channels affected by corruption. Simulated damaged was introduced on the first day of the test set. The uNN-NOMASK, uNN-MASK, and uNN-RETRAIN were tested over 10 different random initializations, while only one SVM was tested for each day and number of channels affected. (A) The accuracy over the 1.4-year test period. (B) The mean accuracy across 100-ms time bins as a function of the number of channels dropped. Error bars represent 95% confidence intervals. (C) The success rate over for each 2.5 s cue over the 1.4-year test period. (D) The mean success rate across 100-ms time bins as a function of the number of channels dropped. Error bars represent 95% confidence intervals.
FIGURE 4(A) The mean accuracy for the three uNN decoders, replotted from Figure 3B with x-axis ticks realigned to match (B). (B) The computational requirements of the uNN decoders as measured by the number of batches required for training after disruptions are introduced. When zero channels are affected, no masking or retraining takes place, and the model would only receive unsupervised updates. The number of batches required is according to an early stopping criterion and is averaged over each of the 10 random initializations applied. Error bars represent 95% confidence intervals. (C) The number of batches required as a function of decoder accuracy.