B IOLOGICALLY INSPIRED SLEEP ALGORITHM FOR INCREASED GENERALIZATION AND ADVERSARIAL ROBUSTNESS IN DEEP NEURAL NETWORKS
Anonymous authors
Paper under double-blind review
A BSTRACT
Current artiﬁcial neural networks (ANNs) can perform and excel at a variety of tasks ranging from image classiﬁcation to spam detection through training on large datasets of labeled data.
While the trained network usually performs well on similar testing data, certain inputs that differ even slightly from the training data may trigger unpredictable behavior.
Due to this limitation, it is possible to generate inputs with very small designed perturbations that can result in misclassiﬁcation.
These adversarial attacks present a security risk to deployed ANNs and indicate a divergence between how ANNs and humans perform classiﬁcation.
Humans are robust at behaving in the presence of noise and are capable of correctly classifying objects that are occluded, blurred, or otherwise distorted.
It has been hypothesized that sleep promotes generalization and improves robustness against noise in animals and humans.
In this work, we utilize a biologically inspired sleep phase in ANNs and demonstrate the beneﬁt of sleep on defending against adversarial attacks as well as increasing ANN classiﬁcation robustness.
We compare the sleep algorithm’s performance on various robustness tasks with two previously proposed adversarial defenses, defensive distillation and ﬁne-tuning.
We report an increase in robustness after sleep to adversarial attacks as well as to general image distortions for three datasets: MNIST, CUB200, and a toy dataset.
Overall, these results demonstrate the potential for biologically inspired solutions to solve existing problems in ANNs and guide the development of more robust, human-like
ANNs.
I NTRODUCTION
Although artiﬁcial neural networks (ANNs) have recently begun to rival human performance on various tasks, ranging from complex games (Silver et al.
to image classiﬁcation (Krizhevsky et al.
(2012)), they have been shown to underperform when the testing data differs in speciﬁc ways even by a small amount from the training data (Geirhos et al.
This lack of generalization presents two issues when ANNs are utilized in the real world.
First, ANNs are often trained on curated datasets of images designed to best capture the image content, whereas in real-world scenarios, they may be tested on disturbed or noisy inputs, not observed during training.
Second, ANNs are susceptible to adversarial attacks, or the deliberate creation of inputs designed to fool ANNs that may be imperceptibly different from correctly classiﬁed inputs (Szegedy et al.
These two issues limit the applicability of ANNs in the real world and present potential security risks when deployed.
There have been two main approaches for investigating ANN robustness: adversarial machine learning and training data manipulation (Ford et al.
Adversarial machine learning aims to develop novel attack methods which perturb the input minimally while changing the ANN’s classiﬁcation (Moosavi-Dezfooli et al.
; Carlini & Wagner (2017); Goodfellow et al.
; Athalye et al.
; Nguyen et al.
as well as to design defense mechanisms which prevent these attacks from affecting ANN behavior (Papernot et al.
(2016b)
; Goodfellow et al.
; Huang et al.
(2015), see Yuan et al.
(2019) for review).
Training data manipulation research typically examines the impact of changing the input distribution during testing and observing the effect on ANN performance.
Geirhos et al.
(2018) showed that ANNs trained on images with one type of distortion do not perform as well when tested on other types of distortions, even if images with both distortions appear identical to the human eye.
Likewise, ANNs trained on unperturbed images exhibit reduced performance when images in the test set are distorted, for example, through horizontal translations, blurring, or the addition of compression artifacts (Dodge & Karam (2016); Vasiljevic et al.
Zhou et al.
Although it has been proposed that adversarial and manipulation robustness can be increased through various mechanisms during the training phase, such as ﬁne-tuning, recent research has shown that these methods are mostly ineffective or their effectiveness is inconclusive (Geirhos et al.
; Uesato et al.
; Athalye et al.
It has been hypothesized that in the mammalian brain sleep helps to create generalized representations of the input learned during the awake state (Stickgold & Walker (2013)).
Sleep has been identiﬁed as being critical for memory consolidation - a process of converting recent memories into long-tern storage (Rasch & Born (2013)).
During sleep, there is reactivation of neurons involved in previously learned activity (Stickgold (2005)) and this reactivation is likely to invoke the same spatio-temporal pattern as the pattern observed during training in the awake state (Wilson & McNaughton, 1994).
Sleep reactivation, or replay, serves to strengthen synapses involved in a learned task through spike-timing dependent plasticity rules (STDP).
Sleep, through STDP, can increase a subject’s ability to form logical connections between memories and to generalize knowledge learned during the awake state (Payne et al.
In one study, subjects learned to ﬁnd an exit to a maze in a virtual 3D environment starting at various positions.
Subjects who were allowed to sleep exhibited a more complex understanding of the overall shape of the maze (Wamsley et al.
Using a biophysical model of cortical network, Gonzalez et al.
and Wei et al.
(2018) showed that sleep promotes reactivation and helps to create distinct representations for unique memories by devoting synapses to speciﬁc memory traces.
This body of neuroscience work suggests that a sleep-like phase may be applied to ANNs to enable the networks to extract the gist of the training data and to rely less on the input statistics, which may be peculiar to the speciﬁcs of the training data and not the underlying distribution.
Our hypothesis is that sleep could aid in reducing a neural network’s susceptibility to adversarial attacks and to increase generalization performance by reducing the impact that imperceptible pixel changes can have on the task output.
In this new work, we propose a sleep-inspired algorithm to defend against adversarial attacks as well as to increase ANN robustness to noise.
We utilize the notion of sleep from biology and apply an off-line unsupervised ”sleep” phase to modify the parameters of a fully connected ANN.
We demonstrate a number of performance improvements over existing defense algorithms, such as ﬁnetuning or adversarial retraining and defensive distillation, on both adversarial and noise robustness.
The contributions are summarized below:
• We analyze how robust our sleep algorithm is to four different types of adversarial attacks on three different datasets (MNIST, CUB200, and a toy dataset). For most conditions (MNIST, toy dataset), after a sleep phase was applied, the attacks consistently resulted in adversarial examples that were more distinct from the original input compared to the adversarial examples designed for the original (before sleep) network.
• We illustrate that the sleep algorithm creates a more robust network whereby performance on noisy and blurred inputs is higher compared to a control or defensively distilled network and is more robust to the other distortions compared to ANNs that are ﬁne-tuned on a single distortion.
• We analyze the impact of the sleep algorithm on task representation and demonstrate that the algorithm creates decision boundaries that more closely resemble the true classes, effectively extracting the gist of the data.
A DVERSARIAL A TTACKS AND D ISTORTIONS
Adversarial attacks aim to create minimal perturbations that, while imperceptible to the human eye, fool ANNs.
These attacks range from white-box to black-box attacks, based on how much information they assume the attacker to possess about the network.
White-box attacks assume that the attacker has access to the network architecture, training data and weights.
These attacks can range from absolute information, such as gradient-based attacks which compute the gradient of the loss with respect to the input (Brendel et al.
(2017)), to score-based attacks which only utilize predicted scores of the model.
Black-box attacks, which assume no knowledge about the network, solely rely on the decision made in order to craft adversarial examples.
Attacks can be targeted such that the at- tacker aims to create an adversarial example that the network predicts as a certain class or untargeted where the attacker’s goal is simply to cause any kind of misclassiﬁcation (Biggio & Roli (2018)).
In this work we consider four adversarial attacks ranging from white-box to black-box attacks.
We assume that the attacker solely wants to cause a misclassiﬁcation, with no respect to the output class.
We present a brief description of each of the four attacks below (see Appendix for examples of images created by these attacks).
Fast Gradient Sign Method (FGSM).
FGSM (Goodfellow et al.
(2014)) computes the sign of the gradient of the loss function ( J ) with respect to the original input x using the weights θ of the network and the target labels y .
[Equation]
This represents the direction to change each pixel in the original input in order to increase the loss function.
Based on the value of (cid:15) , the corresponding perturbation to the original image can range from small to large.
Thus, in this work we use the average of the smallest value of (cid:15) needed to create an adversarial example x (cid:48)
(misclassiﬁed input) for each input in the testing set.
DeepFool.
DeepFool (Moosavi-Dezfooli et al.
(2016)) is an iterative method which approximates the nearest decision boundary to the input at time t and moves the input x t in that direction to compute x t +1 .
This process is repeated until a misclassiﬁcation is produced or the runtime of the simulation is exceeded.
For this attack, we measure the L2-norm between the original input x and the adversarial input x (cid:48) .
Thus, successful defenses should result in a high L2-norm for this algorithm.
Jacobian-based Saliency Map (JSMA).
JSMA (Papernot et al.
(2016a)) aims to craft adversarial examples that minimize the L0-norm of x − x (cid:48) by reducing the number of pixels that are altered.
In summary, the algorithm computes the gradient, as done in FGSM but for all possible classes.
These gradient values represent how changing each pixel contributes to the overall loss function, with large values indicating a signiﬁcant effect on the loss.
These values are used to create a saliency map, where each pixel’s impact on the loss is modelled.
The algorithm utilizes this saliency map to alter individual pixels, repeating the gradient and saliency map computation until an adversarial example is created.
For this type of attack, we utilize the L2-norm to determine defense success.
Boundary Attack.
The Boundary Attack (Brendel et al.
(2017)) is a black-box attack which relies solely on the decision of the ANN to craft an adversarial example.
Given an input x , a random input x (cid:48) 0 is chosen such that f ( x ) (cid:54) =
f ( x (cid:48) 0 ) , where f ( x ) is the label produced by the ANN.
In our work, x (cid:48) 0 is chosen from a uniform distribution.
The attack starts by moving x (cid:48) 0 toward x until it reaches the point where f ( x ) =
f ( x (cid:48) 0 ) , or the decision boundary in between f ( x ) and f ( x (cid:48) 0 ) .
From here, the attack consists of two steps: an orthogonal perturbation and a forward perturbation.
During the orthogonal perturbation, random points along the hypersphere around f ( x (cid:48) t ) are sampled.
Those that are adversarial and closer to x than before are added to the queue for forward perturbation.
During the forward perturbation, a small step is taken from x (cid:48) t to x as long as f ( x ) (cid:54) =
f ( x (cid:48) t ) .
This process is repeated until a convergence criterion is met.
For this attack, we utilize the L2-norm to deﬁne defense success.
Distortions.
Although not speciﬁcally designed to attack an ANN, distortions similarly corrupt
ANN performance.
In this work we consider two simple distortion techniques: blurring and Gaussian noise.
For images, we perform 2-D Gaussian ﬁltering with a blur kernel of varying standard deviation in order to blur the images.
For all inputs, we add Gaussian noise with mean 0 and standard deviation σ .
These distortions are tested with the proposed sleep algorithm as well as using the adversarial defenses discussed below.
A DVERSARIAL D EFENSES
We compare our sleep algorithm with two existing adversarial defenses: defensive distillation and ﬁne-tuning, or adversarial retraining.
Defensive distillation (Papernot et al.
(2016b)) utilizes two training sessions in order to create a distilled network.
First, an initial network is trained on ( X, Y ) , where X is the training data, and Y is the one-hot encoded training labels.
The activation function of this network is changed such that the softmax function of the output layer is computed using a temperature term T as follows:
[Equation]
A higher value for T forces the ANN to produce probability values with large values for each class, whereas lower values for T support a similar representation as the one-hot encoded labels.
After the ﬁrst network is trained, the output of the network (probability values) is used to train a distilled network with the same softmax-temperature function.
Previous work has shown this to be successful at preventing some types of attacks (Papernot et al.
(2016b)).
However, others have shown that it is not successful at defending against modiﬁed versions of those attacks or novel attacks in general (Carlini & Wagner (2016; 2017)).
Based on the previous work which found that temperature values between 20 and 100 effectively prevent adversarial attacks (Papernot et al.
(2016b))
, we use a temperature value of T = 50 in our implementation of defensive distillation.
Adversarial retraining aims to ﬁne-tune the network on adversarial examples with the correct labels as a form of regularization.
Previous work has shown that adversarial retraining can mitigate the effectiveness of some adversarial attacks.
Goodfellow et al.
(2014) showed that adversarial retraining can reduce the error rate on MNIST, demonstrating greater ANN robustness after ﬁne-tuning.
Likewise, Moosavi-Dezfooli et al.
(2016) showed that ﬁne-tuning on DeepFool attacks can reduce the effectiveness of their attacks.
However, they observed that ﬁne-tuning on FGSM attacks has negative results, actually increasing the strength of the attack.
This suggests that ﬁne-tuning may overﬁt the network to certain attacks, while failing to extrapolate to other attacks, similar to results shown for generalization in ANNs (Geirhos et al.
For the adversarial retraining procedure presented here, we train the network on the original input and then ﬁne-tune the network on various adversarial attacks with a reduced learning rate. [Equation]
LEEP ALGORITHM
The basic intuition behind the sleep algorithm is that a period of ofﬂine activity, whereby network weights are modiﬁed according to an unsupervised learning algorithm, allows the parameters of the network to become more reﬂective of the underlying statistics of the task at hand, while not overﬁtting the statistics of the training data.
The pseudocode is presented in Algorithm 1.
In short, an ANN is trained using stochastic gradient descent and the usual backpropagation algorithm (exact parameters used for each of the datasets are shown in Table 2).
After training, the network structure is converted into a spiking neural network (SNN).
After building the SNN, we run a sleep phase which modiﬁes the network connectivity based on spike-timing dependent plasticity (STDP).
After the sleep phase, the SNN network is converted back into the ANN and testing is performed.
4. S PIKING N EURAL N ETWORKS
While most ANNs are modelled after highly simpliﬁed brain dynamics, SNNs seek to more closely model temporal brain dynamics.
In short, SNNs are composed of spiking neurons and model the information transformation and the dependence on exact timing of spikes that occurs in biological networks (Ghosh-Dastidar & Adeli (2009)).
Neurons can range from simple integrate-and-ﬁre type neurons which sum their inputs and produce an output (spike) if this exceeds some ﬁring threshold to more complicated Hodgkin-Huxley type neurons which model sodium-, potassium-, and chloridechannel kinetics (Abbott & Kepler (1990)).
Recent work has shown that a near loss-less conversion between ANNs and SNNs can be achieved by propagating activity through a spiking neural network for a given input and counting the number of times that each output neuron ﬁres (Diehl et al.
To convert an ANN to SNN (Lines 1-3 of pseudocode), we assume the ANN utilizes ReLU neurons with no bias.
This assumption is made so that the output neuron’s activation can be treated as a ﬁring rate, either zero or positive, and that the thresholds of all neurons in a given layer are of the same scale.
The weights from the ANN are directly mapped to the SNN.
Each unit in the SNN is modelled as an integrate-and-ﬁre type neuron, computing the following equation:
[Equation]
Map the weights from ( nn ) with ReLU units to network of integrate-ﬁre units ( snn ) 3:
Apply weight normalization and return scale for each layer ([24]) return snn, scales 4: procedure C ONVERT SNN TO ANN( nn ) 5:
Directly map the weights from integrate-ﬁre network ( nn ) to ReLU network ( ann ) return ann 6: procedure S LEEP ( nn, I, scales ) (cid:46)
I is input 7:
Initialize v (voltage) = 0 vectors for all neurons 8: for t ← 1 to Ts do (cid:46)
Ts - duration of sleep 9:
S (1 , t ) ← Convert input I to Poisson-distributed spiking activity 10: for l ← 2 to n do (cid:46) n - number of layers 11: v ( l, t ) ← v ( l, t − 1) + ( scales ( l − 1) W ( l, l − 1) S ( l − 1 , t ))
(cid:46) W(l,l-1) - weights 12:
S ( l, t ) ← v ( l, t ) > threshold ( l ) (cid:46)
Propagate spikes 13:
W ( l, l − 1) ← (cid:26) W ( l, l − 1) + inc if S ( l, t ) = 1 & S ( l − 1 , t ) =
W ( l, l − 1) − dec if S ( l, t ) = 1 & S ( l − 1 , t ) =(cid:46)
STDP 14: procedure M AIN 15:
Initialize neural network ( ann ) with ReLU neurons and bias = 0.
Train ann using backpropagation.
17: snn, scales = ConvertANNtoSNN ann ) 18: snn = Sleep( snn , Training data X , scales ) 19: ann = ConvertSNNtoANN snn ) Here, τ m represents the decay constant of the membrane potential, v is the voltage at a given time, w i is the weight connecting from neuron i , and s ( i ) is the spiking activity of neuron i , either 1 or 0. 4. S LEEP The key advantage of using a SNN is that biologically inspired training rules can be applied while the network is driven by noisy input.
Empirical data suggest that the brain uses spike-timing dependent plasticity (STDP) (Song et al., 2000), where weight updates depend on the relative timing of pre- and post-synaptic spikes.
It has been shown that STDP results in balanced activity, where all neurons ﬁre in equal proportions (Song et al.
Here we utilize a simpliﬁed version of STDP, resembling the classic Hebbian plasticity rule: if a pre-synaptic spike induces a post-synaptic spike, then the weight between these neurons is increased.
If a post-synaptic spike occurs, but the pre-synaptic neuron does not spike, then the corresponding weight is decreased (in this case postsynaptic spiking may occur because of spiking in other neurons connecting to that post-synaptic neuron).
The sleep training phase we propose here can be described as following.
First, inputs to each neuron of the input layer must be presented as spiking activity in order to propagate activity from the input layer to the hidden layers of the network.
We convert inputs (real-valued pixel intensities or features) to spikes by deﬁning a maximum ﬁring rate f max with units spikessec and computing a Poisson-distributed spike raster, such that inputs with higher values (i.e. brighter pixels) will have higher rate than inputs with lower values, with no spike rates exceeding f max .
Next, activity is propagated through the network as spikes and the STDP rule is applied to update weights.
To simulate sleep, with periods of up-states where activity is elevated and memory traces are thought to be replayed (resembling biological activity during deep sleep), weights are up-scaled to induce high ﬁring rates in later layers.
Other important parameters include the threshold for each layer and the length of sleep.
The parameters used for each dataset are presented in Table 3.
4. E XPERIMENTS AND D ATASETS
Here, we describe the general experimental setup as well as the datasets tested.
First, we trained a control ANN using the training set for each of the main datasets used in this study.
Next, we created a defensively distilled network using T = 50 for the temperature parameter to create the second test network.
Then, we ﬁne-tuned the control ANN on a speciﬁc attack or distortion method to create the third test network.
Finally, we converted the control ANN to an SNN and applied the sleep algorithm as described above to create the fourth test network.
We created adversarial examples for each of these four networks using the attacks we described above (ﬁne-tuned networks are tested on the attacks they were ﬁne-tuned on).
Then, we analyze how successful each attack is to fool each of the four networks using the metrics deﬁned above.
For generalization (blur and noise), we performed the same setup as above creating four different networks.
We then tested each network on varying levels of distortion.
We tested networks ﬁne-tuned on blurred and noisy images to measure how performance generalizes across distortion methods.
We averaged performance across a minimum of three networks for each attack and distortion.
We used three datasets to compare performance:
Patches (a toy dataset created simply for analysis),
MNIST (LeCun et al.
(1998)), and CUB-200 (Welinder et al.
Patches consists of four binary images arranged in a 10x10 square.
Each image has its own label (1-4), and consists ofbright pixels (value set to 1) and 75 dark pixels.
The overlap of bright pixels among the four images (see Appendix] Appendix) is chosen such that the task is not trivial.
The MNIST dataset consists of 70,000 28x28 greyscale images of handwritten digits, with 60,000 in the training set and 10, in the testing set.
CUB-200 is a high resolution dataset of images of birds with 200 bird species, with very few ( 30) images per class.
For this dataset, we used previously extracted ResNetembeddings, where ResNet-50 was pre-trained on ImageNet (He et al.
For CUB-200, we do not report results for blurring, since we are using extracted features, not images.
R ESULTS
We evaluate the sleep algorithm in two settings: (1) Adversarial attacks designed to fool neural networks and (2) generalization distortions designed to reﬂect imperfect viewing conditions or other types of noise.
For adversarial attacks (other than FGSM), we utilize the following metric to evaluate the success of each defense.
Let x (cid:48)
i be the adversarial example created for input x i .
The total score
S A for an attack is the median squared L2-distance for all samples, where N is the dimension of the space:
[Equation]
For FGSM, we deﬁne the following metric which computes the median minimum noise level (cid:15) needed to produce a misclassiﬁcation across all samples:
[Equation]
For MNIST and CUB-200, we evaluate the attacks on all examples in the testing set.
Examples that the networks get wrong before the attack was implemented are discarded from the analysis (in these cases (cid:107) x (cid:48)
i − x (cid:107) 2 k = 0 and (cid:15) i = 0 for all attacks).
For FGSM and distortions, we also include plots of classiﬁcation accuracy as a function of noise level.
For DeepFool and JSMA, we report adversarial attack accuracy (number of examples where f ( x ) =
y and f ( x (cid:48) )
(cid:54) =
f ( x ) , where y is the correct label, over number of examples tested).
Note that these algorithms would always produce an adversarial example if allowed to run forever.
However, due to computational limitations, we included a run-time limitation on the number of iterations for these algorithms (see Appendix).
Thus, a lower adversarial attack accuracy indicates that the attack would need more iterations to run to reach 100% accuracy.
This is a similar measure as distance since more iterations would result in more distinct adversaries for all attacks implemented and the updates at each iteration have the same magnitude for each defense.
5. A DVERSARIAL A TTACKS
Here we report the scores for all different attacks and for the all datasets.
For the FGSM attack, the sleep algorithm increases the median minimum noise needed for misclassiﬁcation for all three datasets compared to the control network (also see Figure 1).
For MNIST dataset, the amount of noise needed to fool the network after the sleep algorithm was almost double of that needed for either the ﬁne-tuning or defensive distillation approaches.
For the Patches dataset, both defensive distillation and ﬁne-tuning increase the robustness of the network.
However, on CUB-200, only ﬁne-tuning and sleep were able to defend, albeit marginally, against the FGSM attack.
Looking at the classiﬁcation accuracy of the network as a function of noise added ( (cid:15) , Figure 1), we observe that [Table] A B
C
A cc u r
[Figure] [Figure] [Figure] ControlDef.
Dist.
FinetuningSleep in the Patches and CUB-200 dataset, sleep tends to have higher classiﬁcation accuracy than the other methods for epsilon < 0.1.
After this point, sleep tends to have equal classiﬁcation accuracies as compared to the other methods.
For MNIST, the baseline classiﬁcation accuracy on the original test set decreases slightly compared to the other methods (80% after sleep).
However, the performance remains high longer than for the other defense methods on images that were correctly classiﬁed.
We observe that performance continued to drop after a sufﬁciently large amount of noise was added.
This is biologically plausible as adding more noise to an image should result in misclassiﬁcations, as the images degrade.
In sum, these results indicate that a sleep phase can successfully mitigate FGSM, more so than a control network.
For DeepFool, sleep has a signiﬁcant effect on the defense score on the MNIST dataset, both reducing the attack success rate and increasing the distance between the adversarial example and the original input by an order of magnitude.
For Patches and CUB-200 this effect is less pronounced, with ﬁne-tuning or the control network performing better.
We hypothesize that sleep was ineffective in preventing the DeepFool attack in tasks with very few exemplars per class (Patches) or a large number of classes (CUB-200).
In CUB-200, there is a large number of classes so the distance between the input and the nearest decision boundary is smaller (this is supported by the fact that JSMA, an L0 attack, does worse than DeepFool for CUB-200 and vice versa for MNIST, control networks).
In this case, sleep is unable to move the decision boundary of one class without impinging on the decision space of another class.
In MNIST, where the decision space for one class is presumably larger, sleep can alter decision boundaries in a way that has a minimal effect on other classes.
Sleep successfully increases the network’s robustness to the JSMA atttacks MNIST and Patche, reducing the attack success rate in the case of MNIST and increasing the distance needed to create an adversary for Patches. On CUB-200, there is a marginal reduction in the adversarial attack accuracy compared to the control network.
Defensive distillation and ﬁne-tuning also reduce JSMA’s effectiveness.
However, for these two defenses, in the case of MNIST, the networks were capable of ﬁnding an adversary for a higher percentage of the testing set.
Thus, the effect of changing a small number of important pixels is mitigated after running the sleep algorithm.
For the Boundary Attack, we found that no defense mechanism helps compared to the control in decreasing the attack’s effectiveness on the MNIST dataset.
However, for CUB-200 and Patches, the sleep algorithm results in a higher defense score than that for the control network.
This lends support to the idea that sleep favorably alters decision boundaries so that it becomes harder to ﬁnd an adversarial example that is close to the original image after the sleep phase.
This also suggests that sleep is not simply obfuscating gradients, which has been a common criticism of several adversarial defenses (Athalye et al.
(2018)), which are tested on white-box attacks.
In fact, given the long run-time for convergence of this algorithm, if we deﬁne a threshold for adversarial attack success (L2-norm > 1), then sleep successfully defends against this attack on the MNIST dataset (see Table 4).
Why does sleep phase help?
It has been shown that sleep tends to promote an increase in stronger weights while pruning weaker weights, thus increasing the width of the weights’ distribution (Gonzalez et al., 2019).
This results in the consolidation of strong memories at the cost of diminishing weak memories.
From this point of view, a memory is a subspace or abstraction in the decision space corresponding to a given class.
Sleep may result in enlarging the subspace the network allocates to a stronger category while shrinking weaker ones (Figure 5A).
The process of strengthening the strong memory also results in making it robust and noise invariant, as seen in Figure 5B where the ﬁrstcategories (numbers 0-7) are strengthened and become more invariant to the FGSM attack, while the last two digits are essentially forgotten and the network cannot conﬁdently predict exemplars from these classes (Figure 5C).
If the noise is less targeted, as in the case of random noise or blurring, sleep does not need to alter the decision space as much to produce better generalization and can maintain a high baseline accuracy, as we demonstrate in the next section.
5. G ENERALIZATION
Figure 2 shows the network performance for noisy and blurry distortions of data for MNIST (A) as well as noisy distortions for the CUB-200 feature embeddings (B, see Figure 3 for results on
Patches).
Overall, ﬁne-tuning on an image distortion results in the best performance for that speciﬁc distortion.
However, as was noted (Geirhos et al.
(2018)), ﬁne-tuning on a speciﬁc distortion does not extend to other types of distortions.
In our analysis, ﬁne-tuning the network on blurred MNIST images results in high performance ( > 80% ) on blurred images.
However, for noisy images, this performance was only marginally above the control network.
The sleep algorithm increased performance for both distortion methods, since this approach is not tailored to any one representation of the training set.
Finally we tested how sleep increases robustness on blur and noise distortions.
In biological systems, sleep increases generalization through replay of memories learned during awake which leads to changes in synaptic weights.
These changes entail both an increase in synaptic weights associated with a speciﬁc task and pruning of synapses involved in other tasks (Gonzalez et al., 2019; Tononi Control Def. Dist. Finetune-noise
Finetune-blur Sleep 0. A cc u r a c y Noise STD
Blur Sigma
MNIST A
B
Noise STD
CUBA cc u r a c y
ControlDef.
Dist.
FinetuningSleep
Cirelli, 2006).
Figures 9 and 10 show that correlations among like digits in the hidden layers of the network are greater after applying sleep than before for noisy and blurred images.
Likewise, pairs of different digits usually become decorrelated after sleep, suggesting synaptic pruning.
We also show that both normalized spiking activity and activations of digit-speciﬁc neurons are higher after sleep than before (Figures 11 and 12, see Appendix for details).
These results suggest that the sleep algorithm increases robustness through biologically plausible learning mechanisms involving replay of relevant activity during sleep phase.
C ONCLUSIONS AND F UTURE D IRECTIONS
In this work, we show that a biologically inspired sleep algorithm can increase an ANN’s robustness to both adversarial attacks and general image distortions.
The algorithm augments the normal (e.g., back-propagation based) training phase of an ANN with an unsupervised learning phase in the equivalent SNN modelled after how the biological brain learns using Hebbian type plasticity rules.
We hypothesize that the unsupervised sleep phase creates more natural feature representations which in turn lead to more natural decision boundaries, thus increasing the robustness of the network.
Although this robustness may come at a cost of overall accuracy, it has been shown that robustness may have multiple important beneﬁts, such as more salient feature representations as well as invariance to input modiﬁcations (Tsipras et al.
We also show that the trade-off between robustness and accuracy does not always occur, particularly for image distortions such as noise or blur.
Future work includes converting the sleep algorithm into a regularization technique to be applied in more standardized machine learning frameworks as well as understanding the theoretical basis for the beneﬁcial role of spike based plasticity rules in increasing network robustness.
A PPENDIX 7. T RAINING P ARAMETERS
Here, we deﬁne the neural network parameters used for each of the three datasets as well as the sleep, defensive distillation
[Table]
and ﬁne-tuning parameters used.
Table 2 shows the neural network training parameters used to train each of the control networks discussed in the paper.
All neural networks were trained with ReLU neurons.
Table 3 shows the parameters used during sleep on each of the three datasets.
Note that these parameters (for MNIST and CUB-200) were chosen by running a genetic algorithm to maximimize performance on the FGSM attack (performance was determined based on the training set so as not to overﬁt to the test set).
For the other three attacks, parameters that maximized FGSM performance were used.
Also, for noise and blur generalization, different parameters were chosen (not shown here).
[Table] [Table]
input neurons,
Sleep duration = length of sleep (number of images presented during sleep, Thresholds = neuronal ﬁring thresholds for each layer of neurons, Synaptic AMPA current = factor to scale the weights by during sleep, Increase and Decrease factor = amount weights are modiﬁed on a STDP event.
For the defensively distilled networks tested in the paper, we ﬁrst train an initial network using a temperature of 50.
Then, we use the training set to compute soft labels and ﬁnetune the initial network on these soft labels for the same number of epochs and with the same learning rate.
For the ﬁne-tuned networks, we take the control networks trained with the parameters shown in
Table 2.
The learning rate is reduced to 0.05 and the network is ﬁne-tuned on a mixture of either adversarial attacks, blur or noise and the original images/features.
For CUB-200, we perform ﬁnetuning for 10 epochs.
7. P ATCHES A NALYSIS
The Patches dataset represents an easily interpretable example where we can understand what happens to the weights after sleep.
Figure 3A shows an example of the dataset.
Here, we have 4 images each belonging to 4 different classes.
25 pixels are whitened in each image and the remainingpixels are dark.
There is a 15 pixel overlap, so that weights connecting from input to output layer must take this into account in order to separate the images.
Figure 3B illustrate the blur and noise distortions tested for this dataset and Figure 3C shows the results for the blur and noise distortions.
After the network is trained, we can analyze the weights connecting from each of the 100 input neurons to the 4 output neurons (see Figure 4, top row).
We theorize that optimally robust behav- [Figure] ior would occur when weights connecting from ON-pixels are positive, weights connecting from overlapping pixels are near 0, and weights connecting from OFF-pixels are negative.
In this case, changing the value of overlapping pixels will have no effect on classiﬁcation.
Changing the value of OFF-pixels will cause the network to predict another class, where OFF-pixels may be ON-pixels or indicative of that class.
Changing the value of ON-pixels will only have a negative impact if the brightness of the pixel is reduced signiﬁcantly.
Thus, in this circumstance, the network should behave robustly.
In the control network we observe that weights connecting from ON-pixels (pixel-value = 1) increase while weights connecting from OFF-pixels remain at 0.
Weights connecting from overlapping pixels remain near 0 or positive.
Defensive distillation causes some weights connecting from overlapping pixels to decrease, likely because the soft labels used in defensive distillation cause overlapping pixel units to alter the probability values computed by the network in such a way that does not truly reﬂect the impact of the overlapping pixels.
In the ﬁne-tuning networks (both on blurred images and noisy images), we observe an increase in ON-pixel weights and an increase in noisiness of OFF-pixel weights.
Likewise, in the sleep network, OFF-pixel weights become negative while ON-pixel weights remain the same.
In these cases, robustness is increased as weights become more similar to our hypothesized ideal weights.
Essentially, the magnitude of input changes need to change classiﬁcation increase since the spread between ON-pixel weights and OFF-pixel weights increases.
We quantify the spread in weights by taking the difference between the average weight connecting from ON-pixels and the average weight connecting from OFF-pixels.
This represents the mean input that each correct output neuron receives.
This result is shown in Figure 3D. Of note is that this weight spread is increased for both the sleep and ﬁnetuning-noise network, suggesting that these defenses bring the weights closer to their ideal values for computing robustness. [Figure] C o n
[Figure]
o l D e f . D i s t i . F i n e t un i n g
N o
i s e
F i n e t un i n g
B l u r
S l ee p
Neuron ID
Neuron ID
Neuron ID
Neuron ID
A
[Figure]
A DVERSARIAL A TTACKS
A
B
C
Digit [Figure] n I n p u t
Neuron
Neuron
I n p u t
Here, we describe the general approach for implementing DeepFool, JSMA, and
Boundary Attack discussed in the paper.
We also of adversaries for of the defense networks from these attacks. A Boundary Attack JSMA
DeepFool
C
B
Control
Def. Dist.
Finetuning
Sleep
Control
Def. Dist.
Finetuning
Sleep
Control
Def. Dist.
Finetuning
Sleep DeepFool. DeepFool (Moosavi-Dezfooli et al.
(2016)), as mentioned above, is an iterative algorithm that, at each iteration, aims to move the adversarial example in the direction of the closest decision boundary until it results in a misclassiﬁcation.
We based our implementation off of that in Rauber et al.
We stopped running the algorithm when either an adversarial example is found or when 100 iterations have passed.
Examples of DeepFool attacks on the MNIST dataset are shown in Figure 6A.
At each iteration we compute a linear approximation of the loss function and take a step in the direction that would be result in a misclassiﬁcation.
The equations used and pseudocode can be found in the original DeepFool paper.
JSMA.
JSMA is also an iterative algorithm which computes the pixel that would change the loss function the most at each algorithm and changes this pixel, until a misclassiﬁcation is produced.
For this method, we set a run-time limit of 500 iterations.
We also remove a pixel from the saliency map when it has beeen updated seven times, so the algorithm can focus on other pixels.
We set the change to each pixel at a constant value, 0.1.
This represents how much each pixel is updated (in the direction that results in a misclassiﬁcation) at each iteration.
Pseudocode can be found in the original publication (Papernot et al.
(2016a)).
We show examples of adversaries created by JSMA in Figure 6B.
Boundary Attack.
The Boundary Attack (Brendel et al.
(2017)) starts with an adversarial example and moves it closer to the decision boundary of the correct class.
At each step of the algorithm, the method performs orthogonal and forward perturbations to move the adversary closer to the original image, thus reducing the distance between the adversary and the original image.
We set both a distance convergence criterion (L2-norm = 1e-7) and a run-time limitation on the attack ( iterations).
Example attacks are shown in Figure 6C.
We note that sometimes the algorithm does not successfully produce an ”imperceptible” adversarial example and instead produces a noisy output (the starting condition is a noisy image).
If we deﬁne a threshold deﬁning a successful adversarial attack (L2-Norm > 1), then we observe the results for MNIST in Table 4. [Table]
7. G ENERALIZATION A NALYSIS
[Figure]
In this section, we analyze how sleep can aid in increasing ANN robustness.
In biological networks, sleep extracts the gist of a task through replay (Lewis & Durrant (2011)).
Thus, we hypothesized that our sleep algorithm works in the same manner.
First, we tested the ability of the sleep network to decorrelate distinct inputs.
We tested this hypothesis by analyzing the effect of running sleep and testing on our two distortion techniques (see Figure 7).
First, we computed the correlations of network activities in each of the hidden layers of the network before and after implementing our defense methods.
For each pair of digits, we computed the average correlation of layer activities in the undistorted (Figure 8), noisy (Figure 9), and blurred (Figure 10) conditions.
Each ﬁgure reports the difference in digit pairwise correlations between the defense method and the control network for each set of inputs.
For our sleep network, it is apparent that in layer 2 and layer 3, the correlations of the same digits (the diagonal) increases after sleep.
Additionally, the correlation of distinct digits typically experiences negative change, representing decorrelation of distinct inputs.
This analysis holds for defensive distillation and both of the ﬁne-tuned networks.
This suggests that the ANN representation of different exemplars of the same digit becomes more similar after sleep or after any of the defense networks when compared to the control.
This is not simply due to an increased overlap of all inputs, since exemplars of different digits become decorrelated after applying a defense method.
Next, we performed the same correlation analysis on noisy and blurred images to see how the representation of distorted images changes after applying a distortion method.
First, we note that ﬁnetuning on noisy images results in stronger correlation of the same (noisy) digit but weaker correlations of different (noisy) digits, as noted above.
However, ﬁne-tuning on blurred images does not have as strong an effect.
Second, sleep seems to have a beneﬁcial effect on the correlation matrices for both blurred and noisy images (comparing the right column of Figures 9 and 10).
This illustrates the beneﬁcial role of sleep in creating distinct representations of digits, where different neuronal ensembles encode different digits.
This change in representation should result in increased robustness since changes to the input must be larger in order to recruit neuronal ensembles that represent other digits.
On top of decorrelating the representation of distinct memories by pruning synapses, biophysical modelling suggests that sleep can also aid in strengthening connections thus making the response of primary neurons involved in memory much stronger (Gonzalez et al.
To test this hypothesis in our networks, we analyzed the ﬁring rate and activations of digit-speciﬁc neurons before and after sleep.
Before describing the analysis, we remind the reader that SNNs can be used to performn classiﬁcation and a near loss-less conversion between ANNs and SNNs has been achieved on the
MNIST task (Diehl et al. (2015)).
To perform classiﬁcation, a digit is presented (as a Poisson spike train) to the network and spikes are propagated throughout the network for a given time period (or number of presentations of the input).
Analyzing network activity in the spiking domain can be easier than in the activation domain (ANNs) since spikes are oftentimes easier to interpret than neuronal activations.
For this reason, we ﬁrst analyze how spike rates of digit-speciﬁc neurons change before and after sleep in the spike domain.
To do this we present all images of a speciﬁed digit to the spiking network and count the number of spikes from each neuron (holding the weights constant).
We deﬁne digitspeciﬁcity by looking at the 100 neurons with the highest ﬁring rates in layer 2.
In Figure 11, we show that the normalized ﬁring rate of these neurons usually increases after sleep (normalized by dividing by the maximum ﬁring rate observed from the SNN).
Next, we perform the same analysis in the activation domain.
Again, we deﬁne digit-speciﬁc neurons by looking at the top 100 neurons with the highest activation for a speciﬁc digit.
We look at the normalized mean activations of these neurons before and after sleep and note that for all digits this value is higher after sleep than before sleep (Figure 12).
This suggests that the neurons in the network are responding more strongly to the presentation of the same digit, thus increasing the robustness of the network as more noise must be added in order to counter the effect of this stronger response.
This also suggests that our algorithm works in a biologically plausible way: both by decorrelating distinct inputs and increasing the strength of similar inputs.
La y e r La y e r
La y e r
Sleep Defensive
Distillation
Finetune
Noise
Finetune
Blur
La y e r
La y e r
La y e r
Sleep Defensive
Distillation
Finetune
Noise
Finetune
Blur
La y e r La y e r
La y e r
Sleep Defensive
Distillation
Finetune
Noise
Finetune
Blur