GC-LIF: Spiking Group-Equivariant Neural Network With Group-Coupled Membrane Dynamics
Organizations: College of ICT, Can Tho University, Can Tho, Vietnam. · Center for Digital Transformation and Communication, Can Tho University, Can Tho, Vietnam. · Department of Computer Science and Engineering, The University of Aizu, Japan.
Abstract
Spiking neural networks (SNNs) promise event-driven, low-power inference, but on vision tasks they still acquire rotation invariance only through data augmentation. Group-equivariant CNNs build such symmetries into the architecture, yet a faithful spiking analogue is missing: a leaky integrate-and-fire (LIF) neuron placed after a -equivariant convolution can break equivariance through its membrane recurrence. We give a necessary and sufficient condition for this not to happen. The layer is equivariant at every time step if and only if the neuron parameters are constant along the joint orbits of the group action on the (group, space) index. We solve it in closed form, and it covers any elementwise stateful neuron. Its sufficiency half specialises a known result for equivariant recurrent networks; its necessity half rules out the heterogeneous per-neuron parameters common in SNNs. The per-channel parameterisation that satisfies it, Group-Coupled LIF (GC-LIF), is combined with and -equivariant convolutions and a spike-rate group max-pool into SGE-Net. On MNIST, KMNIST, Fashion-MNIST, CIFAR-10 and CIFAR-100, evaluating each checkpoint on the clean test set and all its group-transformed copies, SGE-Net trained without rotations is invariant under and and outperforms a plain SNN trained with rotations in nine of ten dataset-group pairs, whereas non-equivariant baselines lose up to percentage points in the worst case. Operator-level tests confirm the theory before and after training, and an analytical estimate places SGE-Net at the arithmetic energy of a standard CNN and below its ANN counterpart at eight time steps.
Figures & tables
| Models | MNIST | FMNIST | KMNIST | CIFAR10 | CIFAR100 |
|---|---|---|---|---|---|
| CNN | 99.20 0.32 | 91.13 2.54 | 93.88 0.51 | 79.93 3.43 | 49.48 2.10 |
| SCNN | 98.68 0.13 | 89.84 0.57 | 89.20 5.53 | 75.77 0.06 | 34.94 5.59 |
| GCNN | 97.86 1.59 | 91.37 1.02 | 91.61 0.70 | 79.81 1.84 | 47.21 6.04 |
| SGE-Net | 95.58 1.46 | 87.78 2.16 | 88.30 0.10 | 73.58 4.38 | 40.50 3.05 |
| Data | Group | Spike disagreement | Logit deviation | Prediction agreement | |||
|---|---|---|---|---|---|---|---|
| init | trained | init | trained | ||||
| KMNIST | 1.0000 | ||||||
| KMNIST | n.d. | 0.5770 | |||||
| KMNIST | 1.0000 | ||||||
| KMNIST | 0.6062 | ||||||
| CIFAR10 | 1.0000 | ||||||
| Group | Clean | Full-orbit | Energy ( J) | |||
|---|---|---|---|---|---|---|
| 4 | 93.42 1.72 | 93.41 1.70 | 0.253 | 21.70 | ||
| 6 | 94.87 1.55 | 94.87 1.52 | 0.245 | 29.09 | ||
| 8 | 95.58 1.46 | 95.58 1.46 | 0.237 | 36.50 | ||
| 12 | 96.02 1.38 | 96.01 1.27 | 0.221 | 51.38 | ||
| 16 | 96.35 1.31 | 96.34 1.29 | 0.205 | 66.22 | ||
| 4 | 93.42 1.72 | 87.21 1.85 | 0.253 | 21.70 |
| Layer | CNN | SCNN | GCNN | SGE-Net |
|---|---|---|---|---|
| conv/lift- | ||||
| conv/gconv- | ||||
| conv/gconv- | ||||
| conv/gconv- | ||||
| classifier | ||||
| Total |
| Model | Params | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| Syn. | Neuron | Total ( J) | Ratio | Syn. | Neuron | Total ( J) | Ratio | ||||
| CNN | 16,890 | N/A | 21.29 | 0.00 | 21.53 | 1.00 | N/A | 21.29 | 0.00 | 21.53 | 1.00 |
| SCNN | 16,890 | 0.274 | 6.26 | 1.24 | 8.95 | 0.42 | 0.264 | 8.01 | 1.66 | 11.59 | 0.54 |
| GCNN | 16,514 | N/A | 84.13 | 0.00 | 84.61 | 3.93 | N/A | 84.13 | 0.00 | 84.61 | 3.93 |
| SGE-Net | 16,514 | 0.245 | 23.72 | 2.48 | 29.09 | 1.35 | 0.237 | 29.33 | 3.31 | 36.50 | 1.69 |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| Setting | Model | MNIST | FMNIST | KMNIST | CIFAR10 | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Prelim. | Final | Prelim. | Final | Prelim. | Final | Prelim. | Final | ||||||
| Clean | CNN | 99.21 | 99.20 | 91.95 | 91.13 | 94.67 | 93.88 | 80.88 | 79.93 | ||||
| SCNN | 98.73 | 98.68 | 90.84 | 89.84 | 89.49 | 89.20 | 76.67 | 75.77 | |||||
| GCNN | 98.16 | 97.86 | 91.82 | 91.37 | 91.93 | 91.61 | 80.84 | 79.81 | |||||
| SGE-Net | 95.75 | 95.58 | 88.64 | 87.78 | 89.57 | 88.30 | 73.61 | 73.58 | |||||
| , ✗ ✗ | CNN | 46.78 | 43.17 | 37.58 | 34.53 | 36.89 | 35.28 | 48.17 | 46.73 | ||||
| Setting | Model | MNIST | FMNIST | KMNIST | CIFAR10 | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Prelim. | Final | Prelim. | Final | Prelim. | Final | Prelim. | Final | ||||||
| , ✗ ✗ | CNN | 40.44 | 41.02 | 29.45 | 25.79 | 25.51 | 24.05 | 39.20 | 38.96 | ||||
| SCNN | 40.69 | 41.35 | 29.55 | 26.46 | 25.42 | 23.65 | 37.91 | 37.91 | |||||
| GCNN | 69.50 | 87.88 | 61.89 | 59.31 | 56.64 | 56.12 | 64.51 | 64.92 | |||||
| SGE-Net | 77.75 | 85.62 | 56.43 | 58.42 | 51.75 | 53.31 | 58.02 | 59.75 | |||||
| , ✓ ✓ | CNN | 95.95 | 95.74 | 85.21 | 83.26 | 73.49 | 72.86 | 72.14 | 67.51 | ||||
| Data | Model | ||||
|---|---|---|---|---|---|
| no disc | disc | no disc | disc | ||
| MNIST | GCNN (ReLU) | 57.74 18.77 | 87.46 0.86 | 71.92 43.17 | 87.03 8.16 |
| MNIST | GCNN ( -level quantiser) | 61.04 5.54 | 84.84 25.45 | 66.62 6.03 | 84.32 3.97 |
| MNIST | GCNN (noisy ReLU) | 64.69 24.04 | 87.76 5.04 | 68.21 2.00 | 87.43 0.29 |
| MNIST | SGE-Net (spiking) | 66.82 23.07 | 85.41 18.30 | 63.32 22.60 | 84.75 7.98 |
| KMNIST | GCNN (ReLU) | 55.02 10.62 | 56.18 10.98 | 54.60 7.43 | 56.99 4.28 |
Equivariant Neural Networks
Equivariant neural networks encode transformations such as rotations, translations, and permutations so model outputs change consistently with their inputs.
See all 159 papers