SHD

Full-data SHD training curves and matched two-layer inference rasters for COBA and PING.

Abstract

We trained two-layer conductance-based networks to recognize spoken digits represented by cochlear spike events. Both COBA and PING learned under the calibrated initialization and optimization recipe, with PING achieving higher development-validation accuracy in the paired replicate. Matched inference rasters show how the trained layers respond to one held-out utterance. This single-seed comparison and illustrative response do not establish competitive test performance or a general benefit from inhibition.

TODO

  1. Use the full utterance time ranges in the next training run (Methods, point 1).
  2. Apply no input corruption in the next training run (Methods, point 2).

Results

COBA training

After 50 epochs, online training accuracy was 84% and validation accuracy 82.6% (Fig. 1B). Validation cross-entropy fell from 3.0055 to 0.5919 (Fig. 1A). All three learned matrices changed and final weights were finite. The condition took 58.4 minutes, including 52.7 minutes training and 5.7 minutes validation, excluding acquisition, setup and transfer. This is one development-holdout replicate, not an official-test evaluation.

Two-layer COBA online training and validation loss and accuracy over the recorded epochs.
Figure 1.

Recorded COBA training curves. We trained two 128-neuron excitatory layers on 7,340 utterances and evaluated 816 held-out utterances. (A) Example-average classification cross-entropy. (B) Classification accuracy; the dotted line marks 5% uniform random guessing across 20 classes. Black lines/circles show online pre-update minibatch measurements aggregated over each epoch; red lines/circles show complete validation with that epoch’s final weights. Epoch zero contains validation only. Each curve is one seed-42 replicate, without averaging or uncertainty bands. Class counts are naturally unequal; the split is utterance-disjoint, not speaker-disjoint.

PING training

With fixed reciprocal feedback in both 128 E/32 I layers, online training accuracy was 89% and validation accuracy 84.4% (Fig. 2B). Validation cross-entropy fell from 3.1025 to 0.4725 (Fig. 2A). All three learned matrices changed and final weights were finite. The condition took 100.8 minutes, including 91.6 minutes training and 9.2 minutes validation. It ran after COBA had completed. Higher validation accuracy in this pair does not establish a general inhibitory advantage or its mechanism.

Two-layer PING online training and validation loss and accuracy over the recorded epochs.
Figure 2.

Recorded PING training curves. We used the same training/validation utterances, initial learned matrices, sample order and corruption streams as COBA. (A) Example-average classification cross-entropy. (B) Classification accuracy, with 5% uniform-guessing accuracy dotted. Black lines/circles are online pre-update measurements aggregated over the epoch; red lines/circles are full validation with final epoch weights. Epoch zero contains validation only. Each curve is one seed-42 replicate, without averaging or uncertainty bands. The official test set was not evaluated.

COBA two-layer response

We replayed the first validation utterance through the final trained weights, without optimizer updates. The true class was 0, and COBA predicted class 0. Both layers responded during the utterance (Fig. 3). This fixed example illustrates activity propagation; it was not selected for correct classification or its raster appearance and cannot summarize performance across utterances.

Excitatory spike rasters and population-count autocorrelations for both trained COBA layers during one held-out utterance.
Figure 3.

Recorded COBA inference raster. We presented one held-out SHD utterance with a fresh seeded insertion/deletion draw and fixed final-epoch weights. (A, C) Layer 1 and layer 2 rasters. Each black tick is one spike; horizontal position is elapsed time from utterance onset and vertical position identifies a neuron. Each raster contains 128 excitatory neurons and no inhibitory population. The full one-second presentation is shown, with no smoothing, trial average or transient removal. The same input events were used for PING in Figure 4. (B, D) Corresponding layer population spike-count autocorrelograms in 1 ms bins. We divided each uncentered lagged product sum by the number of overlapping bins and the squared full-utterance mean count (Equation 2). Black traces show E at positive lags up to 200 ms; zero lag is omitted because of self-pairs. The dotted line at one marks the expected baseline for independent stationary activity. These are population-count correlations, not averages of individual-neuron autocorrelations; each y axis includes the full displayed peak.

PING two-layer response

PING predicted class 0 for the same true-class-0 utterance. The raster shows intermittent excitatory responses and synchronized inhibitory events in both layers (Fig. 4). Identical inhibitory parameters and uniform incoming feedback produce identical inhibitory spike trains within a layer. This response is illustrative; the raster alone does not establish a rhythm’s frequency or show why PING trained more accurately.

Excitatory and inhibitory spike rasters and population-count autocorrelations for both trained PING layers during the matched held-out utterance.
Figure 4.

Recorded PING inference raster. We used the same utterance and corruption draw as COBA, with final-epoch weights fixed and no optimization. (A, C) Layer 1 and layer 2 rasters. Black ticks identify spikes from 128 excitatory neurons, indexed 0–127; red ticks identify spikes from 32 inhibitory neurons stacked above them, indexed 128–159. Time runs horizontally from utterance onset. The thin horizontal line separates populations. Both rasters show the full one-second presentation without smoothing, averaging or removal of the initial transient. (B, D) Corresponding population-count autocorrelograms, calculated separately for E (black) and I (red) with the same 1 ms bins, overlap correction and squared-mean normalization as Figure 3 (Equation 2). Positive lags extend to 200 ms, with zero lag omitted and no trial averaging. The dotted line at one marks the independent stationary baseline; each y axis includes both populations’ full displayed peaks. The single stimulus-evoked response includes its changing activity envelope and does not establish a sustained oscillation.

Methods

The architecture and simulation settings are listed in Table 1.

  1. Prepare SHD. We split 8,156 official training utterances [1] (20 classes, 700 channels) into 7,340 training and 816 validation examples using a seed-42 per-class shuffle. Original spike-train lengths varied: last-spike times ranged from 0.233 to 1.369 s. We kept only events before 1 s to fit the simulation, flooring times into binary 1 ms bins. Speakers could overlap between splits; the official test set was unused.

  2. Corrupt inputs. Per channel and 1 ms bin, we deleted existing spikes with probability 0.0008 and filled empty bins with probability 0.0016 (approximately 1.6 Hz). Both models shared example order and noise draws; validation noise stayed fixed across epochs for consistent comparison. Corruption exposed models to missing and extra spikes, but its benefit was not tested against uncorrupted training.

  3. Initialize hidden weights. Both models shared seed-42 learned matrices. SNNLab’s LowerClampedNormal used parent mean 0.09 μS and SD 0.009 μS. Input→E1 and E1→E2 initial zero fractions were 0.95 and 0.90, followed by division by retained fan-in (35 and 12.8). Negative draws were clamped; initial zeros remained trainable and weights remained nonnegative.

  4. Initialize readout and feedback. Signed E2→readout weights had Gaussian mean zero and SD 0.1 μS, centered across source neurons for each class, without sparsification or fan-in scaling. PING feedback was fixed at 0.05/1280.05/128 μS per E→I edge and 0.05/320.05/32 μS per I→E edge: 0.05 μS summed incoming conductance in either direction. COBA had no inhibitory feedback.

  5. Score utterances. We used a custom rate-normalized mean-voltage score:

    zc=G uˉc rrefmax⁡(rˉE2,rfloor).(1)z_c=G\,\bar u_c\,\frac{r_{\mathrm{ref}}}{\max(\bar r_{E2},r_{\mathrm{floor}})}. \tag{1}

    Here zcz_c is the score for class cc, uˉc\bar u_c its mean non-spiking readout voltage, and rˉE2\bar r_{E2} the per-neuron E2 rate, both over the full second. We used gain G=20G=20, reference rate rref=20 Hzr_{\mathrm{ref}}=20\,\mathrm{Hz} and rate floor rfloor=1 Hzr_{\mathrm{floor}}=1\,\mathrm{Hz}, predicting the largest score. The positive scaling preserves that choice for fixed weights but changes cross-entropy and training gradients.

  6. Train both models. We used SNNLab’s train for 50 epochs with class-stratified batches of 256, merging tails smaller than 20. Custom Adam minimized mean cross-entropy per represented class; input/interlayer/readout learning rates were 5×10−6/2×10−5/0.015\times10^{-6}/2\times10^{-5}/0.01, with betas 0.9/0.999 and no weight decay. The FastSigmoid surrogate used slope 100/15 mV−1100/15\,\mathrm{mV}^{-1} and gradient scale 0.01. States reset per utterance. Reported losses were example means; validation used epoch-end weights, including epoch zero.

  7. Record matched responses. Using SNNLab’s infer, we replayed the first validation utterance with final-epoch weights, one shared fresh corruption draw (seed 46) and no updates. Selection did not depend on prediction or spike appearance.

  8. Calculate autocorrelograms. We applied SNNLab’s spike_autocorrelogram separately to each population over the full utterance:

    Acorr,P[ℓ]=∑k=0Nt−ℓ−1nP[k] nP[k+ℓ](Nt−ℓ) nˉP 2.(2)A_{\mathrm{corr},P}[\ell] =\frac{\displaystyle\sum_{k=0}^{N_t-\ell-1}n_P[k]\,n_P[k+\ell]} {(N_t-\ell)\,\bar n_P^{\,2}}. \tag{2}

    Here Acorr,PA_{\mathrm{corr},P} is dimensionless autocorrelation for population PP (E or I in one layer), nP[k]n_P[k] its count in bin kk, and nˉP\bar n_P its mean across Nt=1000N_t=1000 bins. We used 1 ms bins and integer lags ℓ=1,…,200\ell=1,\ldots,200 (1–200 ms). Independent stationary activity has baseline one; zero lag is omitted, and silent or too-short recordings give undefined curves. No smoothing, transient removal or trial averaging was applied; the stimulus-evoked activity envelope contributes to the correlation.

References

  1. Cramer, B., Stradmann, Y., Schemmel, J., and Zenke, F. The Heidelberg Spiking Data Sets for the Systematic Evaluation of Spiking Neural Networks. IEEE Transactions on Neural Networks and Learning Systems, 33, 2744–2757 (2022). doi:10.1109/TNNLS.2020.3044364.

Parameters

Figure 5.

Executed network circuits. (A) COBA. (B) PING. Diagrams were generated from the retained network definitions using SNNLab’s diagram API. Boxes identify populations and their sizes; arrows show signal direction, with inhibitory connections in red. The circuit view omits measurement reductions and their output boxes; the rate-normalized readout is defined in Equation 1.

ConfigurationCOBAPING
Input700 SHD cochlear spike channels [1]Same
Hidden neuron modelConductance-based leaky integrate-and-fire (COBA LIF)Same for both E and I
Synapses onto hidden neuronsExponentially decaying AMPA conductancesExponentially decaying AMPA and GABA conductances
Hidden layer 1128 excitatory neurons128 excitatory + 32 inhibitory neurons
Hidden layer 2128 excitatory neurons128 excitatory + 32 inhibitory neurons
Feedforward pathwaysInput → E1 → E2 → readoutSame
Within-layer feedbackNoneFixed E→I and I→E connections in each layer
Readout20 non-spiking leaky-integrator unitsSame
Readout synapseSigned leaky-integrator filter, 10 ms decaySame
Classification score20 × mean readout voltage × 20 Hz / max(mean E2 rate, 1 Hz)Same
Cm  (membrane capacitance)C_m\;\text{(membrane capacitance)}E: 1 nFE: 1 nF; I: 0.5 nF
gL  (leak conductance)g_L\;\text{(leak conductance)}E: 0.05 μSE: 0.05 μS; I: 0.025 μS
τpassive=Cm/gL\tau_{\mathrm{passive}}=C_m/g_L20 ms (derived)20 ms for both E and I (derived)
τAMPA\tau_{\mathrm{AMPA}}10 msSame
τGABA\tau_{\mathrm{GABA}}Not used10 ms
Refractory period0 ms0 ms
Simulation timestep / presentation1 ms / 1 sSame
Learned weightsInput→E1, E1→E2, E2→readoutSame; inhibitory feedback remains fixed
Training / validation utterances7,340 / 816Same
Epochs / batch size / seed50 / 256 / 42Same
Optimizer / objectiveAdam / equal-class cross-entropySame
Learning rates: input / interlayer / readout5e-06 / 2e-05 / 0.01Same
Input insertion / deletion1.6 Hz per channel / 0.0008 probabilitySame
Initial hidden weights: Gaussian parent mean / SD0.09 / 0.009 μSSame
Initial zero fractions: input / interlayer0.95 / 0.9Same
Fixed summed feedback: E→I / I→ENone0.05 / 0.05 μS
Table 1.

Architecture and simulation settings. E and I denote excitatory and inhibitory populations; E1 and E2 identify the two hidden excitatory layers. We chose a 20 ms passive time constant and fixed capacitances, then calculated leak conductances as capacitance divided by that time constant. The passive time constant is therefore derived from the configured capacitance and leak conductance, not an additional independent physical parameter. Synaptic conductances change the effective membrane time constant. AMPA and GABA time constants describe conductance decay. Readout voltage and E2 firing rate are averaged over each one-second utterance. Neuronal and synaptic states reset between utterances.

Dataset

Upstream: —

Downstream: —

RunDateDurationSizeRan onSource
exp124-r164-presentUsed1s550.1 KBLocalLocal (6)
exp124-r162-analyseUsed1s95.5 KBLocalLocal
exp124-r149-computeUsed<1s176.8 MBLocalLocal