August 10, 2023
We present a demonstration of image classification using an extreme learning machine (ELM) based on a unique simulated magnetic tunnel junction (MTJ) delayed in time. As the ground state of the MTJ is a magnetic vortex, we refer to it as a vortex-based spin-torque oscillator (STVO). The dynamics of the magnetic vortex is simulated with a model called the data-driven Thiele equation approach (DD-TEA). This allows to avoid the constraints associated with repeated experimental manipulation for hyperparameters search and benchmarking. We showcase the versatility of our implementation by using it successfully for classification tasks on the MNIST, EMNIST-letters and Fashion MNIST datasets. Through simulations, we show that within an ELM with a sufficient number of parameters, the performance reached using the STVO dynamics as a source of nonlinearity is equivalent to the ones obtained with classical software activation functions such as the reLU and the sigmoid. While achieving state-of-the-art accuracy levels on the MNIST dataset, our model’s performance on EMNIST-letters and Fashion MNIST is lower due to the simplicity of the network architecture and the increased complexity of the data. We expect that the DD-TEA framework will enable the exploration of deeper and more complex STVO-based architectures, ultimately leading to improved classification accuracy.
The impact of artificial intelligence (AI) on various industries and our daily lives is undeniable [1]. However, the widespread use of AI has raised concerns about its energy consumption and environmental impact during resource-intensive training and large-scale inference phases [2], [3]. With the rapid development of large generative models, the electricity demand of AI is now growing fast enough to even match that of small countries [4]. Moreover, conventional computers, despite decades of miniaturization and optimization, are nearing their limits in terms of computing power [5]–[7].
To face these challenges, new approaches in artificial intelligence have emerged, aiming to surpass the limitations of digital frameworks [8]. Among them, reservoir computing bypasses the training of the network parameters by using a large recurrent network with random and fixed connections [9]. Within this framework, the input data are projected into a higher-dimensional space in which the categories become linearly separable. The parameters of the readout layer are learned using a simple linear regression, which is computationally efficient and can be performed in a single shot over the whole training set. This approach hence allows to find a global minimum of the loss function while saving time, energy and avoiding the vanishing gradient problem observed with traditional algorithms relying on gradient descent and error backpropagation. While reservoir computing was initially developed for processing time-dependent data, it can be generalized to the classification of static data with extreme learning machines (ELM) [10]. ELMs are feed-forward neural networks where a randomly initialized hidden layer stays fixed and acts as a random feature mapper. The output layer is the only part of the network requiring training, which is usually achieved through a simple linear regression, similarly to reservoir computing. The fixed nature of the reservoir (or of the hidden layer in the case of ELMs) makes it well-suited for hardware implementations, enabling the creation of dedicated hardware analog processors to optimize performance and energy efficiency [11]. Research has also demonstrated that such reservoirs could be implemented using a single nonlinear device delayed in time without sacrificing performance [12]–[14], by trading on the execution time. By sampling the output of the device repeatedly, one is able to emulate a large number of virtual nodes connected in time instead of space. This method, called time-multiplexing, further eases the hardware implementation of these systems [14]–[17].
Efficient data classification using physical reservoir computing has been demonstrated in various works utilizing spintronic oscillators known as vortex-based spin-torque oscillators [18] (STVOs). STVOs are magnetic tunnel junctions (MTJs) whose behavior can be described using spintronics, the study of the transport of spin in magnetic materials and nanostructures [19]. The ground state magnetization of STVOs is a magnetic vortex, composed of in-plane curling magnetization with an out-of-plane magnetization vortex core at the center (Fig. 1a). When a current density signal is injected into a STVO (Fig. 1b), the vortex core undergoes circular oscillations in the plane of the MTJ (pink circular arrow in Fig. 1a) due to an effect called the spin-transfer torque [18], [20], [21] (STT). Thanks to another phenomenon known as tunnel magnetoresistance[22] (TMR), stable oscillations of the electrical resistance are obtained, as well as voltage oscillations in the MHz range across the MTJ (Fig. 1c). The amplitude of these voltage oscillations is nonlinearly related to the amplitude of the input signal, making STVOs suitable for nonlinear data transformation in reservoir computing applications [23]. This nonlinearity can be retrieved from the evolution of an internal state quantity of the STVO: the time-varying reduced position of the vortex core \(s(t)=\vert\vert X(t)\vert\vert/R\), where \(\vert\vert X(t)\vert\vert\) is the time-dependent position of the vortex core in the plane of the MTJ and \(R\) is the radius of the MTJ [24]. STVOs offer numerous advantages, such as low power consumption, minimal noise and compatibility with complementary metal-oxide-semiconductor (CMOS) technology [25], making them attractive candidates for nonlinear data transformation in reservoir computing applications [17], [26]–[29].
Despite the progress in STVO-based hardware computing systems, a significant challenge lies in their development, optimization and testing at large scale. Indeed, the performance of these systems is highly dependent on the choice of the hyperparameters, such as the number of virtual nodes in the reservoir (or the hidden layer in the case of ELMs), the bias current density \(J_\text{dc}\), the peak-to-peak amplitude of the input signal, and the sampling rate. Despite being crucial, hyperparameters search remains a time-consuming and resource-intensive task, especially within experimental frameworks. To overcome these limitations, simulating STVOs and their complex dynamics proves to be the easiest approach. Micromagnetic simulations [30] (MMS) offer highly accurate results but are time and energy-consuming, making them impractical for large scale simulations. A computationally lighter alternative is the Thiele equation approach (TEA), which uses a single ordinary differential equation for each direction of space to model the STVO dynamics [31]. TEA is unfortunately merely qualitative in the steady and transient oscillatory states, which respectively describe stable magnetization oscillations and transitions from a stable state to another one. Both of these regimes however play a key role in nonlinearly transforming the data [17], [24]. A recent solution to this challenge is the data-driven Thiele equation approach (DD-TEA) developed by Abreu Araujo et al. [24]. This semi-analytical model accurately describes the STVO dynamics in both steady and transient states by using numerical parameters extracted from a small set of MMS. DD-TEA enables STVO simulation with the same level of accuracy as MMS, but with a significant acceleration factor of \(9\) orders of magnitude [32]. While the numerical parameters used in this work are extracted for a specific STVO geometry, the underlying methodology consisting in fitting the coefficients of a reduced analytical model to a small set of micromagnetic simulations is not tied to this particular device. It can in principle be also transposed to other oscillator-based nanodevices whose dynamics admit an analytical description, making the present approach a template rather than a single-device solution.
The present study shows the ability of a simulated STVO-based ELM to achieve state-of-the-art accuracy for image classification on the MNIST dataset. First, we present the implementation of a time-multiplexed ELM based on a single STVO delayed in time, as well as the data preprocessing and training procedures. We then explain how the STVO dynamics is simulated using the DD-TEA framework. Finally, we evaluate the network’s performance in classifying various types of images, such as handwritten digits from MNIST [33], latin characters from EMNIST-letters [34] and clothing items from Fashion MNIST [35] (FMNIST). We compare the STVO dynamics to more conventional nonlinear functions (reLU and sigmoid) in order to highlight the usability of the vortex dynamics as a hardware activation function. Those studies are carried out for various numbers of virtual nodes in the ELM to demonstrate the potential of DD-TEA for large-scale hyperparameters research.
The data-driven Thiele equation approach underlying our STVO model was introduced in Ref. [24], and a first proof of concept of its use for accelerated neuromorphic computing was reported in Ref. [32], with an early single demonstration on the MNIST dataset. The present work is however self-contained and extends these earlier reports in both model fidelity and scope. The STVO nonlinearity is computed here with the full high-precision form of the model (Eqs. 7 –10 ), rather than a reduced approximation, ensuring quantitative agreement with micromagnetic simulations across the whole operating range. More importantly, we move from a single demonstration to a systematic study: we benchmark the same single-STVO ELM on three distinct image datasets, compare the STVO dynamics with standard software activation functions to establish their equivalence, and take advantage of the speed of the model to map the performance against the reservoir size. All the results reported below are produced within a single pipeline, so that the present study can be read and reproduced independently of any earlier work.
ELMs first send the input data in a random higher-dimension space, before transforming the projected data nonlinearly. If the dimension of the projection space is sufficiently high, the categories can be separated using a simple linear classifier, which is the only part of the model requiring training. The present work presents a partially physical ELM for classifying the MNIST dataset, where the nonlinear transformation is performed by a single STVO delayed in time, similarly to the single chain time-delayed reservoir presented in Ref. [13].
We first apply a denoising and dimensionality reduction procedure to the input data using principal component analysis [36], [37] (PCA). PCA extracts the directions (i.e., linear combinations of pixels) along which the variance of the images is the highest. This allows us to reduce the dimension of the images from \(784\) pixels to \(44\) pixels combinations while still explaining \(80\%\) of the variance in the training set. First, each image is flattened into a column vector \(\mathbf{x}\) of \(784\) pixels whose value is the grayscale intensity normalized between \(0\) and \(1\). Once the PCA components matrix \(\mathbf{C}\) is computed based on the training data, the significant components are extracted from each input image \(\mathbf{x}\) accordingly to Eq. 1 (see blue part in Fig. 2).
\[\mathbf{x}' = \mathbf{C}\mathbf{x} \label{eq:pca}\tag{1}\]
Each sample \(\mathbf{x}'\) is then projected into a space of dimension \(N_\theta >> 44\) using a random matrix \(\mathbf{M}\) whose elements are drawn from \(U(-1, 1)\) (Eq. 2 ). Note that \(\mathbf{M}\) is defined only once and stays fixed for the whole training and inference phases. More pragmatically, this operation corresponds to propagating the data into a hidden layer of \(N_\theta\) nodes with random weights (orange part in Fig. 2).
\[\mathbf{x}'' = \mathbf{M}\mathbf{x'} \label{eq:masking}\tag{2}\]
The projected data samples \(\mathbf{x}''\) are then nonlinearly transformed using the STVO dynamics, which effectively corresponds to the application of a neural activation function. Each \(\mathbf{x}''\) is first converted into a current density signal \(\mathbf{J}\) by scaling each of its elements linearly into a predefined range. The STVO transformation of \(\mathbf{J}\) is simulated by computing the reduced position of the vortex core \(\mathbf{s}\) using \(\mathbf{J}\) as an input signal (Eq. 3 ) and the model described later in subsection 2.2 (central part in Fig. 2). The value \(1\) is stacked on top of the vector to account for the bias term in the linear classification step (see next subsection).
\[\mathbf{s} = s(\mathbf{J}) = \begin{bmatrix}1\\s(J (x''_1))\\ \vdots \\s(J(x''_{N_\theta}))\end{bmatrix} \label{eq:nonlinearity}\tag{3}\]
The STVO output \(\mathbf{s}\) of all the samples in the training set are concatenated to form the matrix \(\mathbf{S}_\text{train}\) (dimension \((N_\theta +1) \times 60000\)). In parallel, the target matrix \(\mathbf{T}_\text{train}\) (dimension \(10 \times 60000\)) is formed by encoding the category of each training sample in a one-hot vector (zeros everywhere except for a one at the index of the correct class). The weights matrix of the output layer \(\mathbf{W}_\text{out}\) (dimension \(10 \times (N_\theta +1)\)), which must satisfy the first part of Eq. 4 , is computed with the Moore-Penrose pseudo-inverse [38]–[40] of \(\mathbf{S}_\text{train}\) (second part of Eq. 4 ).
\[\mathbf{W}_\text{out}\mathbf{S}_\text{train} = \mathbf{T}_\text{train} \Leftrightarrow \mathbf{W}_\text{out} = \mathbf{T}_\text{train}\mathbf{S}_\text{train}^\dagger \label{eq:Wdef}\tag{4}\]
The linear system from Eq. 4 is strongly overdetermined, as the number of virtual nodes remains far below the number of training samples (\(N_\theta \ll 60{,}000\)). Therefore, the regular Moore-Penrose pseudo-inverse can be used rather than a ridge regression, without observing any overfitting (Sec. 3). We still verified this assumption by evaluating the use of a ridge regularization parameter \(\lambda\) on a validation set. The optimal parameter is negligible and the resulting test accuracy is similar from that of the pseudo-inverse, confirming that regularization is not necessary in this case.
As a result, \(\mathbf{W}_\text{out}\) effectively contains the parameters of a linear mapping between a (new) STVO output \(\mathbf{s}\) and the most likely one-hot categorical vector \(\mathbf{\hat{y}}\) (Eq. 5 ). The corresponding predicted digit \(\hat{d}\) is obtained by selecting the index of the maximum element in \(\mathbf{\hat{y}}\) using the argmax function (Eq. 6 ) as depicted in the rightmost part of Fig. 2.
\[\mathbf{\hat{y}} = \mathbf{W}_\text{out}\mathbf{s} \label{eq:output}\tag{5}\]
\[\hat{d} = \text{argmax}(\mathbf{\hat{y}}) \label{eq:argmax}\tag{6}\]
To compute Eq. 3 we simulate the reduced position \(s\) of the vortex core for the input current density signal \(\mathbf{J}\) made from the input data. This can be analytically computed within the DD-TEA [24] framework using the recurrent Eq. 7 . The parameters \(\alpha_i\), \(\beta_i\) and \(n_i\) are derived from the TEA and rule the time-dependent nonlinearity of the STVO dynamics. In the DD-TEA framework, they are polynomials of the input signal \(\mathbf{J}\) fitted through MMS [24] (Eqs. 8 , 9 and 10 ). The input signal \(\mathbf{J}\) is obtained by scaling the input data into a current density range of \(\Delta J \approx 2.84\) MA/cm² centered around a bias density of \(J_\text{dc} \approx 6.32\) MA/cm² (i.e., \(\mathbf{J} \in [J_\text{dc}-\Delta J/2, J_\text{dc}+\Delta J/2]\)), corresponding to a working current \(I_\text{w} \approx 1.99\) mA and a signal voltage span of \(\Delta V = 125\) mV across the oscillator. The \(D_\text{t}\) parameter represents the duration for which each value in \(\mathbf{J}\) is injected into the STVO, and the sampling period of the output signal \(\mathbf{s}\). It is chosen shorter than the characteristic transient time of the STVO to ensure that the observed dynamics is always nonlinear, and is set to \(D_\text{t} = 10\) ns in this work. The initial value \(s_0\) is the value of \(s\) when \(J_\text{dc}\) is injected into the oscillator for a duration \(t >> D_\text{t}\). In the context of reservoir computing, the model described in Eq. (7 ) plays the role of the internal nonlinear function implemented by the virtual STVOs in the reservoir [11], [15], [17], [41]–[43].
\[s_i = s_{i-1}{\left[\left(1+\dfrac{s_{i-1}^{n_i}}{\alpha_{i}/\beta_i}\right)\exp\left(-n_i\alpha_i D_t\right)-\dfrac{s_{i-1}^{n_i}}{\alpha_i/\beta_i}\right]}^{-1/n_i} \label{eq:hptea}\tag{7}\]
\[\label{eq:alpha} \alpha_i = 6.64 J_{i} -39.97\tag{8}\]
\[\label{eq:beta} \beta_i = -0.43 J_{i}-25.92\tag{9}\]
\[\begin{align} \label{eq:n} n_i = &-95.97 J_{i} + 25.54 J_{i}^2 -2.87 J_{i}^3\notag\\&+ 0.18 J_{i}^4 + 157.14 \end{align}\tag{10}\]
Our simulated hardware ELM is benchmarked on the MNIST dataset, as well as on two additional image datasets in order to assess its generalizability. The first one is the EMNIST-letters dataset: an extension of MNIST to handwritten letters [34], consisting of \(124,800\) training images and \(20,800\) testing images classified into \(26\) classes corresponding to the \(26\) letters of the alphabet. Each class contains lower and upper case letters. The second dataset is FMNIST [35] and is composed of \(60,000\) training images and \(10,000\) testing images, all extracted from the Zalando website. These images are separated into \(10\) classes representing clothing items: {t-shirt/top, trouser, pullover, dress, coat, sandal, shirt, sneaker, bag, ankle boot}.
A STVO-based ELM is simulated and trained for each dataset accordingly to the procedure described in the previous section. The number of PCA components selected is chosen to ensure an explained variance of \(80\%\) in each training set, allowing to reduce the dimension of the data from \(784\) to \(38\) (resp. \(24\)) for the EMNIST-letters (resp. FMNIST) dataset. For each dataset, we compute the test accuracy (i.e., the ratio of correctly classified unseen samples) and the normalized root-mean-square error (NRMSE) defined as in Eq. 11 [41], with \(\mathbf{\hat{y}}\) the output of the ELM, \(\mathbf{t}\) the corresponding target, and \(\langle\cdot\rangle\) the average operator. \[\label{eq:nrmse} \text{NRMSE} = \sqrt{\dfrac{\left\langle\vert\vert\mathbf{\hat{y}}-\mathbf{t}\vert\vert^2\right\rangle}{\left\langle\vert\vert\mathbf{t}-\left\langle\mathbf{t}\right\rangle\vert\vert^2\right\rangle}}\tag{11}\] The metrics are averaged over \(10\) iterations using different randomly generated versions of \(\mathbf{M}\), and the whole process is repeated for an increasing number of virtual neurons in the nonlinear layer (i.e., an increasing \(N_\theta\)). This highlights the potential of DD-TEA for performing high-throughput parametric studies for hyperparameter optimization.
Finally, the performance are compared to the same architecture with more conventional nonlinear transfer functions such as the reLU and the sigmoid. This is achieved by artificially replacing Eq. (7 ) by the corresponding expressions in the simulations. The identity activation is also be tested as an activation function to underline the role of the nonlinearity in the classification ability of the ELM.
The test accuracy and the NRMSE of our STVO-based ELM on the MNIST dataset are presented in Fig. 4. Without surprise, the accuracy (resp. NRMSE) increases (resp.decreases) with an increasing number of nodes \(N_\theta\) in the nonlinear layer. As a matter of fact, the number of nodes is directly proportional to the number of tunable parameters in the model (i.e., the number of elements in the learned matrix \(\mathbf{W}\)). The shaded areas on the plots represent the range where the input data \(\mathbf{x}\) is projected into a smaller dimension space due to the low amount of virtual nodes in the nonlinear layer (i.e., \(N_\theta < N_f\)), leading to data compression instead of expansion. As a result, the performance is poor in this range, and similar to the one obtained with a linear classifier with \(N_\theta\) parameters. Yet, the performance still increase with \(N_\theta\) in this regime, as it increases the size of the compression space and hence the quality of the linear classifier. As soon as the size of the projection space reaches the dimension of the input data (i.e., \(N_\theta\geq 44\)), the performance obtained with the linear classifier saturates because the projected data simply become linear expansions of the input data, and the absence of nonlinearity prevents the learning of complex patterns (red curve in Fig. 4). More pragmatically, if we consider a linear transfer function represented by \(s\left(\mathbf{x}''\right) = k\mathbf{x}''\) with \(k\) a constant, the predicted category of an image \(\mathbf{x}\) can be summarized by \[\label{eq:inf95summary} \hat{d} = \text{argmax}\left(\mathbf{T}_\text{train}\mathbf{S}^\dagger_\text{train}k\mathbf{M}\mathbf{C}\mathbf{x}\right).\tag{12}\] Substituting \(\mathbf{S}_\text{train}\) by \(k\mathbf{M}\mathbf{C}\mathbf{X}_{\text{train}}\), where \(\mathbf{X}_{\text{train}}\) is the concatenation of the \(60000\) training samples, yields \[\begin{align} \label{eq:inf95summary2} \hat{d} &= \text{argmax}\left(\mathbf{T}_\text{train}{\left(k\mathbf{M}\mathbf{C}\mathbf{X}_{\text{train}}\right)}^\dagger k\mathbf{M}\mathbf{C}\mathbf{x}\right)\notag\\ &= \text{argmax}\left(\mathbf{T}_\text{train}\mathbf{X}_{\text{train}}^\dagger {\left(k\mathbf{M}\mathbf{C}\right)}^\dagger k\mathbf{M}\mathbf{C}\mathbf{x}\right). \end{align}\tag{13}\] As \({\left(k\mathbf{M}\mathbf{C}\right)}^\dagger\left(k\mathbf{M}\mathbf{C}\right)=\mathbf{I}\) (the identity matrix), Eq. (12 ) finally reduces to Eq. (14 ), which is independent from the network’s parameters, and where the matrix \((\mathbf{T}_\text{train}\mathbf{X}_\text{train}^\dagger)\) contains the parameters of the linear regression between the training data and the training targets, applied to the input data \(\mathbf{x}\). \[\label{eq:linear} \hat{d} = \text{argmax}\left(\mathbf{T}_\text{train}\mathbf{X}_{\text{train}}^\dagger\mathbf{x}\right)\tag{14}\] This is however not a case of overfitting. Indeed, as the parameters are learned by linear regression over the whole training set, overfitting should not be observed as long as \(N_\theta\) is lower than the number of training samples i.e., 60000.
In all the nonlinear cases (STVO, reLU, and sigmoid), the performance improve steadily (accuracy increases and NRMSE decreases) with the number of parameters \(N_\theta\). Eventually at higher \(N_\theta\) values, the accuracy reaches a value common to all the activation functions, well above the saturation threshold reached with the identity activation, and similarly with the NRMSE. The origin of this convergence becomes clear by looking at the effective nonlinearity implemented by the STVO dynamics. Figure 3a shows the steady-state reduced position \(s_\text{f}(J)\) over the operating current window. It can be seen that the STVO linearity acts as a smooth threshold-and-saturation function. If normalized, its shape qualitatively interpolates between the rectifying reLU and the saturating sigmoid (Fig. 3b), which explains why the three functions yield comparable random feature maps. More fundamentally, a property of extreme learning machines is that any fixed nonlinearity provides a feature map rich enough for the linear readout as soon as \(N_\theta\) is large enough to expand rather than compress the input, so the precise form of the activation does not matter anymore. The presence of a large performance gap with respect to the identity activation for \(N_\theta > 44\) confirms that some form of nonlinearity is however necessary.
The STVO dynamics also exhibit a short-term memory due to the transient coupling between successive virtual nodes, that the reLU and sigmoid do not possess. For the image classification tasks considered in this work, this memory plays no role, and the three activations reach the same performance. For temporal tasks requiring fading memory like speech recognition or time series prediction, this equivalence would not hold anymore, and the STVO dynamics would reach higher performance than the reLU and the sigmoid.
The comparison of the accuracy of the classification using the STVO-based ELM for the three datasets is represented in Fig. 5. It can be seen that for a very low number of virtual nodes in the reservoir, the accuracy reaches the random choice accuracy level i.e., \(10\%\) for MNIST and FMNIST, and \(3.85\%\) for EMNIST-letters. The accuracy eventually reaches \(97.8\%\) on MNIST and \(87.5\%\) (resp. \(86.5\%\)) on the EMNIST-letters (resp. FMNIST) dataset with \(5000\) nodes in the nonlinear layer. The relatively low score achieved in FMNIST is due to the lack of convolutional layers in the architecture, something that is known to be of prior importance for the classification of complex images [44]. To test this hypothesis, we replaced the PCA preprocessing by a fixed random convolutional feature extractor (random filters followed by a reLU nonlinearity and max-pooling), reduced to the same dimensionality and fed to the identical ELM. Even without any training of the convolutional filters, this consistently improves the FMNIST test accuracy (e.g.from \(85.7\%\) to \(86.2\%\) at \(N_\theta = 1000\)), confirming that convolutional processing is beneficial. Trained convolutional layers would be expected to yield significantly larger gains. Concerning EMNIST-letters, we suspect that the lower quality of the recognition is due to the presence of lower case and upper case characters in each category, which can be regarded as the presence of two distinct datasets with the same labels. This assumption is consolidated by computing the average intraclass variance in each dataset. The variance of each class \(\mathbf{K}\) is computed accordingly to \(\text{Var}_\text{intra}(\mathbf{K}) = \left(\sum_{i}(\mathbf{x}_i-\overline{\mathbf{K}})^2\right)/\vert\mathbf{K}\vert\) and averaged over all the classes for each dataset. It is equal to \(3401\) in FMNIST, \(3452.9\) in MNIST, and \(4408.7\) in EMNIST-letters, which is about \(28\%\) larger than in the two former datasets.
The top accuracy reached on MNIST known to this day is \(99.87\%\) [45]. Our solution reaches an accuracy (\(97.8\%\)) competitive with other reservoir computing-based approaches in software and hardware [46]–[48], while relying on a much simpler architecture. Other works indicate that higher performance could be achieved by increasing further the number of nodes in the reservoir [49]. The top accuracy on EMNIST-letters and FMNIST are respectively \(95.96\%\) and \(96.91\%\) [50], [51]. However, these works are based on neural networks far more complex than ours, involving many specialized layers such as convolutional layers and attention modules, which strongly hinders their potential for hardware implementation and energy-efficient inference.
We designed an extreme learning machine based on non-linear spintronics oscillators called STVOs. This approach aims at solving cognitive tasks more energy-efficiently than conventional software solutions by leveraging the low power consumption of STVOs as nonlinear data processors. The system has a simple architecture and only requires the learning of the readout weights through linear regression, while the rest of the parameters is random and fixed. Moreover, the use of a single STVO delayed in time allows to further reduce the hardware complexity of the system, while still achieving state-of-the-art performance on the MNIST dataset.
To overcome the experimental load inherent to the optimization and testing of such a system, we used simulations of STVOs, allowing to obtain an explicit expression for the STVO dynamics that we used as neuronal nonlinearity. This approach allowed to simulate an entire STVO-based ELM several orders of magnitude faster than previously available frameworks. As DD-TEA has been quantitatively validated against micromagnetic simulations, the reference numerical method for STVO dynamics [24], [32], the present results rest on a physically grounded model. Experimental validation on a physical device remains an essential next step for assessing the influence of device and measurement noise on the distribution of the STVO responses. The simulation framework introduced here is precisely intended to prepare such experiments by narrowing the hyperparameter space beforehand.
Because the images are processed as static inputs, the system operates as an ELM rather than as a memory-dependent reservoir. Therefore, the temporal multiplexing through a single STVO is used here as a hardware-efficient way to emulate a large fixed nonlinear feature map, not to exploit fading memory. While other approaches treat each image as a genuine temporal signal [52] or rely on pretrained convolutional features within an ESN [53], our static solution with PCA trades such temporal encoding and feature-extraction for simplicity. In a physical realization, time-multiplexing \(N_\theta\) virtual nodes requires injecting and sampling \(N_\theta\) successive values through the single oscillator, i.e. storing the masked sequence and spending a real time of \(N_\theta \times D_\text{t}\) per inference. For \(N_\theta = 5000\) and \(D_\text{t} = 10\) ns this results in \(50\) μs per image (without considering reset time and read-out overheads). This constraint of single-node time-multiplexed hardware could be relaxed by spatially multiplexing several STVOs at the cost of additional chip area and increased complexity.
We showed that a STVO-based ELM is able to classify images of handwritten digits from the MNIST dataset with state-of-the-art accuracy levels. The nonlinearity of STVO dynamics was found to be equivalent to more conventional nonlinear function such as the reLU and the sigmoid. Moreover, the approach is easily generalizable to other datasets like the EMNIST-letters and the fashion MNIST datasets. Better performance can likely be obtained on the two latter tasks by designing a more specialized architecture, for example involving convolutional data treatment before the reservoir [54].
The use of the DD-TEA framework allowed to carry on extensive studies of the performance with respect to the number of reservoir nodes. This could be reproduced with other hyperparameters such as the input and output shapes of the data or the properties of the input matrix \(\mathbf{M}\). Physical operating parameters having an influence on the STVO dynamics could also be optimized the same way, allowing an efficient preparation of an experimental setup.
Spin-torque oscillators have also been used for addressing temporal tasks such as spoken-digit recognition and chaotic time-series prediction [16], [26], [28], [55]. Extending the present DD-TEA framework to recurrent single-STVO reservoirs, in which the fading memory of the magnetization dynamics becomes a computational resource, is a natural next step. More generally, we think that the DD-TEA framework could allow to develop and test other neural architectures based on STVOs to perform machine learning tasks with increased efficiency. In parallel, increasing attention has been given to the matrix-vector multiplication step that allows to multiply the neurons ouputs by the weights of the following layer. Recent works have demonstrated analog implementation of the so-called multiply-and-accumulate (MAC) operation using CMOS technology [56], and others using the magnetic or RF properties of MTJs similar to the ones considered in this work [57]–[59]. This opens the path to the realization of hardware neuromorphic systems using MTJs for both nonlinear transformation and signal propagation though the layers, and hence to more energy-efficient machine learning solutions.
The data-driven STVO model, the extreme learning machine pipeline, and the scripts used to produce all the figures and reported metrics are available from the authors upon reasonable request. The MNIST [33], EMNIST-letters [34] and Fashion MNIST [35] datasets are publicly available.
Computational resources have been provided by the Consortium des Équipements de Calcul Intensif (CÉCI), funded by the Fonds de la Recherche Scientifique de Belgique (F.R.S.-FNRS) under Grant No. 2.5020.11 and by the Walloon Region. F.A.A. is a Research Associate of the F.R.S.-FNRS.