Abstract
Through the success of deep learning in various domains, artificial neural networks are currently among the most used artificial intelligence methods. Taking inspiration from the network properties of biological neural networks (e.g. sparsity, scalefreeness), we argue that (contrary to general practice) artificial neural networks, too, should not have fullyconnected layers. Here we propose sparse evolutionary training of artificial neural networks, an algorithm which evolves an initial sparse topology (Erdős–Rényi random graph) of two consecutive layers of neurons into a scalefree topology, during learning. Our method replaces artificial neural networks fullyconnected layers with sparse ones before training, reducing quadratically the number of parameters, with no decrease in accuracy. We demonstrate our claims on restricted Boltzmann machines, multilayer perceptrons, and convolutional neural networks for unsupervised and supervised learning on 15 datasets. Our approach has the potential to enable artificial neural networks to scale up beyond what is currently possible.
Introduction
Artificial neural networks (ANNs) are among the most successful artificial intelligence methods nowadays. ANNs have led to major breakthroughs in various domains, such as particle physics^{1}, deep reinforcement learning^{2}, speech recognition, computer vision, and so on^{3}. Typically, ANNs have layers of fullyconnected neurons^{3}, which contain most of the network parameters (i.e. the weighted connections), leading to a quadratic number of connections with respect to their number of neurons. In turn, the network size is severely limited, due to computational limitations.
By contrast to ANNs, biological neural networks have been demonstrated to have a sparse (rather than dense) topology^{4,5}, and also hold other important properties that are instrumental to learning efficiency. These have been extensively studied in ref. ^{6} and include scalefreeness^{7} (detailed in Methods section) and smallworldness^{8}. Nevertheless, ANNs have not evolved to mimic these topological features^{9,10}, which is why in practice they lead to extremely large models. Previous studies have demonstrated that, following the training phase, ANN models end up with weights histograms that peak around zero^{11,12,13}. Moreover, in our previous work^{14}, we observed a similar fact. Yet, in the machine learning stateoftheart, sparse topological connectivity is pursued only as an aftermath of the training phase^{13}, which bears benefits only during the inference phase.
In a recent paper, we introduced compleX Boltzmann machines (XBMs), a sparse variant of restricted Boltzmann machines (RBMs), conceived with a sparse scalefree topology^{10}. XBMs outperform their fullyconnected RBM counterparts and are much faster, both in the training and the inference phases. Yet, being based on a fixed sparsity pattern, XBMs may fail to properly model the data distribution. To overcome this limitation, in this paper, we introduce a sparse evolutionary training (SET) procedure, which takes into consideration data distributions and creates sparse bipartite layers suitable to replace the fullyconnected bipartite layers in any type of ANNs.
SET is broadly inspired by the natural simplicity of the evolutionary approaches, which were explored successfully in our previous work on evolutionary function approximation^{15}. The same evolutionary approaches have been explored for network connectivity in ref. ^{16}, and for the layers architecture of deep neural networks^{17}. Usually, in the biological brain, the evolution processes are split in four levels: phylogenic at generations time scale, ontogenetic at a daily (or yearly) time scale, epigenetic at a seconds to days scale, and inferential at a milliseconds to seconds scale^{18}. A classical example which addresses all these levels is NeuroEvolution of Augmenting Topologies (NEAT)^{19}. In short, NEAT is an evolutionary algorithm which seeks to optimize both the parameters (weights) and the topology of an ANN for a given task. It starts with small ANNs with few nodes and links, and gradually considers adding new nodes and links to generate more complex structures to the extent that they improve performance. While NEAT has shown some impressive empirical results^{20}, in practice, NEAT and, most of its direct variants have difficulty scaling due to their very large search space. To the best of our knowledge, they are only capable of solving problems, which are much smaller than the ones currently solved by the stateoftheart deep learning techniques, e.g. object recognition from raw pixel data of large images. In ref. ^{21}, Miconi has tried to use NEAT like principles (e.g. addition, deletion) in combination with stochastic gradient descent (SGD) to train recurrent neural networks for small problems, due to a still large search space. Very recently in refs. ^{22,23}, it has been shown that evolution strategies and genetic algorithms, respectively, can train successfully ANNs with up to four million parameters as a viable alternative to DQN^{2} for reinforcement learning tasks, but they need over 700 CPUs to do so. To avoid being trapped in the same type of scalability issues, in SET, we focus on using the best from both worlds (i.e. traditional neuroevolution and deep learning). E.g., evolution just at the epigenetic scale for connections to yield a sparse adaptive connectivity, structured multilayer architecture with fixed amounts of layers and neurons to obtain ANN models easily trained by standard training algorithms, e.g. SGD, and so on.
Here, we claim that topological sparsity must be pursued starting with the ANN design phase, which leads to a substantial reduction in connections and, in turn, to memory and computational efficiency. We show how ANNs perform perfectly well with sparsely connected layers. We found that sparsely connected layers, trained with SET, can replace any fullyconnected layers in ANNs, at no decrease in accuracy, while having quadratically fewer parameters even in the ANN design phase (before training). This leads to reduced memory requirements and may lead to quadratically faster computational times in both phases (i.e. training and inference). We demonstrate our claims on three popular ANN types (RBMs, multilayer perceptrons (MLPs), and convolutional neural networks (CNNs)), on two types of tasks (supervised and unsupervised learning), and on 15 benchmark datasets. We hope that our approach will enable ANNs having billions of neurons and evolved topologies to be capable of handling complex realworld tasks that are intractable using stateoftheart methods.
Results
SET method
With SET, the bipartite ANN layers start from a random sparse topology (i.e. Erdös–Rényi random graph^{24}), evolving through a random process during the training phase towards a scalefree topology. Remarkably, this process does not have to incorporate any constraints to force the scalefree topology. But our evolutionary algorithm is not arbitrary: it follows a phenomenon that takes place in realworld complex networks (such as biological neural networks and protein interaction networks). Starting from an Erdős–Rényi random graph topology and throughout millenia of natural evolution, networks end up with a more structured connectivity, i.e. scalefree^{7} or smallworld^{8} topologies.
The SET algorithm is detailed in Box 1 and exemplified in Fig. 1. Formally, let us define a sparse connected (SC^{k}) layer in an ANN. This layer has n^{k} neurons, collected in a vector h^{k} = \(\left[ {h_1^k,h_2^k, \ldots ,h_{n^k}^k} \right]\). Any neuron from h^{k} is connected to an arbitrary number of neurons belonging to the layer below, h^{k−1}. The connections between the two layers are collected in a sparse weight matrix \({\bf{W}}^k \in {\bf{R}}^{n^{k  1} \times n^k}\). Initially, W^{k} is a Erdös–Rényi random graph, in which the probability of a connection between the neurons \(h_i^k\) and \(h_j^{k  1}\) is given by
whereby ε ∈ R^{+} is a parameter of SET controlling the sparsity level. If \(\varepsilon \ll n^k\) and \(\varepsilon \ll n^{k + 1}\) then there is a linear number of connections (i.e. nonzero elements), \(n^W = \left {{\mathbf{W}}^k} \right\) = \(\varepsilon \left( {n^k + n^{k  1}} \right)\), with respect to the number of neurons in the sparse layers. In the case of fullyconnected layers the number of connections is quadratic, i.e. n^{k}n^{k−1}.
However, it may be that this random generated topology is not suited to the particularities of the data that the ANN model tries to learn. To overcome this situation, during the training process, after each training epoch, a fraction ζ of the smallest positive weights and of the largest negative weights of SC^{k} is removed. These removed weights are the ones closest to zero, thus we do not expect that their removal will notably change the model performance. This has been shown, for instance, in refs. ^{13,25} using more complex approaches to remove unimportant weights. Next, to let the topology of SC^{k} to evolve so as to fit the data, an amount of new random connections, equal to the amount of weights removed previously, is added to SC^{k}. In this way, the number of connections in SC^{k} remains constant during the training process. After the training ends, we keep the topology of SC^{k} as the one obtained after the last weight removal step, without adding new random connections. To illustrate better these processes, we make the following analogy. If we assume a connection as the entity which evolves over time, the removal of the least important connections corresponds, roughly, to the selection phase of natural evolution, while the random addition of new connections corresponds, roughly, to the mutation phase of natural evolution.
It is worth highlighting that in the initial phase of conceiving the SET procedure, the weightremoval and weightaddition steps after each training epoch were introduced based on our own intuition. However, in the last phases of preparing this paper, we have found that there is a similarity between SET and a phenomenon which takes place in biological brains, named synaptic shrinking during sleep. This phenomenon has been demonstrated in two recent papers^{26,27}. In short, it was found that during sleep the weakest synapses in the brain shrink, while the strongest synapses remain unaltered, supporting the hypothesis that one of the core functions of sleeping is to renormalize the overall synaptic strength increased while awake^{27}. By keeping the analogy, this is—in a way—what happens also with the ANNs during the SET procedure.
We evaluate SET in three types of ANNs, RBMs^{28}, MLPs, and CNNs^{3} (all three are detailed in the Methods section), to experiment with both unsupervised and supervised learning. In total, we evaluate SET on 15 benchmark datasets, as detailed in Table 1, covering a wide range of fields in which ANNs are employed, such as biology, physics, computer vision, data mining, and economics. We also assess SET in combination with two different training methods, i.e. contrastive divergence^{29} and SGD^{3}.
Performance on RBMs
First, we have analyzed the performance of SET on a bipartite undirected stochastic ANN model, i.e. RBM^{28}, which is popular for its unsupervised learning capability^{30} and high performance as a feature extractor and density estimator^{31}. The new model derived from the SET procedure was dubbed SETRBM. In all experiments, we set ε = 11, and ζ = 0.3, performing a small random search just on the MNIST dataset, to be able to assess if these two metaparameters are dataset specific or if their values are general enough to perform well also on different datasets.
There are few studies on RBM connectivity sparsity^{10}. Still, to get a good estimation of SETRBM capabilities we compared it against RBM_{FixProb}^{10} (a sparse RBM model with a fixed Erdős–Rényi topology), fullyconnected RBMs, and with the stateoftheart results of XBMs from ref. ^{10}. We chose RBM_{FixProb} as a sparse baseline model to be able to understand better the effect of SETRBM adaptive connectivity on its learning capabilities, as both models, i.e. SETRBM and RBM_{FixProb}, are initialized with an Erdös–Rényi topology. We performed experiments on 11 benchmark datasets coming from various domains, as depicted in Table 1, using the same splitting for training and testing data as in ref. ^{10}. All models were trained for 5000 epochs using contrastive divergence^{29} (CD) with 1, 3, and 10 CD steps, a learning rate of 0.01, a momentum of 0.9, and a weight decay of 0.0002, as discussed in ref. ^{32}. We evaluated the generative performance of the scrutinized models by computing the logprobabilities on the test data using annealed importance sampling (AIS)^{33}, setting all parameters as in refs. ^{10,33}. We have used MATLAB for this set of experiments. We implemented SETRBM and RBM_{FixProb} ourselves; while for RBM and AIS we have adapted the code provided by Salakhutdinov and Murray^{33}.
Figure 2 depicts the model’s performance on the DNA dataset; while Supplementary Fig. 1 presents results on all datasets, using varying numbers of hidden neurons (i.e. 100, 250, and 500 hidden neurons for the UCI evaluation suite datasets; and 500, 2500, and 5000 hidden neurons for the CalTech 101 Silhouettes and MNIST datasets). Table 2 summarizes the results, presenting the best performer for each type of model for each dataset. In 7 out of 11 datasets, SETRBM outperforms the fullyconnected RBM, while reducing the parameters by a few orders of magnitude. For instance, on the MNIST dataset, SETRBM reaches −86.41 nats (natural units of information), with a 5.29fold improvement over the fullyconnected RBM, and a parameters reduction down to 2%. In 10 out of 11 datasets, SETRBM outperforms XBM, which represents the stateoftheart results on these datasets for sparse variants of RBM^{10}. It is interesting to see in Table 2 that RBM_{FixProb} reaches its best performance on each dataset in the case when the maximum number of hidden neurons is considered. Even if SETRBM has the same amount of weights with RBM_{FixProb}, it reaches its maximum performance on 3 out of the 11 datasets studied just when a medium number of hidden neurons is considered (i.e. DNA, Mushrooms, and CalTech 101 Silhouettes 28 × 28).
Figure 2 and Supplementary Fig. 1 present striking results on stability. Fullyconnected RBMs show instability and overfitting issues. For instance, using one CD step on the DNA dataset the RBMs have a fast learning curve, reaching a maximum after several epochs. After that, the performance start to decrease giving a sign that the models start to be overfitted. Moreover, as expected, the RBM models with more hidden neurons (Fig. 2b, c, e, f, h, i) overfit even faster than the one with less hidden neurons (Fig. 2a, d, g). A similar behavior can be seen in most of the cases considered, culminating with a very spiky learning behavior in some of them (Supplementary Fig. 1). Contrary to fullyconnected RBMs, the SET procedure stabilizes SETRBMs and avoids overfitting. This situation can be observed more often when a high number of hidden neurons is chosen. For instance, if we look at the DNA dataset, independently on the values of n^{h} and n^{CD} (Fig. 2), we may observe that SETRBMs are very stable after they reach around −85 nats, having almost a flat learning behavior after that point. Contrary, on the same dataset, the fullyconnected RBMs have a very short initial good learning behavior (for few epochs) and, after that, they go up and down during the 5000 epochs analyzed, reaching the minimum performance of −160 nats (Fig. 2i). Note that these good stability and overfitting avoidance capacities are induced not just by the SET procedure, but also by the sparsity itself, as RBM_{FixProb}, too, has a stable behavior in almost all the cases. This happens due to the very small number of optimized parameters of the sparse models in comparison with the high number of parameters of the fullyconnected models (as reflected by the stacked bar from the right yaxis of each panel of Fig. 2 and Supplementary Fig. 1) which does not allow the learning procedure to overfit the sparse models on the training data.
Furthermore, we verified our initial hypothesis about sparse connectivity in SETRBM. Figure 3 and Supplementary Fig. 2 show how the hidden neurons’ connectivity naturally evolves towards a scalefree topology. To assess this fact, we have used the null hypothesis from statistics^{34}, which assumes that there is no relation between two measured phenomena. To see if the null hypothesis between the degree distribution of the hidden neurons and a powerlaw distribution can be rejected, we have computed the pvalue^{35,36} between them using a onetailed test. To reject the null hypothesis the pvalue has to be lower than a statistically significant threshold of 0.05. In all cases (all panels of Fig. 3), looking at the pvalues (y axes to the right of the panels), we can see that at the beginning of the learning phase the null hypothesis is not rejected. This was to be expected, as the initial degree distribution of the hidden neurons is binomial due to the randomness of the Erdös–Rényi random graphs^{37} used to initialize the SETRBMs topology. Subsequently, during the learning phase, we can see that, in many cases, the pvalues decrease considerably under the 0.05 threshold. When these situations occur, it means that the degree distribution of the hidden neurons in SETRBM starts to approximate a powerlaw distribution. As to be expected, the cases with fewer neurons (Fig. 3a, b, d, e, g) fail to evolve to scalefree topologies, while the cases with more neurons always evolve towards a scalefree topology (Fig. 3c, f, h, i). To summarize, in 70 out of 99 cases studied (all panels of Supplementary Fig. 2), the SETRBMs hidden neurons’ connectivity evolves clearly during the learning phase from an Erdös–Rényi topology towards a scalefree one.
Moreover, in the case of the visible neurons, we have observed that their connectivity tends to evolve into a pattern that is dependent on the domain data. To illustrate this behavior, Fig. 4 shows what happens with the amount of connections for each visible neuron during the SETRBM training process on the MNIST and CalTech 101 datasets. It can be observed that initially the connectivity patterns are completely random, as given by the binomial distribution of the Erdös–Rényi topology. After the models are trained for several epochs, some visible neurons start to have more connections and others fewer and fewer. Eventually, at the end of the training process, some clusters of the visible neurons with clearly different connectivities emerge. Looking at the MNIST dataset, we can observe clearly that in both cases analyzed (i.e. 500 and 2500 hidden neurons) a cluster with many connections appeared in the center. At the same time, on the edges, another cluster appeared in which each visible neuron has zero or very few connections. The cluster with many connections corresponds exactly to the region where the digits appear in the images. On the Caltech 101 dataset, a similar behavior can be observed, except the fact that due to the high variability of shapes on this dataset the less connected cluster still has a considerable amount of connections. This behavior of the visible neurons’ connectivity may be used, for instance, to perform dimensionality reduction by detecting the most important features on highdimensional datasets, or to make faster the SETRBM training process.
Performance on MLPs
To better explore the capabilities of SET, we have also assessed its performance on classification tasks based on supervised learning. We developed a variant of MLP^{3}, dubbed SETMLP, in which the fullyconnected layers have been replaced with sparse layers obtained through the SET procedure, with ε = 20, and ζ = 0.3. We kept the ζ parameter as in the previous case of SETRBM, while for the ε parameter we performed a small random search just on the MNIST dataset. We compared SETMLP to a standard fullyconnected MLP, and to a sparse variant of MLP having a fixed Erdős–Rényi topology, dubbed MLP_{FixProb}. For the assessment, we have used three benchmark datasets (Table 1), two coming from the computer vision domain (MNIST and CIFAR10), and one from particle physics (the HIGGS dataset^{1}). In all cases, we have used the same data processing techniques, network architecture, training method (i.e. SGD^{3} with fixed learning rate of 0.01, momentum of 0.9, and weight decay of 0.0002), and a dropout rate of 0.3 (Table 3). The only difference between MLP, MLP_{FixProb}, and SETMLP, consisted in their topological connectivity. We have used Python and the Keras library (https://github.com/fchollet/keras) with Theano backend^{38} for this set of experiments. For MLP we have used the standard Keras implementation, while we implemented ourselves SETMLP and MLP_{FixProb} on top of the standard Keras libraries.
The results depicted in Fig. 5 show how SETMLP outperforms MLP_{FixProb}. Moreover, SETMLP always outperforms MLP, while having two orders of magnitude fewer parameters. Looking at the CIFAR10 dataset, we can see that with only just 1% of the weights of MLP, SETMLP leads to significant gains. At the same time, SETMLP has comparable results with stateoftheart MLP models after these have been carefully fine tuned. To quantify, the second best MLP model in the literature on CIFAR10 reaches about 74.1% classification accuracy^{39} and has 31 million parameters: while SETMLP reaches a better accuracy (74.84%) having just about 0.3 million parameters. Moreover, the best MLP model in the literature on CIFAR10 has 78.62% accuracy^{40}, with about 12 million parameters, while also benefiting from a pretraining phase^{41,42}. Although we have not pretrained the MLP models studied here, we should mention that SETRBM can be easily used to pretrain a SETMLP model to further improve performance.
With respect to the stability and overfitting issues, Fig. 5 shows that SETMLP is also very stable, similarly to SETRBM. Note that due to the use of the dropout technique, the fullyconnected MLP is also quite stable. Regarding the topological features, we can see from Fig. 5 that, similarly to what was found in the SETRBM experiments (Fig. 3), the hidden neuron connections in SETMLP rapidly evolve towards a powerlaw distribution.
To understand better the effect of various regularization techniques, and activation functions, we performed a small controlled experiment on the FashionMNIST dataset. We chose this dataset because it has a similar size with the MNIST dataset, being at the same time a harder classification problem. We used MLP, MLP_{FixProb}, and SETMLP with three hidden layers of 1000 hidden neurons each. Then, we varied for each model the following: (1) the weights regularization method (i.e. L1 regularization with a rate of 0.0000001, L2 regularization with a rate of 0.0002, and no regularization), (2) the use (or not use) of Nesterov momentum, and (3) two activation functions (i.e. SReLU^{43} and ReLU^{44}). The regularization rates were found by performing a small random search procedure with L1 and L2 levels between 0.01 and 0.0000001 to try maximizing the performance of all the three models. In all cases, we used SGD with 0.01 learning rate to train the models. The results depicted in Fig. 6 show that, in this specific scenario, SETMLP achieves the best performance if no regularization or L2 regularization is used for the weights, while L1 regularization does not offer the same level of performance. To summarize, SETMLP achieves the best results on the FashionMNIST dataset with the following settings: SReLU activation function, without Nesterov momentum, and without (or with L2) weights regularization. These being, in fact, the settings that we used in the MLP experiments discussed above. It is worth highlighting that independently on the specific setting, the general conclusion drawn up to now still holds. SETMLP achieves a similar (or better) performance to that of MLP, while having a much smaller number of connections. Also, SETMLP always clearly outperforms MLP_{FixProb}.
Performance on CNNs
As one of the most used ANN models nowadays are CNNs^{3}, we have briefly studied how SET can be used in the CNN architectures to replace their fully connected layers with sparse evolutionary counterparts. We considered a standard small CNN architecture, i.e. conv(32,(3,3))dropout(0.3)conv(32,(3,3))poolingconv(64,(3,3))dropout(0.3)conv(64,(3,3))poolingconv(128,(3,3))dropout(0.3)conv(128,(3,3))pooling), where the numbers in brackets for the convolutional layers mean (number of filters, (kernel size)), and for the dropout layers represent the dropout rate. Then, on top of the convolutional layers, we have used: (1) two fully connected layers to create a standard CNN, (2) two sparse layers with a fixed Erdős–Rényi topology to create a CNN_{FixProb}, and (3) two evolutionary sparse layers to create a SETCNN. For each model, each of the two layers on top was followed by a dropout (0.3) layer. On top of these, the CNN, CNN_{FixProb}, and SETCNN contained also a softmax layer. Even if SReLU seems to offer a slightly better performance, we used ReLU as activation function for the hidden neurons due to its wide utilization. We used SGD to train the models. The experiments were performed on the CIFAR10 dataset. The results are depicted in Fig. 7. They show, same as in the previous experiments with restricted Boltzmann machine and multilayer perceptron, that SETCNN can achieve a better accuracy than CNN, even if it has just about 4% of the CNN connections. To quantify this, we mention that in our experiments SETCNN reaches a maximum of 90.02% accuracy, CNN_{FixProb} achieves a maximum of 88.26% accuracy, while CNN achieves a maximum of 87.48% accuracy. Similar with the RBM experiments, we can observe that CNN is subject to a small overfitting behavior, while CNN_{FixProb} and SETCNN are very stable. Even if our goal was just to show that SET can be combined also with the widely used CNNs and not to optimize the CNN variants architectures to increase the performance, we highlight that, in fact, SETCNN achieves a performance comparable with stateoftheart results. The benefit of using SET in CNNs is twofold: to reduce the total number of parameters in CNNs and to permit the use of larger CNN models.
Last but not least, during all the experiments performed, we observed that SET is quite stable with respect to the choice of metaparameters ε and ζ. There is no way to say that our choices offered the best possible performance, even if we finetuned them just on one dataset, i.e. MNIST, and we evaluated their performance on all 15 datasets. Still, we can say that a ζ = 0.3 for both, SETRBM and SETMLP, and an ε specific for each model type, SETRBM (ε = 11), SETMLP (ε = 20), and SETCNN (ε = 20) were good enough to outperform stateoftheart.
Considering the different datasets under scrutiny, we stress that we have assessed both imageintensive and nonimage sets. On image datasets, CNNs^{3} typically outperform MLPs. However, CNNs are not viable on other types of highdimensional data, such as biological data (e.g.^{45}), or theoretical physics data (e.g.^{1}). In those cases, MLPs will be a better choice. This is in fact the case of the HIGGS dataset (Fig. 5e, f), where SETMLP achieves 78.47% classification accuracy and has about 90,000 parameters. Whereas, one of the best MLP models in the literature achieved a 78.54% accuracy, while having three times more parameters^{40}.
Discussion
In this paper, we have introduced SET, a simple and efficient procedure to replace ANNs’ fullyconnected bipartite layers with sparse layers. We have validated our approach on 15 datasets (from different domains) and on three widely used ANN models, i.e. RBMs, MLPs, and CNNs. We have evaluated SET in combination with two different training methods, i.e. contrastive divergence and SGD, for unsupervised and supervised learning. We showed that SET is capable of quadratically reducing the number of parameters of bipartite neural networks layers from the ANN design phase, at no decrease in accuracy. In most of the cases, SETRBMs, SETMLPs, and SETCNNs outperform their fullyconnected counterparts. Moreover, they always outperform their nonevolutionary counterparts, i.e. RBM_{FixProb}, MLP_{FixProb}, and CNN_{FixProb}.
We can conclude that the SET procedure is coherent with realworld complex networks, whereby nodes’ connections tend to evolve into scalefree topologies^{46}. This feature has important implications in ANNs: we could envision a computational time reduction by reducing the number of training epochs, if we would use for instance preferential attachment algorithms^{47} to evolve faster the topology of the bipartite ANN layers towards a scalefree one. Of course, this possible improvement has to be treated carefully, as forcing the model topology to evolve unnaturally faster into a scalefree topology may be prone to errors—for instance, the data distribution may not be perfectly matched. Another possible improvement would be to analyze how to remove the unimportant weights. In this article, we showed that it is efficient for SET to directly remove the connections with weights closest to zero. Note that we have tried also to remove connections randomly, and, as expected, this led to dramatic reductions in accuracy. Likewise, when we tried to remove the connections with the largest weights, the SETMLP model was not able to learn at all, performing similarly to a random classifier. However, we do not exclude the possibility that there may be better, more sophisticated approaches to removing connections, e.g. using gradient methods^{25}, or centrality metrics from network science^{48}.
SET can be widely adopted to reduce the fullyconnected layers into sparse topologies in other types of ANNs, e.g., recurrent neural networks^{3}, deep reinforcement learning networks^{2,49}, and so on. For a large scale utilization of SET, from the academic environment to industry, one more step has to be achieved. Currently, all stateoftheart deep learning implementations are based on very welloptimized dense matrix multiplications on graphics processing units (GPUs), while sparse matrix multiplications are extremely limited in performance^{50,51}. Thus, until optimized hardware for SETlike operations will appear (e.g., sparse matrix multiplications), one would have to find some alternative solutions. E.g., lowlevel parallel computations of neurons activations based just on their incoming connections and data batches to still perform dense matrix multiplications and to have a lowmemory footprint. If these software engineering challenges are solved, SET may prove to be the basis for much larger ANNs, perhaps on a billionnode scale, to run in supercomputers. Also, it may lead to the building of small but powerful ANNs, which could be directly trained on lowresource devices (e.g. wireless sensor nodes, mobile phones), without the need of first training them on supercomputers and then to move the trained models to lowresource devices, as is currently done by stateoftheart approaches^{13}. These powerful capabilities will be enabled by the linear relation between the number of neurons and the amount of connections between them yielded by SET. ANNs built with SET will have much more representational power, and better adaptive capabilities than the current stateoftheart ANNs, and we hope that they will create a new research direction in artificial intelligence.
Methods
Artificial neural networks
ANNs^{52} are mathematical models, inspired by biological neural networks, which can be used in all three machine learning paradigms (i.e. supervised learning^{53}, unsupervised learning^{53}, and reinforcement learning^{54}). These make them very versatile and powerful, as quantifiable by the remarkable success registered recently by the last generation of ANNs (also known as deep ANNs or deep learning^{3}) in many fields from computer vision^{3} to gaming^{2,49}. Just like their biological counterparts, ANNs are composed by neurons and weighted connections between these neurons. Based on their purposes and architectures, there are many models of ANNs, such as RBMs^{28}, MLPs^{55}, CNNs^{56}, recurrent neural networks^{57}, and so on. Many of these ANN models contain fullyconnected layers. A fullyconnected layer of neurons means that all its neurons are connected to all the neurons belonging to its adjacent layer in the ANN architecture. For the purpose of this paper, in this section we briefly describe three models that contain fullyconnected layers, i.e. RBMs^{28}, MLPs^{55}, and CNNs^{3}.
A restricted Boltzmann machine is a twolayer, generative, stochastic neural network that is capable to learn a probability distribution over a set of inputs^{28} in an unsupervised manner. From a topological perspective, it allows only interlayer connections. Its two layers are: the visible layer, in which the neurons represent the input data; and the hidden layer, in which the neurons represent the features automatically extracted by the RBM model from the input data. Each visible neuron is connected to all hidden neurons through a weighted undirected connection, leading to a fullyconnected topology between the two layers. Thus, the flow of information is bidirectional in RBMs, from the visible layer to the hidden layer, and from the hidden layer to the visible layer, respectively. RBMs, beside being very successful in providing very good initialization weights to the supervised training of deep artificial neural network architectures^{42}, are also very successful as stand alone models in a variety of tasks, such as density estimation to model human choice^{31}, collaborative filtering^{58}, information retrieval^{59}, multiclass classification^{60}, and so on.
MultiLayer Perceptron^{55} is a classical feedforward ANN model that maps a set of input data to the corresponding set of output data. Thus, it is used for supervised learning. It is composed by an input layer in which the neurons represent the input data, an output layer in which the neurons represent the output data, and an arbitrary number of hidden layers in between, with neurons representing the hidden features of the input data (to be automatically discovered). The flow of information in MLPs is unidirectional, starting from the input layer towards the output layer. Thus, the connections are unidirectional and exist just between consecutive layers. Any two consecutive layers in MLPs are fullyconnected. There are no connections between the neurons belonging to the same layer, or between the neurons belonging to layers which are not consecutive. In ref. ^{61}, it has been demonstrated that MLPs are universal function approximators, so they can be used to model any type of regression or classification problems.
CNNs^{3} are a class of feedforward neural networks specialized for image recognition, representing the stateoftheart on these type of problems. They typically contain an input layer, an output layers, and a number of hidden layers in between. From bottom to top, the first hidden layers are the convolutional layers, inspired by the biological visual cortex, in which each neuron receives information just from the previous layer neurons belonging to its receptive field. Then, the last hidden layers are fully connected ones.
In general, working with ANN models involves two phases: (1) training (or learning), in which the weighted connections between neurons are optimized using various algorithms (e.g. backpropagation procedure combined with SGD^{62,63} used in MLPs or CNNs, contrastive divergence^{29} used in RBMs) to minimize a loss function defined by their purpose; and (2) inference, in which the optimized ANN model is used to fulfill its purpose.
Scalefree complex networks
Complex networks (e.g. biological neural networks, actors and movies, power grids, transportation networks) are everywhere, in different forms, and different fields (from neurobiology to statistical physics^{4}). Formally, a complex network is a graph with nontrivial topological features, humanmade or naturemade. One of the most wellknown and deeply studied type of topological features in complex networks is scalefreeness, due to the fact that a wide range of realworld complex networks have this topology. A network with a scalefree topology^{7} is a sparse graph^{64} that approximately has a powerlaw degree distribution P(d) ~ d^{−γ}, where the fraction P(d) from the total nodes of the network has d connections to other nodes, and the parameter γ usually stays in the range γ ∈ (2, 3).
Data availability
The data used in this paper are public datasets, freely available online, as reflected by their corresponding citations from Table 1. Prototype software implementations of the models used in this study are freely available online at https://github.com/dcmocanu/sparseevolutionaryartificialneuralnetworks
References
Baldi, P., Sadowski, P. & Whiteson, D. Searching for exotic particles in highenergy physics with deep learning. Nat. Commun. 5, 4308 (2014).
Mnih, V. et al. Humanlevel control through deep reinforcement learning. Nature 518, 529–533 (2015).
LeCun, Y., Bengio, Y. & Hinton, G. Deep learning. Nature 521, 436–444 (2015).
Strogatz, S. H. Exploring complex networks. Nature 410, 268–276 (2001).
Pessoa, L. Understanding brain networks and brain organization. Phys. Life Rev. 11, 400–435 (2014).
Bullmore, E. & Sporns, O. Complex brain networks: graph theoretical analysis of structural and functional systems. Nat. Rev. Neurosci. 10, 186–198 (2009).
Barabási, A.L. & Albert, R. Emergence of scaling in random networks. Science 286, 509–512 (1999).
Watts, D. J. & Strogatz, S. H. Collective dynamics of ‘smallworld’ networks. Nature 393, 440–442 (1998).
Mocanu, D. C. On the synergy of network science and artificial intelligence. In Proc. 25th International Joint Conference on Artificial Intelligence(ed. Kambhampati, S.) 4020–4021 (AAAI Press, New York, 2016).
Mocanu, D. C., Mocanu, E., Nguyen, P. H., Gibescu, M. & Liotta, A. A topological insight into restricted boltzmann machines. Mach. Learn. 104, 243–270 (2016).
Dieleman, S. & Schrauwen, B. Accelerating sparse restricted boltzmann machine training using nongaussianity measures. In Proc. Deep Learning and Unsupervised Feature Learning, Vol. 9 (eds Bengio Y., Bergstra J. & Le Q.) http://hdl.handle.net/1854/LU3118568 (Lake Tahoe, 2012).
Yosinski, J. & Lipson, H. Visually debugging restricted boltzmann machine training with a 3d example. In Representation Learning Workshop, 29th International Conference on Machine Learning (Edinburgh, 2012).
Han, S., Pool, J., Tran, J. & Dally, W. Learning both weights and connections for efficient neural network. In Proc. Advances in Neural Information Processing Systems (eds Cortes, C., Lawrence, N. D., Lee, D. D., Sugiyama, M. & Garnett, R.) Vol. 28, 1135–1143 (MIT Press Cambridge, Montreal, 2015).
Mocanu, D. C. et al. Noreference video quality measurement: added value of machine learning. J. Electron. Imaging 24, 061208 (2015).
Whiteson, S. & Stone, P. Evolutionary function approximation for reinforcement learning. J. Mach. Learn. Res. 7, 877–917 (2006).
McDonnell, J. R. & Waagen, D. Evolving neural network connectivity. In Proc. IEEE International Conference on Neural Networks, Vol. 2, 863–868 (IEEE, San Francisco, 1993).
Miikkulainen, R. et al. Evolving deep neural networks. Preprint at https://arxiv.org/abs/1703.00548 (2017).
Kowaliw, T., Bredeche, N., Chevallier, S., & Doursat, R. Artificial neurogenesis: an introduction and selective review. In Growing Adaptive Machines: Combining Development and Learning in Artificial Neural Networks (Kowaliw, T., Bredeche, N. & Doursat, R.) 1–60 (Springer, Berlin, Heidelberg, 2014).
Stanley, K. O. & Miikkulainen, R. Evolving neural networks through augmenting topologies. Evol. Comput. 10, 99–127 (2002).
Hausknecht, M., Lehman, J., Miikkulainen, R. & Stone, P. A neuroevolution approach to general atari game playing. IEEE Trans. Comput. Intell. AI 6, 355–366 (2014).
Miconi, T. Neural networks with differentiable structure. Preprint at https://arxiv.org/abs/1606.06216 (2016).
Salimans, T., Ho, J., Chen, X., Sidor, S. & Openai, I. S. Evolution strategies as a scalable alternative to reinforcement learning. Preprint at https://arxiv.org/abs/1703.03864 (2017).
Such, F. P. et al. Deep neuroevolution: genetic algorithms are a competitive alternative for training deep neural networks for reinforcement learning. Preprint at https://arxiv.org/abs/1712.06567 (2018).
Erdös, P. & Rényi, A. On random graphs i. Publ. Math.Debr. 6, 290–297 (1959).
Weigend, A. S., Rumelhart, D. E. & Huberman, B. A. Generalization by weightelimination with application to forecasting. In Proc. Advances in Neural Information Processing Systems, Vol. 3, 875–882 (MorganKaufmann, Colorado, 1991).
Diering, G. H. et al. Homer1a drives homeostatic scalingdown of excitatory synapses during sleep. Science 355, 511–515 (2017).
de Vivo, L. et al. Ultrastructural evidence for synaptic scaling across the wake/sleep cycle. Science 355, 507–510 (2017).
Smolensky, P. Information processing in dynamical systems: foundations of harmony theory. In Parallel Distributed Processing: Explorations in the Microstructure of Cognition (eds Rumelhart, D. E., McClelland, J. L. & CORPORATE PDP Research Group) 194–281 (MIT Press, Cambridge, 1986).
Hinton, G. E. Training products of experts by minimizing contrastive divergence. Neural Comput. 14, 1771–1800 (2002).
Bengio, Y. Learning deep architectures for ai. Found. Trends Mach. Learn. 2, 1–127 (2009).
Osogami, T. & Otsuka, M. Restricted boltzmann machines modeling human choice. Proc. Adv. Neural Inf. Process. Syst. 27, 73–81 (2014).
Hinton, G. A practical guide to training restricted boltzmann machines. In Neural Networks: Tricks of the Trade, Vol. 7700 of Lecture Notes in Computer Science (eds Montavon, G., Orr, G. B. & Müller, K.R.) 599–619 (Springer, Berlin Heidelberg, 2012).
Salakhutdinov, R. & Murray, I. On the quantitative analysis of deep belief networks. In Proc. 25th International Conference on Machine Learning, 872–879 (ACM, Helsinki, 2008).
Everitt, B. The Cambridge Dictionary of Statistics (Cambridge University Press, Cambridge, UK; New York, 2002).
Nuzzo, R. Scientific method: statistical errors. Nature 506, 150–152 (2014).
Clauset, A., Shalizi, C. R. & Newman, M. E. J. Powerlaw distributions in empirical data. SIAM Rev. 51, 661–703 (2009).
Newman, M. E., Strogatz, S. H. & Watts, D. J. Random graphs with arbitrary degree distributions and their applications. Phys. Rev. E 64, 026118 (2001).
AlRfou, R., et al. Theano: a Python framework for fast computation of mathematical expressions. Preprint at https://arxiv.org/abs/1605.02688 (2016).
Urban, G. et al. Do deep convolutional nets really need to be deep and convolutional? In Proc. 5th International Conference on Learning Representations (OpenReview.net, Toulon, 2016).
Lin, Z., Memisevic, R. & Konda, K. How far can we go without convolution: improving fullyconnected networks. Preprint at https://arxiv.org/abs/1511.02580 (2015).
Hinton, G. E. & Salakhutdinov, R. R. Reducing the dimensionality of data with neural networks. Science 313, 504–507 (2006).
Hinton, G. E., Osindero, S. & Teh, Y.W. A fast learning algorithm for deep belief nets. Neural Comput. 18, 1527–1554 (2006).
Jin, X. et al. Deep learning with sshaped rectified linear activation units. In Proc. 30th AAAI Conference on Artificial Intelligence (eds Schuurmans, D. & Wellman, M.) 1737–1743 (AAAI Press, Phoenix, 2016).
Nair, V. & Hinton, G. E. Rectified linear units improve restricted boltzmann machines. In Proc. 27th International Conference on Machine Learning (eds Fürnkranz, J. & Joachims, T.) 807–814 (Omnipress, Haifa, 2010).
Danziger, S. A. et al. Functional census of mutation sequence spaces: the example of p53 cancer rescue mutants. IEEE ACM Trans. Comput. Biol. 3, 114–125 (2006).
Barabási, A.L. Network Science (Cambridge University Press, Glasgow, 2016).
Albert, R. & Barabási, A.L. Statistical mechanics of complex networks. Rev. Mod. Phys. 74, 47–97 (2002).
Mocanu, D. C., Exarchakos, G. & Liotta, A. Decentralized dynamic understanding of hidden relations in complex networks. Sci. Rep. 8, 1571 (2018).
Silver, D. et al. Mastering the game of go with deep neural networks and tree search. Nature 529, 484–489 (2016).
Lebedev, V. & Lempitsky, V. Fast ConvNets using groupwise brain damage. In Proc. 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2554–2564 (IEEE, Las Vegas, 2016).
Changpinyo, S., Sandler, M. & Zhmoginov, A. The power of sparsity in convolutional neural networks. Preprint at https://arxiv.org/abs/1702.06257 (2017).
Bishop, C. M. Pattern Recognition and Machine Learning (Information Science and Statistics) (SpringerVerlag New York, Inc., Secaucus, 2006).
Hastie, T., Tibshirani, R. & Friedman, J. The Elements of Statistical Learning. (Springer New York Inc., New York, NY, USA, 2001).
Sutton, R. S. & Barto, A. G. Introduction to Reinforcement Learning. (MIT Press, Cambridge, MA, USA, 1998).
Rosenblatt, F. Principles of Neurodynamics: Perceptrons and the Theory of Brain Mechanisms (Spartan, Washington, 1962).
LeCun, Y., Bottou, L., Bengio, Y. & Haffner, P. Gradientbased learning applied to document recognition. Proc. IEEE 86, 2278–2324 (1998).
Graves, A. et al. A novel connectionist system for unconstrained handwriting recognition. IEEE Trans. Pattern Anal. Mach. Intell. 31, 855–868 (2009).
Salakhutdinov, R., Mnih, A. & Hinton, G. Restricted boltzmann machines for collaborative filtering. In Proc. 24th International Conference on Machine Learning (ed. Ghahramani, Z.) 791–798 (ACM, Corvallis, 2007).
Gehler, P. V., Holub, A. D. & Welling, M. The rate adapting poisson model for information retrieval and object recognition. In Proc. 23rd International Conference on Machine Learning (eds Cohen, W. & Moore, A.) 337–344 (ACM, Pittsburgh, 2006).
Larochelle, H. & Bengio, Y. Classification using discriminative restricted boltzmann machines. In Proc. 25th International Conference on Machine Learning (eds McCallum, A. & Roweis, S.) 536–543 (ACM, Helsinki, 2008).
Cybenko, G. Approximation by superpositions of a sigmoidal function. Math. Control Signal 2, 303–314 (1989).
Rumelhart, D., Hintont, G. & Williams, R. Learning representations by backpropagating errors. Nature 323, 533–536 (1986).
Bottou, L. & Bousquet, O. The tradeoffs of large scale learning. In Proc. Advances in Neural Information Processing Systems Vol. 20 (eds Platt, J. C., Koller, D., Singer, Y. & Roweis, S. T.) 161–168 (NIPS Foundation, Vancouver, 2008).
Del Genio, C. I., Gross, T. & Bassler, K. E. All scalefree networks are sparse. Phys. Rev. Lett. 107, 178701 (2011).
Larochelle, H. & Murray, I. The neural autoregressive distribution estimator. In Proc. 14th International Conference on Artificial Intelligence and Statistics(eds Gordon, G., Dunson, D. & Dudík, M.) 29–37 (JMLR, Fort Lauderdale, 2011).
Marlin, B. M., Swersky, K., Chen, B. & de Freitas, N. Inductive principles for restricted boltzmann machine learning. In Proc. 13th International Conference on Artificial Intelligence and Statistics (eds Teh, Y. W. & Titterington, M.) 509–516 (JMLR, Sardinia, 2010).
LeCun, Y., Bottou, L., Bengio, Y. & Haffner, P. Gradientbased learning applied to document recognition. Proc. IEEE 86, 2278–2324 (1998).
Krizhevsky, A. Learning Multiple Layers of Features from Tiny Images. Master’s thesis (2009).
Xiao, H., Rasul, K. & Vollgraf, R. Fashionmnist: a novel image dataset for benchmarking machine learning algorithms. Preprint at https://arxiv.org/abs/1708.07747 (2017).
Author information
Authors and Affiliations
Contributions
D.C.M. and E.M. conceived the initial idea. D.C.M., E.M., P.S., P.H.N., M.G., and A.L. designed the experiments and analyzed the results. D.C.M. performed the experiments. D.C.M., E.M., P.S., P.H.N., M.G., and A.L. wrote the manuscript.
Corresponding author
Ethics declarations
Competing interests
The authors declare no competing interests.
Additional information
Publisher's note: Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Electronic supplementary material
Rights and permissions
Open Access This article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons license, and indicate if changes were made. The images or other third party material in this article are included in the article’s Creative Commons license, unless indicated otherwise in a credit line to the material. If material is not included in the article’s Creative Commons license and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this license, visit http://creativecommons.org/licenses/by/4.0/.
About this article
Cite this article
Mocanu, D.C., Mocanu, E., Stone, P. et al. Scalable training of artificial neural networks with adaptive sparse connectivity inspired by network science. Nat Commun 9, 2383 (2018). https://doi.org/10.1038/s41467018043163
Received:
Accepted:
Published:
DOI: https://doi.org/10.1038/s41467018043163
This article is cited by

Deep limits of residual neural networks
Research in the Mathematical Sciences (2023)

Quick and robust feature selection: the strength of energyefficient sparse training for autoencoders
Machine Learning (2022)

Nonsmooth Bayesian learning for artificial neural networks
Journal of Ambient Intelligence and Humanized Computing (2022)

Predicting ionospheric precursors before large earthquakes using neural network computing and the potential development of an earthquake early warning system
Natural Hazards (2022)

Model reduction of feed forward neural networks for resourceconstrained devices
Applied Intelligence (2022)
Comments
By submitting a comment you agree to abide by our Terms and Community Guidelines. If you find something abusive or that does not comply with our terms or guidelines please flag it as inappropriate.