Energy efficient synaptic plasticity
Abstract
Many aspects of the brain’s design can be understood as the result of evolutionary drive toward metabolic efficiency. In addition to the energetic costs of neural computation and transmission, experimental evidence indicates that synaptic plasticity is metabolically demanding as well. As synaptic plasticity is crucial for learning, we examine how these metabolic costs enter in learning. We find that when synaptic plasticity rules are naively implemented, training neural networks requires extremely large amounts of energy when storing many patterns. We propose that this is avoided by precisely balancing labile forms of synaptic plasticity with more stable forms. This algorithm, termed synaptic caching, boosts energy efficiency manifold and can be used with any plasticity rule, including backpropagation. Our results yield a novel interpretation of the multiple forms of neural synaptic plasticity observed experimentally, including synaptic tagging and capture phenomena. Furthermore, our results are relevant for energy efficient neuromorphic designs.
eLife digest
The brain expends a lot of energy. While the organ accounts for only about 2% of a person’s bodyweight, it is responsible for about 20% of our energy use at rest. Neurons use some of this energy to communicate with each other and to process information, but much of the energy is likely used to support learning. A study in fruit flies showed that insects that learned to associate two stimuli and then had their food supply cut off, died 20% earlier than untrained flies. This is thought to be because learning used up the insects’ energy reserves.
If learning a single association requires so much energy, how does the brain manage to store vast amounts of data? Li and van Rossum offer an explanation based on a computer model of neural networks. The advantage of using such a model is that it is possible to control and measure conditions more precisely than in the living brain.
Analysing the model confirmed that learning many new associations requires large amounts of energy. This is particularly true if the memories must be stored with a high degree of accuracy, and if the neural network contains many stored memories already. The reason that learning consumes so much energy is that forming longterm memories requires neurons to produce new proteins. Using the computer model, Li and van Rossum show that neural networks can overcome this limitation by storing memories initially in a transient form that does not require protein synthesis. Doing so reduces energy requirements by as much as 10fold.
Studies in living brains have shown that transient memories of this type do in fact exist. The current results hence offer a hypothesis as to how the brain can learn in a more energy efficient way. Energy consumption is thought to have placed constraints on brain evolution. It is also often a bottleneck in computers. By revealing how the brain encodes memories energy efficiently, the current findings could thus also inspire new engineering solutions.
Introduction
The human brain only weighs 2% of the total body mass but is responsible for 20% of resting metabolism (Attwell and Laughlin, 2001; Harris et al., 2012). The brain’s energy need is believed to have shaped many aspects of its design, such as its sparse coding strategy (Levy and Baxter, 1996; Lennie, 2003), the biophysics of the mammalian action potential (Alle et al., 2009; Fohlmeister, 2009), and synaptic failure (Levy and Baxter, 2002; Harris et al., 2012). As the connections in the brain are adaptive, one can design synaptic plasticity rules that further reduce the energy required for information transmission, for instance by sparsifying connectivity (Sacramento et al., 2015). But in addition to the costs associated to neural information processing, experimental evidence suggests that memory formation, presumably corresponding to synaptic plasticity, is itself an energetically expensive process as well (Mery and Kawecki, 2005; Plaçais and Preat, 2013; Jaumann et al., 2013; Plaçais et al., 2017).
To estimate the amount of energy required for plasticity, Mery and Kawecki (2005) subjected fruit flies to associative conditioning spaced out in time, resulting in longterm memory formation. After training, the fly’s food supply was cut off. Flies exposed to the conditioning died some 20% quicker than control flies, presumably due to the metabolic cost of plasticity. Likewise, fruit flies doubled their sucrose consumption during the formation of aversive longterm memory (Plaçais et al., 2017), while forcing starving fruit flies to form such memories reduced lifespan by 30% (Plaçais and Preat, 2013). A massed learning protocol, where pairings are presented rapidly after one another, leads to less permanent forms of learning that don’t require protein synthesis. Notably this form of learning is energetically less costly (Mery and Kawecki, 2005; Plaçais and Preat, 2013). In rats (Gold, 1986) and humans (Hall et al., 1989, but see Azari, 1991) beneficial effects of glucose on memory have been reported, although the intricate regulation of energy complicates interpretation of such experiments (Craft et al., 1994).
Motivated by the experimental results, we analyze the metabolic energy required to form associative memories in neuronal networks. We demonstrate that traditional learning algorithms are metabolically highly inefficient. Therefore, we introduce a synaptic caching algorithm that is consistent with synaptic consolidation experiments, and distributes learning over transient and persistent synaptic changes. This algorithm increases efficiency manifold. Synaptic caching yields a novel interpretation to various aspects of synaptic physiology, and suggests more energy efficient neuromorphic designs.
Results
Inefficiency of perceptron learning
To examine the metabolic energy cost associated to synaptic plasticity, we first study the perceptron. A perceptron is a single artificial neuron that attempts to binary classify input patterns. It forms the core of many artificial networks and has been used to model plasticity in cerebellar Purkinje cells. We consider the common case where the input patterns are random patterns each associated to a randomly chosen binary output. Upon presentation of a pattern, the perceptron output is calculated and compared to the desired output. The synaptic weights are modified according to the perceptron learning rule, Figure 1A. This is repeated until all patterns are classified correctly (Rosenblatt, 1962, see Materials and methods). Typically, the learning takes multiple iterations over the whole dataset (’epochs’).
As it is not well known how much metabolic energy is required to modify a biological synapse, and how this depends on the amount of change and the sign of the change, we propose a parsimonious model. We assume that the metabolic energy for every modification of a synaptic weight is proportional to the amount of change, no matter if this is positive or negative. The total metabolic cost $M$ (in arbitrary units) to train a perceptron is the sum over the weight changes of synapses
where $N$ is the number of synapses, ${w}_{i}$ denotes the synaptic weight at synapse $i$, and $T$ is the total number of timesteps required to learn the classification. The exponent $\alpha $ is set to one, but our results below are similar whenever $0\le \alpha \lesssim 2$, Figure 1—figure supplement 1. As there is evidence that synaptic depression involves different pathways than synaptic potentiation (e.g. Hafner et al., 2019), we also tried a variant of the cost function where only potentiation costs energy and depression does not. This does not change our results, Figure 1—figure supplement 1.
Learning can be understood as a search in the space of synaptic weights for a weight vector that leads to correct classification of all patterns, Figure 1B. The synaptic weights approximately follow a random walk (Materials and methods), and the metabolic cost is proportional to the length of this walk under the ${L}_{1}$ norm, Equation 1. The perceptron learning rule is energy inefficient, because repeatedly, weight modifications made to correctly classify one pattern are partly undone when learning another pattern. However, as both processes require energy, this is inefficient.
The energy required by the perceptron learning rule depends on the number of patterns $P$ to be classified. The set of correct weights spans a cone in $N$dimensional space (grey region in Figure 1B). As the number of patterns to be classified increases, the cone containing correct weights shrinks and the random walk becomes longer (Gardner, 1987). Near the critical capacity of the perceptron ($P=2N$), the number of epochs required diverges as ${(2P/N)}^{2}$, Opper (1988). The energy required, which is proportional to the number of updates that the weights undergo, follows a similar behavior, Figure 1C.
It is useful to consider the theoretical minimal energy required to classify all patterns. The most energy efficient algorithm would somehow directly set the synaptic weights to their desired final values. Geometrically, the random walk trajectory of the synaptic weights to the target is replaced by a path straight to the correct weights (green arrow in Figure 1B). Given the initial weights ${w}_{i}(0)$ and the final weights ${w}_{i}(T)$, the energy required in this idealized case is
While the minimal energy also grows with memory load (Materials and methods), it increases less steeply, Figure 1C.
We express the metabolic efficiency of a learning algorithm as the ratio between the energy the algorithm requires and the minimal energy (the gap between the two logscale curves in Figure 1C). As the number of patterns increases, the inefficiency of the perceptron rule rapidly grows as (see Materials and methods)
which fits the simulations very well, Figure 1D, black curve and dashed blue curve.
There is evidence that both cerebellar and cortical neurons are operating close to their maximal memory capacity (Brunel et al., 2004; Brunel, 2016). Indeed, it would appear wasteful if this were not the case. However, the above result demonstrates that for instance classifying 1900 patterns by a neuron with 1000 synapses with the traditional perceptron learning requires about ∼900 times more energy than minimally required. As the fruitfly experiments indicate that even storing a single association in longterm memory is already metabolically expensive, storing many memories would thus require very large amounts of energy if the biology would naively implement these learning rules.
Synaptic caching
How can the conflicting demands of energy efficiency and high storage capacity be met? The minimal energy argument presented above suggests a way to increase energy efficiency. There are forms of plasticity – anesthesiaresistant memory in flies and earlyLTP/LTD in mammals – that decay and do not require protein synthesis. Such transient synaptic changes can be induced using a massed, instead of a spaced, stimulus presentation protocol. Fruitfly experiments show that this form of plasticity is much less energydemanding than longterm memory (Mery and Kawecki, 2005; Plaçais and Preat, 2013; Plaçais et al., 2017). In mammals, there is evidence that synaptic consolidation, but not transient plasticity, is suppressed under lowenergy conditions (Potter et al., 2010). Inspired by these findings, we propose that the transient form of plasticity constitutes a synaptic variable that accumulates the synaptic changes across multiple updates in a less expensive transient form of memory; only occasionally the changes are consolidated. We call this synaptic caching.
Specifically, we assume that each synapse is comprised of a transient component ${s}_{i}$ and a persistent component ${l}_{i}$. The total synaptic weight is their sum, ${w}_{i}={s}_{i}+{l}_{i}$. We implement synaptic caching as follows, Figure 2A: For every presented pattern, changes in the synaptic strength are calculated according to the perceptron rule and are accumulated in the transient component that decays exponentially to zero. If, however, the absolute value of the transient component of a synapse exceeds a certain consolidation threshold, all synapses of that neuron are consolidated (vertical dashed line in Figure 2A); the value of the transient component is added to the persistent weight; and the transient weight is reset to zero.
The efficiency gain of synaptic caching depends on the limitations of transient plasticity. If the transient synaptic component could store information indefinitely at no metabolic cost, consolidation could be postponed until the end of learning and the energy would equal the minimal energy Equation 2. Hence the efficiency gain would be maximal. However, we assume that the efficiency gain of synaptic caching is limited because of two effects: (1) The transient component decays exponentially (with a timeconstant $\tau $). (2) There might be a maintenance cost associated to maintaining the transient component. Biophysically, transient plasticity might correspond to an increased/decreased vesicle release rate (Padamsey and Emptage, 2014; Costa et al., 2015) so that it diverges from its optimal value (Levy and Baxter, 2002).
To estimate the energy saved by synaptic caching, we assume that the maintenance cost is proportional to the transient weight itself and incurred every timestep $\mathrm{\Delta}t$ (shaded area in the top traces of Figure 2A)
While experiments indicate that transient plasticity is metabolically far less demanding than the persistent form, the precise value of the maintenance cost is not known. We encode it in the constant $c$; the theory also includes the case that $c$ is zero. It is straightforward to include a cost term for changing the transient weight (Materials and methods); such a cost would reduce the efficiency gain attainable by synaptic caching.
Next, we need to include the energetic cost of consolidation. Currently it is unknown how different components of synaptic consolidation, such as signaling, protein synthesis, transport to the synapses and changing the synapse, contribute to this cost. We assume the metabolic cost to consolidate the synaptic weights is ${M}_{\mathrm{cons}}={\sum}_{i}{\sum}_{t}{l}_{i}(t){l}_{i}(t1)$. This is identical to Equation 1, but in contrast to standard perceptron learning where synapses are consolidated every time a weight is updated, now changes in the persistent component ${l}_{i}$ only occur when consolidation occurs. One could add a maintenance cost term to the persistent weight as well, in that case postponing consolidation would save even more energy.
Efficiency gain from synaptic caching
To maximize the efficiency gain achieved by synaptic caching one needs to tune the consolidation threshold, Figure 2B. When the threshold is low, consolidation occurs often and the energy approaches the one without synaptic caching. When on the other hand the consolidation threshold is high, the expensive consolidation process occurs rarely, but the maintenance cost of transient plasticity is high; moreover, the decay will lead to forgetting of unconsolidated memories, slowing down learning and increasing the energy cost. Thus, the consolidation energy decreases for larger thresholds, whereas the maintenance energy increases, Figure 2B (see Materials and methods). As a result of this tradeoff, there is an optimal threshold – which depends on the decay and the maintenance cost – that balances persistent and transient forms of plasticity. To analyze the efficiency gain below, we numerically optimize the consolidation threshold.
First, we consider the case when the transient component does not decay. Figure 2C shows the energy required to train the perceptron. When the maintenance cost is absent ($c=0$), consolidation is best postponed until the end of the learning and the energy is as low as the theoretical minimal bound. As $c$ increases, it becomes beneficial to consolidate more often, that is the optimal threshold decreases, Figure 2C bottom panel. The required energy increases until the maintenance cost becomes so high that it is better to consolidate after every update, the transient weights are not used, and no energy is saved with synaptic caching. The efficiency is well estimated by analysis presented in the Materials and methods, Figure 2C (theory).
Next, we consider what happens when the transient plasticity decays. We examine the energy and learning time as a function of the decay rate for various levels of maintenance cost, Figure 3. As stated above, if there is no decay, efficiency gain can be very high; the consolidation threshold has no impact on the learning time, Figure 3 bottom. In the other limit, when the decay is rapid (rightmost region), it is best to consolidate frequently as otherwise information is lost. As expected, the metabolic cost is high in this case.
The regime of intermediate decay is quite interesting. When maintenance cost is high, it is of primary importance to keep learning time short, and in fact the learning time can be lower than in a perceptron without decay, Figure 3, bottom, light curves. When on the other hand maintenance cost is low, the optimal solution is to set the consolidation threshold high so as to minimize the number of consolidation events, even if this means a longer learning time, Figure 3, bottom, dark curves.
For intermediate decay rates, the consolidation threshold trades off between learning time and energy efficiency, Figure 3—figure supplement 1A. That is, by setting the consolidation threshold the perceptron can learn either rapidly or efficiently. Such a tradeoff could be of biological relevance. We found a similar tradeoff in multilayer perceptrons (see below), Figure 3—figure supplement 1B. (although we found no evidence that learning can be sped up there).
In summary, when the transient component decays the learning dynamics is altered, and synaptic caching can not only reduce metabolic cost but can also reduce learning time.
Next, to show that synaptic caching is a general principle, we implement synaptic caching in a spiking neural network with a biologically plausible perceptronlike learning rule proposed by D'Souza et al. (2010). The optimal scenario, where the transient weights do not decay and have no maintenance cost, is assumed. The network is able to save 80% of the energy with synaptic caching, Figure 2—figure supplement 1. Hence, efficiency gains from synaptic caching do not rely on exact implementation.
In the above implementation of synaptic caching, consolidation of all synapses was triggered when transient plasticity at a single synapse exceeded a certain threshold. This resembles the synaptic tagging and capture phenomenon where plasticity induction leads to transient changes and sets a tag; only strong enough stimulation results in proteins being synthesized and being delivered to all tagged synapses, consolidating the changes (Frey and Morris, 1997; Barrett et al., 2009). There is a number of ways synapses could interact, Figure 4A. First, consolidation might be set to occur whenever transient plasticity at a synapse crosses the threshold and only that synapse is consolidated. Second, a hypothetical signal might send to the soma and consolidation of all synapses occurs once transient plasticity at any synapse crosses the threshold (used in Figures 2 and 5). Third, a hypothetical signal might be accumulated in or near the soma and consolidation of all synapses occurs once this total transient plasticity across synapses crosses the threshold. Only cases 2 and 3 are consistent with synaptic tagging and capture experiments, where consolidation of one synapse also leads to consolidation of another synapse that would otherwise decay back to baseline (Frey and Morris, 1997; Sajikumar et al., 2005). However, all variants lead to comparable efficiency gains, Figure 4B.
In summary, we see that synaptic caching can in principle achieve large efficiency gains, bringing efficiency close to the theoretical minimum.
Synaptic caching in multilayer networks
Since the perceptron is a rather restrictive framework, we wondered whether the efficiency gain of synaptic caching can be transferred to multilayer networks. Therefore, we implement a multilayer network trained with backpropagation. Backpropagation networks learn the associations of patterns by approaching the minimum of the error function through stochastic gradient descent. We use a network with one hidden layer with by default 100 units to classify handwritten digits from the MNIST dataset. As we train the network, we intermittently interrupt the learning to measure the energy consumed for plasticity thus far and measure the performance on a heldout testset. This yields a curve relating energy to accuracy.
Similar to a perceptron, learning without synaptic caching is metabolically expensive in a backpropagation network. Until reaching maximal accuracy, energy rises approximately exponentially with accuracy, after which additional energy do not lead to further improvement. When the learning rate is sufficiently small, the metabolic cost of plasticity is independent of the learning rate. At larger learning rates, learning no longer converges and energy goes up steeply without an increase in accuracy, Figure 5A. With the exception of these very large rates, these results show that lowering the learning rate does not save energy.
Similar to the perceptron, we evaluate how much energy would be required to directly set the synaptic weights to their final values. Traditional learning without synaptic caching is once again energetically inefficient, expending at least ∼20 times more energy compared to this theoretical minimum whatever the desired accuracy level is, Figure 5B. However, by splitting the weights into persistent synaptic weights and transient synaptic caching weights, the network can save substantial amounts of energy. As for the perceptron, depending on the decay and the maintenance cost the energy ranges from as little as the minimum to as much as the energy required without caching. Thus, the efficiency gain of synaptic caching found for the perceptron carries over to multilayer networks.
It might seem that smaller networks would be metabolically less costly, because small networks simply contain fewer synapses to modify. On the other hand, we saw above that for the perceptron metabolic costs rise rapidly when cramming many patterns into it. We wondered therefore how energy cost depends on network size in the multilayer network. Since the number of input units is fixed to the image size and the number of output units equals the ten output categories, we adjust the number of hidden units.
The network fails to reach the desired accuracy if the number of hidden units is too small, Figure 5C. When the network size is barely above the minimum requirement, the network has to compensate the lack of hidden units with longer training time and hence a larger energy expenditure. However, very large networks also require more energy. These results show that from an energy perspective there exists an optimal number of neurons to participate in memory formation. The optimal number depends on the accuracy requirement; as expected, higher accuracies require more hidden units and energy.
Discussion
Experiments on formation of a longterm memory of a single association suggest that synaptic plasticity is an energetically expensive process. We have shown that energy requirements rise steeply as memory load or designated accuracy level increase. This indicates tradeoffs between energy consumption, and network capacity and performance. To improve efficiency, we have proposed an algorithm named synaptic caching that temporarily stores changes in the synaptic strength in transient forms of plasticity, and only occasionally consolidates into the persistent forms. Depending on the characteristics (decay and maintenance cost) of transient plasticity, this can lead to large energy savings in the energy required for synaptic plasticity. We stress that from an algorithmic point of view, synaptic caching can be applied to any synaptic learning algorithm (unsupervised, reinforcement, supervised) and does not have specific requirements. Further savings might be possible by adjusting the consolidation threshold as learning progresses and by being pathwayspecific (Leibold and MonsalveMercado, 2016).
The implementation of a consolidation threshold is similar to what has been observed in physiology, in particular in the synaptic tagging and capture literature (Redondo and Morris, 2011). Our results thus give a novel interpretation of those findings. Synaptic consolidation is known to be affected by reward, novelty and punishment (Redondo and Morris, 2011), which is compatible with a metabolic perspective as energy is expended only when the stimulus is worth remembering. In addition, our results for instance explain why consolidation is competitive, but transient plasticity is less so (Sajikumar et al., 2014), namely the formation of longterm memory is precious. Consistent with this, there is evidence that encouraging consolidation increases energy consumption (Plaçais et al., 2017). We also predict that the transient weight changes act as an accumulative threshold for consolidation. That is, sufficient transient plasticity should trigger consolidation, even in the absence of other consolidation triggers. Future characterization of the energy budget of synaptic plasticity should allow more precise predictions of our theory.
Combining persistent and transient storage mechanisms is a strategy well known in traditional computer systems to provide a faster and often energetically cheaper access to memory. In computer systems, permanent storage of memories typically requires transmission of all information across multiple transient cache systems until reaching a longterm storage device. The transfer of information is often a bottleneck in computer architectures and consumes considerable power in modern computers (Kestor et al., 2013). However, in the nervous system transient and persistent synapses appear to exist next to each other. Local consolidation in a synapse does not require moving information. Using this setup, biology appears to have found a more efficient way to store information.
Memory stability has long fascinated researchers (Richards and Frankland, 2017), and in some cases forgetting can be beneficial (Brea et al., 2014). Splitting plasticity into transient and persistent forms might prevent catastrophic forgetting in networks (Leimer et al., 2019). Here, we argue that the main benefit of more transient forms of plasticity is to permit the network to explore the weight space to find a desirable weight configuration using less energy. While this work focuses solely on the metabolic cost of synaptic plasticity, the brain also expends significant amounts of energy on spiking, synaptic transmission, and maintaining resting potential. Learning rules can be designed to reduce costs associated to computation once learning has finished (Sacramento et al., 2015). It would be of interest to next understand the precise interaction of computation and plasticity cost during and after learning.
Materials and methods
Energy efficiency of the perceptron
Request a detailed protocolFor perceptron, we can calculate the energy efficiency of both the classical perceptron and the gain achieved by synaptic caching. We first consider the case that transient plasticity does not decay, as this allows important theoretical simplifications. In the perceptron learning to classify binary patterns Equation 8, the weight updates are either $+\eta $ or $\eta $, where $\eta $ is the learning rate, so that the energy spent (Equation 1, $\alpha =1$) per update per synapse equals $\eta $. Hence the total energy spent to classify all patterns ${M}_{\mathrm{perc}}=NK\eta $, where $K$ is the total number of updates. Opper (1988) showed that learning time diverges as $K\sim {(2P/N)}^{2}$. We found the numerator numerically to yield $K=2P/{(2P/N)}^{2}$.
To calculate the efficiency, we compare this to the minimal energy necessary to reach the final weight vector in the perceptron. We approximate the weight trajectory followed by the perceptron algorithm by a random walk. After $K$ updates of stepsize $\eta $ the weights approximate a Gaussian distribution with zero mean and variance $K{\eta}^{2}$. By shortcutting the random walk, the minimal energy required to reach the weight vector is ${M}_{\mathrm{min}}=N\u27e8{w}_{i}\u27e9=\sqrt{\frac{2}{\pi}}\eta N\sqrt{K}$. Hence, we find for the inefficiency (see Figure 1D)
Simulations show that the variance in the weights is actually about 20% smaller than a random walk, likely reflecting correlations in the learning process not captured in the random walk approximation. This explains most of the slight deviation in the ineffeciency between theory and simulation, Figure 1D.
Efficiency of synaptic caching
Request a detailed protocolTo calculate the efficiency gained with synaptic caching, we need to calculate both the consolidation energy and the maintenance energy. The consolidation energy equals the number of consolidation events times the size of the updates. The size of the weight updates is equal to the consolidation threshold $\theta $, while the number of consolidation events follows from a random walk argument as $NK/{\lceil \theta /\eta \rceil}^{2}$. The ceiling function expresses the fact that when the threshold is smaller than learning rate, consolidation will always occur; we temporarily ignore this scenario. In addition, at the end of learning all remaining transient plasticity is consolidated, which requires an energy $N\u27e8{s}_{i}(T)\u27e9$. Assuming that the probability distribution of transient weights, ${P}_{s}(s)$, has reached steady state at the end of learning, it has a triangular shape (see below) and mean absolute value $\u27e8{s}_{i}(T)\u27e9=\frac{1}{3}\theta $, so that the total consolidation energy
The energy associated to the transient plasticity is (again assuming that ${P}_{s}(s)$ has reached steady state)
where $T$ is the number of timesteps required for learning. We find numerically that $T=\frac{{P}^{3/2}}{{(2P/N)}^{2}}$. Hence the total energy when using synaptic caching is ${M}_{\mathrm{cache}}={M}_{\mathrm{cons}}+{M}_{\mathrm{trans}}=N\left[{\eta}^{2}K/\theta +\frac{1}{3}\theta (1+cT)\right]$. The optimal threshold $\widehat{\theta}$ is given by $\frac{d}{d\theta}\left[{M}_{\mathrm{cons}}+{M}_{\mathrm{trans}}\right]=0$, or
at which the energy is ${M}_{\mathrm{cache}}=2\eta N\sqrt{K}\sqrt{1+cT}/\sqrt{3}$. And so the efficiency of synaptic caching is $\frac{{M}_{\mathrm{cache}}}{{M}_{\mathrm{min}}}=\sqrt{\frac{2\pi}{3}}\sqrt{1+cT}$. However, as consolidation can maximally occur only once per timestep, ${M}_{\mathrm{cons}}$ cannot exceed ${M}_{\mathrm{perc}}$ so that the inefficiency is
This equation reasonably matches the simulations, Figure 2C (labeled ’theory’).
One can include a cost for changing the transient weight, so that ${M}_{\mathrm{trans}}=c{\sum}_{i}{\sum}_{t}{s}_{i}(t)+b{\sum}_{i}{\sum}_{t}{s}_{i}(t+1){s}_{i}(t)$, where $b$ codes the cost of making a change. Assuming that consolidating immediately after a weight change does not incur this cost, this yields an extra term in Equation4 of $bNK(11/{\lceil \theta /\eta \rceil}^{2})$. Such costs will reduce the efficiency gain achievable by synaptic caching. When $b\ge 1$, it is always cheaper to consolidate.
Decaying transient plasticity
Request a detailed protocolWhen transient plasticity decays, the situation is more complicated as the learning time depends on the strength of the decay and to our knowledge no analytical expression exists for it. However, it is still possible to estimate the power, that is the energy per time unit, for both the transient component, denoted ${m}_{\mathrm{trans}}$, and the consolidation component, ${m}_{\mathrm{cons}}$. Under the random walk approximation every time the perceptron output does not match the desired output, the transient weight ${s}_{i}$ is updated with an amount $\mathrm{\Delta}{s}_{i}$ drawn from a distribution $Q$, with zero mean and variance ${\sigma}^{2}$. Given the update probability $p$, that is the fraction of patterns not yet classified correctly, one has ${Q}_{s}(\eta )={Q}_{s}(\eta )=p/2$ and ${Q}_{s}(0)=1p$, so that ${\sigma}_{s}^{2}=p{\eta}^{2}$. We assume that the synaptic update rate decreases very slowly as learning progresses, hence $p$ is quasistationary.
Every timestep $\mathrm{\Delta}t=1$ the transient weights decay with a timeconstant $\tau $. The synapse is consolidated and ${s}_{i}$ is reset to zero whenever the absolute value of the caching weight ${s}_{i}$ exceeds $\theta $. Given $p$ and $\tau $, we would like to know: 1) how often consolidation events occur which gives consolidation power and 2) the maintenance power ${m}_{\mathrm{trans}}=cN\u27e8{s}_{i}\u27e9$. This problem is similar to the random walk to threshold model used for integrateandfire neurons, but here there are two thresholds: $\theta $ and $\theta $.
Under the assumptions of small updates and a smooth resulting distribution, the evolution of the probability distribution ${P}_{s}({s}_{i})$ is described by the FokkerPlanck equation, which in the steady state gives
The last term is a source term that describes the reinsertion of weights by the reset process. The boundary conditions are ${P}_{s}({s}_{i}=\pm \theta )=0$. While ${P}_{s}({s}_{i})$ is continuous in ${s}_{i}$, the source introduces a cusp in ${P}_{s}({s}_{i})$ at the reset value. Conservation of probability ensures that $r$ equals the outgoing flux at the boundaries. One finds
where $\mathrm{erfi}(x)=i\mathrm{erf}(ix)$, ${\sigma}^{2}=\frac{\tau}{\mathrm{\Delta}t}{\sigma}_{s}^{2}$ and with normalization factor
where ${}_{2}F_{2}$ is the generalized hypergeometric function. (In the limit of no decay this becomes a triangular distribution ${P}_{s}({s}_{i})=[\theta {s}_{i}]/{\theta}^{2}$.)
We obtain maintenance power
For small $\theta /\sigma $, that is small decay, this is linear in $\theta $, ${m}_{\mathrm{trans}}\approx \frac{cN\theta}{3}$. It saturates for large $\theta $ because then the decay dominates and the threshold is hardly ever reached.
The consolidation rate follows from Fick’s law
The consolidation power is
In the limit of no decay one has $r={\sigma}^{2}/{\theta}^{2}$, so that ${m}_{\mathrm{cons}}=pN{\eta}^{2}/\theta $. Strictly speaking this approximates learning with a random walk process and assumes local consolidation, Figure 4A. However, Equations 6 and 7 give a good prediction of the simulation when provided with the timevarying update probability from the simulation, Figure 6.
Simulations
Perceptron
Request a detailed protocolUnless stated otherwise, we use a perceptron with $N=1000$ input units to classify $P=N$ random binary (±1 with equal probability) input patterns ${\mathit{\bm{x}}}^{(p)}$, each to be associated to a randomly assigned desired binary output ${d}^{(p)}$. Each input unit is connected with a weight ${w}_{i}$ signifying the strength of the connection. An ’alwayson’ bias unit with corresponding weight is included to adjust the threshold of the perceptron. The perceptron output $y$ of a pattern is determined by the Heaviside step function $\mathrm{\Theta}$, $y=\mathrm{\Theta}(\mathrm{w}.\mathit{\bm{x}})$. If for a given pattern $p$, the output does not match the desired pattern output, $\mathit{\bm{w}}$ is adjusted according to
where the learning rate $\eta $ can be set to one without loss of generality. The perceptron algorithm cycles through all patterns until classified correctly. In principle, the magnitude of the weight vector, and hence the minimal energy, can be arbitrarily small for a noisefree binary perceptron. However, this paradox is resolved as soon as robustness to any postsynaptic noise is required.
Multilayer networks
Request a detailed protocolFor the multilayer networks trained on MNIST, we use networks with one hidden layer, logistic units, and onehot encoding at the output. Weights are updated according to the mean squared error backpropagation rule without regularization.
Simulation scripts for both the perceptron and the multilayer network can be found at https://github.com/vanrossumlab/li_vanrossum_19. (Li and van Rossum, 2020; copy archived at https://github.com/elifesciencespublications/li_vanrossum_19).
References
 1

2
An energy budget for signaling in the grey matter of the brainJournal of Cerebral Blood Flow & Metabolism 21:1133–1145.https://doi.org/10.1097/0000464720011000000001

3
Effects of glucose on memory processes in young adultsPsychopharmacology 105:521–524.https://doi.org/10.1007/BF02244373

4
State based model of longterm potentiation and synaptic tagging and capturePLOS Computational Biology 5:e1000259.https://doi.org/10.1371/journal.pcbi.1000259

5
A normative theory of forgetting: lessons from the fruit flyPLOS Computational Biology 10:e1003640.https://doi.org/10.1371/journal.pcbi.1003640
 6

7
Is cortical connectivity optimized for storing information?Nature Neuroscience 19:749–755.https://doi.org/10.1038/nn.4286
 8
 9
 10
 11
 12

13
Maximum storage capacity in neural networksEurophysics Letters 4:481–485.https://doi.org/10.1209/02955075/4/4/016

14
Glucose modulation of memory storage processingBehavioral and Neural Biology 45:342–349.https://doi.org/10.1016/S01631047(86)80022X
 15
 16
 17
 18

19
Quantifying the energy cost of data movement in scientific applicationsIEEE International Symposium on Workload Characterization (IISWC).https://doi.org/10.1109/IISWC.2013.6704670

20
Asymmetry of neuronal combinatorial codes arises from minimizing synaptic weight changeNeural Computation 28:1527–1552.https://doi.org/10.1162/NECO_a_00854
 21

22
The cost of cortical computationCurrent Biology 13:493–497.https://doi.org/10.1016/S09609822(03)001350

23
Energy efficient neural codesNeural Computation 8:531–543.https://doi.org/10.1162/neco.1996.8.3.531

24
Energyefficient neuronal computation via quantal synaptic failuresThe Journal of Neuroscience 22:4746–4755.https://doi.org/10.1523/JNEUROSCI.221104746.2002

25
li_vanrossum_19, version 9fe7761GitHub.
 26

27
Learning times of neural networks: exact solution for a PERCEPTRON algorithmPhysical Review A 38:3824–3826.https://doi.org/10.1103/PhysRevA.38.3824

28
Two sides to longterm potentiation: a view towards reconciliationPhilosophical Transactions of the Royal Society B: Biological Sciences 369:20130154.https://doi.org/10.1098/rstb.2013.0154
 29
 30
 31

32
Making memories last: the synaptic tagging and capture hypothesisNature Reviews Neuroscience 12:17–30.https://doi.org/10.1038/nrn2963
 33

34
Brain TheoryPrinciples of neurodynamics: perceptrons and the theory of brain mechanisms' to Van Der Malsburg C. (1986) Frank Rosenblatt: Principles of Neurodynamics: Perceptrons and the Theory of Brain Mechanisms, Brain Theory, Berlin, Heidelberg, Springer, 10.1007/9783642709111_20.

35
Energy efficient sparse connectivity from imbalanced synaptic plasticity rulesPLOS Computational Biology 11:e1004265.https://doi.org/10.1371/journal.pcbi.1004265
 36
 37
Decision letter

Peter LathamReviewing Editor; University College London, United Kingdom

Ronald L CalabreseSenior Editor; Emory University, United States

Walter SennReviewer; University of Bern, Switzerland
In the interests of transparency, eLife publishes the most substantive revision requests and the accompanying author responses.
Acceptance summary:
The authors investigate the issue of energy consumption in synapses. They show that under a strategy in which much of the information about weights is kept in temporary storage while permanent weight changes are rare, energy consumption can be reduced by an order of magnitude. There has been a great deal of work on energy consumption associated with action potentials and synaptic plasticity, but this is, to our knowledge, the first to consider energy efficiency in the context of learning. As such it fills an important gap in our understanding of synaptic plasticity. This paper should appeal to anybody who is interested either in synaptic plasticity or energy efficiency in the brain. It may also be important for learning in artificial systems, where energy costs for training networks that are small by brain standards can exceed millions of kilowatthours.
Decision letter after peer review:
Thank you for submitting your article "Energy efficient synaptic plasticity" for consideration by eLife. Your article has been reviewed by three peer reviewers, one of whom is a member of our Board of Reviewing Editors, and the evaluation has been overseen by Ronald Calabrese as the Senior Editor. The following individual involved in review of your submission has agreed to reveal their identity: Walter Senn (Reviewer #2).
The reviewers have discussed the reviews with one another and the Reviewing Editor has drafted this decision to help you prepare a revised submission.
Summary:
The authors investigate the issue of energy consumption in synapses. They show that under a strategy in which much of the information about weights is kept in temporary storage while permanent weight changes are rare, energy consumption can be reduced by an order of magnitude.
This is an interesting paper that formalizes the concept of energy efficiency in learning. While work exists that considers the energy consumption of action potentials and synaptic plasticity, the formalization of energy efficiency in the context of learning is new, and worth a publication. The paper is well written and the math seems to be sound.
Essential revisions:
1) The authors introduce two components of the synaptic strength, a quickly decaying component that is transcribed into a longlasting component when a threshold is crossed. The quickly decaying component requires less energy, and thus learning becomes a tradeoff between energy efficient storage in the decaying component and energy costly storage in the sustained component.
The decay will lead to forgetting of unconsolidated memories and to a slowdown of learning, together with an increase of energy. As far as we could tell, the paper does not consider the speed of learning, and instead only asks for an energy efficient learning up to a certain degree of accuracy. The authors should provide learning curves for different consolidation thresholds and decay rates. Ideal would be a plot of energy versus learning time – presumably the lower the energy, the longer it takes to achieve a given set of accuracy, although we admit that's only a guess. On the other hand, there may be an optimal threshold.
2) Previous work [e.g., Ziegler, Zenke,.…, Gerstner, 2015, "From synapses to behavioural modelling"; Zenke et al., 2017, "Continual Learning Through Synaptic Intelligence"] has shown a benefit for the 2stage synapses model in terms of learning and forgetting. Is there a similar benefit for this 2stage model? This may simply be a Discussion point, referring to the analysis asked in the point (1) above.
3) From a biological point of view, it is clear that the change as well as the maintenance of synaptic weight can cost energy. Nevertheless, we find it strange that the authors analyze the perceptron learning rule according to a changeonly energy cost function while the synaptic caching rule is analyzed by a combination of change (latephase) and maintenance (earlyphase) cost function. How critical is the phasecost relation? Is it also efficient if the latephase costs maintenance (e.g., by having a larger synaptic apparatus) and the earlyphase costs energy dependent on its change? What is the energy consumption of the perceptron learning rule considering the maintenance cost function?
4) Please test for comparison the energy consumption of other learning rules performing perceptron learning (e.g., D’Souza et al., 2010).
https://doi.org/10.7554/eLife.50804.sa1Author response
Essential revisions:
1) The authors introduce two components of the synaptic strength, a quickly decaying component that is transcribed into a longlasting component when a threshold is crossed. The quickly decaying component requires less energy, and thus learning becomes a tradeoff between energy efficient storage in the decaying component and energy costly storage in the sustained component.
The decay will lead to forgetting of unconsolidated memories and to a slowdown of learning, together with an increase of energy. As far as we could tell, the paper does not consider the speed of learning, and instead only asks for an energy efficient learning up to a certain degree of accuracy. The authors should provide learning curves for different consolidation thresholds and decay rates. Ideal would be a plot of energy versus learning time – presumably the lower the energy, the longer it takes to achieve a given set of accuracy, although we admit that's only a guess. On the other hand, there may be an optimal threshold.
The reviewers raise an interesting, subtle issue that we now discuss in much more detail in Results subsection “Efficiency gain from synaptic caching”. First, we first study a potential tradeoff between energy and learning time in the perceptron. We now plot the relation between the learning time and the energy spend (Figure 3—figure supplement 1A). Depending on maintenance cost c, there are regimes where these form a tradeoff:
When the decay is slow and c is small, the most energysaving scenario is to accumulate as many changes as possible in the transient plasticity that still allow the perceptron to converge, resulting in long learning time. In this case there is a tradeoff between learning time and energy.
However, when c is bigger, the perceptron has to choose an smaller threshold so that maintenance cost is limited. In this regime the energyoptimal threshold is close to the threshold that gives minimal learning time (Figure 3—figure supplement 1A).
Finally, when there is no decay, the energy cost depends on consolidation threshold, but the learning time does not change regardless of the threshold. Hence, there is no interaction between learning time and energy cost.
We also researched this issue in MLPs and found a similar picture: in some regimes there is a tradeoff, but not in others (Figure 3—figure supplement 1B).
2) Previous work [e.g., Ziegler, Zenke,.…, Gerstner, 2015, "From synapses to behavioural modelling"; Zenke et al., 2017, "Continual Learning Through Synaptic Intelligence"] has shown a benefit for the 2stage synapses model in terms of learning and forgetting. Is there a similar benefit for this 2stage model? This may simply be a Discussion point, referring to the analysis asked in the point (1) above.
The main benefit of having 2stage synapses is to save energy that would otherwise be expended if the network has to consolidate every single weight change. As described above, a decaying 2state synapse model can under some circumstances reduce perceptron training time. However, we are not sure whether this generalizes to multilayer perceptrons.
We now cite a recent study by Leimer, Herzog and Senn that shows how a similar 2state model can prevent catastrophic forgetting (we thought this was the most relevant paper as the Zenke study does not employ a 2state model, while Ziegler et al. do not show a functional advantage).
3) From a biological point of view, it is clear that the change as well as the maintenance of synaptic weight can cost energy. Nevertheless, we find it strange that the authors analyze the perceptron learning rule according to a changeonly energy cost function while the synaptic caching rule is analyzed by a combination of change (latephase) and maintenance (earlyphase) cost function. How critical is the phasecost relation? Is it also efficient if the latephase costs maintenance (e.g., by having a larger synaptic apparatus) and the earlyphase costs energy dependent on its change? What is the energy consumption of the perceptron learning rule considering the maintenance cost function?
We now discuss these two variants in detail:
If the transient component (earlyphase) costs energy dependent on its change, the theory can be straightforwardly extended (see Materials and methods, subsection “Efficiency of synaptic caching”). As expected this extra cost can only diminish the benefit of synaptic caching.
If, on the other hand, there is a latephase maintenance cost during learning, postponing consolidation will reduce costs further (see Results subsection, “Synaptic caching”).
These arguments only consider the cost of learning and not the cost of computing once the network has learned. We are currently studying this much more complicated situation. We have shown earlier that learning rules can be designed to reduce costs associated to computation (Sacramento et al., 2015), but the precise interaction of computation cost and plasticity cost is not clear at the moment.
4) Please test for comparison the energy consumption of other learning rules performing perceptron learning (e.g., D’Souza et al., 2010).
We have now implemented the learning rule from D’Souza et al., 2010, and find that indeed large energy savings can still be achieved (see new Figure 2—figure supplement 1), showing that the synaptic caching principle is general and does not depend on implementation details.
https://doi.org/10.7554/eLife.50804.sa2Article and author information
Author details
Funding
Leverhulme Trust (RPG2017404)
 Mark CW van Rossum
Engineering and Physical Sciences Research Council (EP/R030952/1)
 Mark CW van Rossum
The funders had no role in study design, data collection and interpretation, or the decision to submit the work for publication.
Acknowledgements
This project is supported by the Leverhulme Trust with grant number RPG2017–404. MvR is supported by Engineering and Physical Sciences Research Council (EPSRC) grant EP/R030952/1. We would like to thank Joao Sacramento and Simon Laughlin for discussion and inputs.
Senior Editor
 Ronald L Calabrese, Emory University, United States
Reviewing Editor
 Peter Latham, University College London, United Kingdom
Reviewer
 Walter Senn, University of Bern, Switzerland
Publication history
 Received: August 3, 2019
 Accepted: February 10, 2020
 Accepted Manuscript published: February 13, 2020 (version 1)
 Version of Record published: March 19, 2020 (version 2)
Copyright
© 2020, Li and van Rossum
This article is distributed under the terms of the Creative Commons Attribution License, which permits unrestricted use and redistribution provided that the original author and source are credited.
Metrics

 998
 Page views

 198
 Downloads

 0
 Citations
Article citation count generated by polling the highest count across the following sources: Crossref, PubMed Central, Scopus.
Download links
Downloads (link to download the article as PDF)
Download citations (links to download the citations from this article in formats compatible with various reference manager tools)
Open citations (links to open the citations from this article in various online reference manager services)
Further reading

 Cell Biology
 Neuroscience

 Neuroscience