SpikeReg: Energy-Efficient 3D Deformable Medical Image Registration with Spiking Neural Networks
Organizations: School of Electrical and Computer Engineering, College of Engineering, University of Tehran, Tehran 1439957131, Iran · Max Planck Institute for Brain Research, 60438 Frankfurt am Main, Germany · School of Computer Engineering, Iran University of Science and Technology (IUST), Tehran 16846-13114, Iran
Abstract
Deformable medical image registration aligns anatomical structures across images but remains computationally dense at 3D resolution. Spiking neural networks (SNNs) offer sparse event-driven computation, yet have not been systematically studied for deformable medical image registration. We introduce SpikeReg, a spiking U-Net for 3D brain MRI registration. SpikeReg is initialized from an analog ANN registration teacher, converted by layer-wise weight transfer and activation-percentile threshold calibration, and fine-tuned with a surrogate-gradient objective combining local cross-correlation, diffusion regularization, and spike-rate sparsity. On the OASIS Learn2Reg validation split ( image pairs), SpikeReg reaches Dice , with no significant paired Dice difference from the ANN teacher (, ), at a mean spike rate and a projected arithmetic-energy reduction under an event-sparse SynOps/MAC proxy relative to the dense-ANN baseline. We additionally report two negative findings: displacement distillation from the ANN teacher hurts performance, and ANN teachers trained with a label-Dice loss fail to transfer through rate-code conversion. Together these results show that dense geometric prediction can be performed under sparse event-driven computation, opening a path toward neuromorphic medical image registration.