FRAME-C: A knowledge-augmented deep learning pipeline for classifying multi-electrode array electrophysiological signals
Organizations: AI, Optimization and Pattern Recognition Research Group, Dept. of Mechanical Eng., University of Melbourne, Australia · Illawarra Health and Medical Research Institute, School of Biological Sciences, Faculty of Science, Medicine and Health, University of Wollongong, Australia
Abstract
Amyotrophic lateral sclerosis (ALS) is a fatal neurodegenerative disorder characterized by motor neuron degeneration, with changes to neuronal excitability being an early phenotype for ALS identified in patients and animal models. Recent advancements in induced pluripotent stem cell (iPSC) technology have enabled the generation of human iPSC-derived neuronal cultures, which, when combined with multi-electrode array (MEA) electrophysiology, provide rich spatial and temporal electrophysiological data. Traditionally, MEA data is analyzed using handcrafted features based on potentially imperfect domain knowledge, which while useful may not fully capture all the relevant characteristics inherent in the MEA data. Machine learning, in particular deep learning has the potential to automatically learn relevant characteristics (features) from raw data, without solely relying on handcrafted feature extraction. However, handcrafted features remain critical for encoding domain knowledge and improving model interpretability, especially in scenarios with limited or noisy data, as is often the case in most experimental studies. This study introduces FRAME-C, a knowledge-augmented machine learning pipeline that combines domain knowledge, raw spike waveform data, and deep learning techniques to classify MEA signals and identify ALS-specific phenotypes. FRAME-C leverages deep learning to extract important features from spike waveforms, while also incorporating handcrafted features such as spike amplitude, inter-spike interval, and spike duration, thus preserving key spatial and temporal information. We validated FRAME-C on both simulated and real-world MEA data from human iPSC-derived neuronal cultures, demonstrating its superior performance compared to existing methods for MEA classification.
Figures & tables
| Cell line | Sex | Age (years) |
|---|---|---|
| sALS 1 | M | 51 |
| Healthy Control 1 | M | 50 |
| sALS 2 | M | 46 |
| Healthy Control 2 | M | 46 |
| sALS 3 | M | 50 |
| Healthy Control 3 | M | 52 |
| Classification pipeline | Automated feature extraction | Handcrafted features (domain knowledge) | Deep learning classifier | |||
|---|---|---|---|---|---|---|
| Informer | TimesNet | CNN | LSTM | |||
| Raw MEA (baseline) | Yes | No | 49.08 | 52.43 | 53.65 | 49.84 |
| Zhao et al. Zhao et al. (2019) | No | Yes (binary spike locations) | N/A 1 1 1 Not evaluated, since the pipeline by Zhao et al. was originally proposed with a CNN classifier. | N/A 1 1 1 Not evaluated, since the pipeline by Zhao et al. was originally proposed with a CNN classifier. | 52.13 ( 2.83) | N/A 1 1 1 Not evaluated, since the pipeline by Zhao et al. was originally proposed with a CNN classifier. |
| FRAME-C V1 | Yes (from spike waveforms) | No | 58.76 (+19.72) | 59.49 (+13.47) | 57.60 (+7.36) | 59.44 (+19.26) |
| FRAME-C V2 | No | Yes | 59.77 (+21.78) | 61.05 (+16.44) | 55.92 (+4.23) | 61.17 (+22.74) |
| FRAME-C V3 | Yes (from spike waveforms) | Yes | 62.31 (+26.95) | 63.49 (+21.09) | 62.58 (+16.64) | 63.08 (+26.56) |
| Classification pipeline | Automated feature extraction | Handcrafted features (domain knowledge) | Deep learning classifier | |||
|---|---|---|---|---|---|---|
| Informer | TimesNet | CNN | LSTM | |||
| Raw MEA recording | Yes | No | 53.35 | 50.45 | 53.35 | 50.60 |
| Pipeline by Zhao et al. Zhao et al. (2019) | No | Yes (binary spike locations) | N/A 1 1 1 Not evaluated, since the pipeline by Zhao et al. was originally proposed with a CNN classifier. | N/A 1 1 1 Not evaluated, since the pipeline by Zhao et al. was originally proposed with a CNN classifier. | 56.42 (+5.76) | N/A 1 1 1 Not evaluated, since the pipeline by Zhao et al. was originally proposed with a CNN classifier. |
| FRAME-C V1 | Yes (from spike waveforms) | No | 57.71 (+8.17) | 62.47 (+23.83) | 63.09 (+18.26) | 62.29 (+23.10) |
| FRAME-C V2 | No | Yes | 64.73 (+21.33) | 65.12 (+29.08) | 66.21 (+24.11) | 66.45 (+31.32) |
| FRAME-C V3 | Yes (from spike waveforms) | Yes | 66.87 (+25.35) | 68.80 (+36.37) | 66.12 (+23.93) | 68.14 (+34.62) |
| Classification pipeline | Automated feature extraction | Handcrafted features (domain knowledge) | Deep learning classifier | |||
|---|---|---|---|---|---|---|
| Informer | TimesNet | CNN | LSTM | |||
| Raw MEA recording | Yes | No | 57.92 | 57.31 | 53.04 | 50.00 |
| Pipeline by Zhao et al. Zhao et al. (2019) | No | Yes (binary spike locations) | N/A 1 1 1 Not evaluated, since the pipeline by Zhao et al. was originally proposed with a CNN classifier. | N/A 1 1 1 Not evaluated, since the pipeline by Zhao et al. was originally proposed with a CNN classifier. | 75.62 (+42.57) | N/A 1 1 1 Not evaluated, since the pipeline by Zhao et al. was originally proposed with a CNN classifier. |
| FRAME-C V1 | Yes (from spike waveforms) | No | 74.78 (+29.11) | 72.72 (+26.89) | 71.88 (+35.52) | 74.05 (+48.10) |
| FRAME-C V2 | No | Yes | 64.73 (+11.76) | 71.40 (+24.59) | 69.81 (+31.62) | 71.94 (+43.88) |
| FRAME-C V3 | Yes (from spike waveforms) | Yes | 74.25 (+28.19) | 73.85 (+28.86) | 72.13 (+35.99) | 75.08 (+50.16) |
| Feature | Permutation feature importance (%) |
|---|---|
| Spike duration | 23.40 |
| Spike amplitude | 1.60 |
| Inter-spike interval | 0.00 |
| Burst duration | 0.00 |
| Number of spikes per burst | 0.50 |
| Burst spike rate | 1.00 |
| Feature | Permutation feature importance (%) |
|---|---|
| Inter-spike interval | 12.30 |
| Spike amplitude | 2.90 |
| Number of spikes per burst | 1.10 |
| Spike duration | 0.80 |
| Burst spike rate | 0.40 |
| Burst duration | 1.00 |
| Feature | Timepoint | Median (Ctrl) | Median (ALS) | -value | Sig. | Cliff’s | ||
|---|---|---|---|---|---|---|---|---|
| Peak amplitude ( V) | Early (month 1) | 23 | 23 | 4.714 | 4.596 | 0.1875 | ns | 0.229 |
| Mid (month 2) | 24 | 24 | 4.702 | 4.649 | 0.3808 | ns | 0.149 | |
| Late (month 3) | 24 | 24 | 4.669 | 4.573 | 0.5848 | ns | 0.094 | |
| Spike interval (ms) | Early (month 1) | 23 | 23 | 643.0 | 857.5 | 0.2065 | ns | 0.219 |
| Mid (month 2) | 24 | 24 | 777.0 | 1059.5 | 0.1835 | ns | 0.226 | |
| Late (month 3) | 24 | 24 | 1370.0 | 1142.5 | 0.7650 | ns | 0.052 |
| Method | No data augmentation (%) | With data augmentation (%) |
|---|---|---|
| CNN classifier (Zhao et al. Zhao et al. (2019) ) | 57.24 | 56.44 |
| FRAME-C V3 with Informer | 63.27 | 65.92 |
| FRAME-C V3 with TimesNet | 65.74 | 66.76 |
| FRAME-C V3 with CNN | 66.28 | 66.66 |
| FRAME-C V3 with LSTM | 65.58 | 67.82 |
| Simulated wells (recordings) per cell type | Pipeline by Zhao et al. (%) | FRAME-C V3 with LSTM (%) |
|---|---|---|
| 2 | 51.12 | 75.06 (+46.83) |
| 4 | 75.44 | 82.05 (+8.76) |
| 8 | 79.13 | 84.58 (+6.89) |
| 16 | 94.07 | 94.21 ( +0.15 ) |