Skip to content

Commit 88c27cf

Browse files
committed
Cite A Spike In Performance in SpikingActivation
1 parent 3e17bbb commit 88c27cf

File tree

1 file changed

+16
-2
lines changed

1 file changed

+16
-2
lines changed

keras_spiking/layers.py

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -223,7 +223,7 @@ class SpikingActivationCell(KerasSpikingCell):
223223
spiking output will be the same as the integral of the base activation output).
224224
Note that if the base activation is outputting a negative value then the spikes
225225
will have height ``-1/dt``. Multiple spikes per timestep are also possible, in
226-
which case the output will be ``n/dt`` (where ``n`` is the number of spikes).
226+
which case the output will be ``n/dt`` (where ``n`` is the number of spikes). [1]_
227227
228228
Notes
229229
-----
@@ -256,6 +256,13 @@ class SpikingActivationCell(KerasSpikingCell):
256256
kwargs : dict
257257
Passed on to `tf.keras.layers.Layer
258258
<https://www.tensorflow.org/api_docs/python/tf/keras/layers/Layer>`_.
259+
260+
References
261+
----------
262+
.. [1] Voelker, A. R., Rasmussen, D., & Eliasmith, C. (2020). A Spike in
263+
Performance: Training Hybrid-Spiking Neural Networks with Quantized Activation
264+
Functions. arXiv preprint arXiv:2002.03553.
265+
(https://export.arxiv.org/abs/2002.03553)
259266
"""
260267

261268
def __init__(
@@ -387,7 +394,7 @@ class SpikingActivation(KerasSpikingLayer):
387394
spiking output will be the same as the integral of the base activation output).
388395
Note that if the base activation is outputting a negative value then the spikes
389396
will have height ``-1/dt``. Multiple spikes per timestep are also possible, in
390-
which case the output will be ``n/dt`` (where ``n`` is the number of spikes).
397+
which case the output will be ``n/dt`` (where ``n`` is the number of spikes). [1]_
391398
392399
When applying this layer to an input, make sure that the input has a time axis
393400
(the ``time_major`` option controls whether it comes before or after the batch
@@ -444,6 +451,13 @@ class SpikingActivation(KerasSpikingLayer):
444451
kwargs : dict
445452
Passed on to `tf.keras.layers.Layer
446453
<https://www.tensorflow.org/api_docs/python/tf/keras/layers/Layer>`_.
454+
455+
References
456+
----------
457+
.. [1] Voelker, A. R., Rasmussen, D., & Eliasmith, C. (2020). A Spike in
458+
Performance: Training Hybrid-Spiking Neural Networks with Quantized Activation
459+
Functions. arXiv preprint arXiv:2002.03553.
460+
(https://export.arxiv.org/abs/2002.03553)
447461
"""
448462

449463
def __init__(

0 commit comments

Comments
 (0)