Spiking Diffusion Models
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
Recent years have witnessed Spiking Neural Networks (SNNs) gaining attention for their ultra-low energy consumption and high biological plausibility compared with traditional Artificial Neural Networks (ANNs). Despite their distinguished properties, the application of SNNs in the computationally intensive field of image generation is still under exploration. In this paper, we propose the Spiking Diffusion Models (SDMs), an innovative family of SNN-based generative models that excel in producing high-quality samples with significantly reduced energy consumption. In particular, we propose a Temporal-wise Spiking Mechanism (TSM) that allows SNNs to capture more temporal features from a bio-plasticity perspective. In addition, we propose a threshold-guided strategy that can further improve the performances by up to 16.7% without any additional training. We also make the first attempt to use the ANN-SNN approach for SNN-based generation tasks. Extensive experimental results reveal that our approach not only exhibits comparable performance to its ANN counterpart with few spiking time steps, but also outperforms previous SNN-based generative models by a large margin. Moreover, we also demonstrate the high-quality generation ability of SDM on large-scale datasets, e.g., LSUN bedroom. This development marks a pivotal advancement in the capabilities of SNN-based generation, paving the way for future research avenues to realize low-energy and low-latency generative applications. Our code is available at https://github.com/AndyCao1125/SDM.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 58 of its 84 references that are in the map. First posted on arXiv 29 Aug 2024.
Builds on
58 of 84 references.
Closest ancestors · 17
Same direction, newest first
- 2023Fully Spiking Denoising Diffusion Implicit Models4 Dec
- 2023Spiking-Diffusion: Vector Quantized Discrete Diffusion Model with Spiking Neural Networks20 Aug
- 2023Spiking Denoising Diffusion Probabilistic Models29 Jun
- 2023One Transformer Fits All Distributions in Multi-Modal Diffusion at Scale12 Mar
- 2022Refining Generative Process with Discriminator Guidance in Score-based Diffusion Models28 Nov
- 2022Classifier-Free Diffusion Guidance26 Jul
- 2022EGSDE: Unpaired Image-to-Image Translation via Energy-Guided Stochastic Differential Equations14 Jul
- 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps2 Jun
- 2022Elucidating the Design Space of Diffusion-Based Generative Models1 Jun
- 2022Analytic-DPM: an Analytic Estimate of the Optimal Reverse Variance in Diffusion Probabilistic Models17 Jan
- 2021Variational Diffusion Models1 Jul
- 2021Cascaded Diffusion Models for High Fidelity Image Generation30 May
5 more
Other references · 41
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2023GPT-4 Technical Reportcited by 18,132
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2018A Style-Based Generator Architecture for Generative Adversarial Networkscited by 5,986
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
- 2014Deep Learning Face Attributes in the Wildcited by 3,679
- 2022DreamFusion: Text-to-3D using 2D Diffusioncited by 3,006
- 2016Improved Techniques for Training GANscited by 2,938
- 2022Video Diffusion Modelscited by 2,280
- 2020DiffWave: A Versatile Diffusion Model for Audio Synthesiscited by 1,339
- 2023ProlificDreamer: High-Fidelity and Diverse Text-to-3D Generation with Variational Score Distillationcited by 1,207
29 more
- 2019Surrogate Gradient Learning in Spiking Neural Networkscited by 604
- 2021Grad-TTS: A Diffusion Probabilistic Model for Text-to-Speechcited by 445
- 2017Efficient Processing of Deep Neural Networks: A Tutorial and Surveycited by 427
- 2017Spatio-Temporal Backpropagation for Training High-performance Spiking Neural Networkscited by 399
- 2020Incorporating Learnable Membrane Time Constant to Enhance Learning of Spiking Neural Networkscited by 340
- 2021Deep Residual Learning in Spiking Neural Networkscited by 326
- 2020Going Deeper With Directly-Trained Larger Spiking Neural Networkscited by 268
- 2022Spikformer: When Spiking Neural Network Meets Transformercited by 263
- 2018Bi-Real Net: Enhancing the Performance of 1-bit CNNs With Improved Representational Capability and Advanced Training Algorithmcited by 233
- 2022Versatile Diffusion: Text, Images and Variations All in One Diffusion Modelcited by 199
- 2022Temporal Efficient Training of Spiking Neural Network via Gradient Re-weightingcited by 192
- 2019Spiking-YOLO: Spiking Neural Network for Energy-Efficient Object Detectioncited by 174
- 2021Optimal Conversion of Conventional Artificial Neural Networks to Spiking Neural Networkscited by 134
- 2019Enabling Spike-based Backpropagation for Training Deep Neural Network Architecturescited by 119
- 2023Optimal ANN-SNN Conversion for High-accuracy and Ultra-low-latency Spiking Neural Networkscited by 116
- 2021A Free Lunch From ANN: Towards Efficient, Accurate Spiking Neural Networks Calibrationcited by 111
- 2022Attention Spiking Neural Networkscited by 103
- 2021Optimal ANN-SNN Conversion for Fast and Accurate Inference in Deep Spiking Neural Networkscited by 74
- 2022Equivariant Energy-Guided SDE for Inverse Molecular Designcited by 72
- 2015Spiking Deep Networks with LIF Neuronscited by 63
- 2021PokeBNN: A Binary Pursuit of Lightweight Accuracycited by 48
- 2023Fast-SNN: Fast Spiking Neural Network by Converting Quantized ANNcited by 42
- 2021Training Feedback Spiking Neural Networks by Implicit Differentiation on the Equilibrium Statecited by 39
- 2021Fully Spiking Variational Autoencodercited by 22
- 2021Learning to Time-Decode in Spiking Neural Networks Through the Information Bottleneckcited by 15
- 2021Spiking-GAN: A Spiking Generative Adversarial Network Using Time-To-First-Spike Codingcited by 8
- 2021Spiking Generative Adversarial Networks With a Neural Network Discriminator: Local Training, Bayesian Models, and Continual Meta-Learningcited by 8
- 2023Spiking Generative Adversarial Network with Attention Scoring Decodingcited by 7
- 2021Encoding Event-Based Data With a Hybrid SNN Guided Variational Auto-encoder in Neuromorphic Hardwarecited by 5
Built on by
Citing papers per year: 2025: 3.
3 papers.
Similar, not cited
Close in wording, no citation either way
- 2024SDiT: Spiking Diffusion Model with Transformer
- 2025Biologically Inspired Spiking Diffusion Model with Adaptive Lateral Selection Mechanism
- 2024Latent Diffusion for Neural Spiking Data
- 2024SPikE-SSM: A Sparse, Precise, and Efficient Spiking State Space Model for Long Sequences Learning
- 2024SpikingSSMs: Learning Long Sequences with Sparse and Parallel Spiking State Space Models
- 2023In search of dispersed memories: Generative diffusion models are associative memory networks
- 2024Brain-inspired Action Generation with Spiking Transformer Diffusion Policy Model
- 2024SDP: Spiking Diffusion Policy for Robotic Manipulation with Learnable Channel-Wise Membrane Thresholds