Convolutional neural networks with radio-frequency spintronic nano-devices

Convolutional neural networks are state-of-the-art and ubiquitous in modern signal processing and machine vision. Nowadays, hardware solutions based on emerging nanodevices are designed to reduce the power consumption of these networks. Spintronics devices are promising for information processing because of the various neural and synaptic functionalities they offer. However, due to their low OFF/ON ratio, performing all the multiplications required for convolutions in a single step with a crossbar array of spintronic memories would cause sneak-path currents. Here we present an architecture where synaptic communications have a frequency selectivity that prevents crosstalk caused by sneak-path currents. We first demonstrate how a chain of spintronic resonators can function as synapses and make convolutions by sequentially rectifying radio-frequency signals encoding consecutive sets of inputs. We show that a parallel implementation is possible with multiple chains of spintronic resonators to avoid storing intermediate computational steps in memory. We propose two different spatial arrangements for these chains. For each of them, we explain how to tune many artificial synapses simultaneously, exploiting the synaptic weight sharing specific to convolutions. We show how information can be transmitted between convolutional layers by using spintronic oscillators as artificial microwave neurons. Finally, we simulate a network of these radio-frequency resonators and spintronic oscillators to solve the MNIST handwritten digits dataset, and obtain results comparable to software convolutional neural networks. Since it can run convolutional neural networks fully in parallel in a single step with nano devices, the architecture proposed in this paper is promising for embedded applications requiring machine vision, such as autonomous driving.

[1]  A. Boes,et al.  11 TOPS photonic convolutional accelerator for optical neural networks , 2021, Nature.

[2]  Juan Trastoy,et al.  Detection of the Microwave Emission from a Spin-Torque Oscillator by a Spin Diode , 2020 .

[3]  Yi Li,et al.  Recent Progress on Memristive Convolutional Neural Networks for Edge Intelligence , 2020, Adv. Intell. Syst..

[4]  J. Yang,et al.  Memristive crossbar arrays for brain-inspired computing , 2019, Nature Materials.

[5]  Xiaohui Liu,et al.  Hardware Implementation of Energy Efficient Deep Learning Neural Network Based on Nanoscale Flash Computing Array , 2019, Advanced Materials Technologies.

[6]  Tadashi Shibata,et al.  Coupled-Oscillator Associative Memory Array Operation for Pattern Recognition , 2015, IEEE Journal on Exploratory Solid-State Computational Devices and Circuits.

[7]  Damien Querlioz,et al.  Neuromorphic computing with nanoscale spintronic oscillators , 2017, Nature.

[8]  Damien Querlioz,et al.  Physics for neuromorphic computing , 2020, Nature Reviews Physics.

[9]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[10]  S. Yuasa,et al.  Spin-torque diode effect in magnetic tunnel junctions , 2005, Nature.

[11]  Vincent Gripon,et al.  A Nonvolatile Associative Memory-Based Context-Driven Search Engine Using 90 nm CMOS/MTJ-Hybrid Logic-in-Memory Architecture , 2014, IEEE Journal on Emerging and Selected Topics in Circuits and Systems.

[12]  Timo Aila,et al.  A Style-Based Generator Architecture for Generative Adversarial Networks , 2018, 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[13]  Vincent Cros,et al.  Field dependence of spin-transfer-induced vortex dynamics in the nonlinear regime , 2012 .

[14]  Baoshun Zhang,et al.  Ultralow-current-density and bias-field-free spin-transfer nano-oscillator , 2013, Scientific Reports.

[15]  Danna Zhou,et al.  d. , 1840, Microbial pathogenesis.

[16]  Mohammad Zahedinejad,et al.  Memristive control of mutual spin Hall nano-oscillator synchronization for neuromorphic computing , 2020, Nature Materials.

[17]  Big data needs a hardware revolution , 2018, Nature.

[18]  P. Bortolotti,et al.  Analog and Digital Phase Modulation and Signal Transmission with Spin-Torque Nano-Oscillators , 2019, Physical Review Applied.

[19]  Sachhidh Kannan,et al.  Sneak-Path Testing of Crossbar-Based Nonvolatile Random Access Memories , 2013, IEEE Transactions on Nanotechnology.

[20]  Baoshun Zhang,et al.  Giant spin-torque diode sensitivity in the absence of bias magnetic field , 2016, Nature communications.

[21]  J. C. Sankey,et al.  Spin-torque ferromagnetic resonance measurements of damping in nanomagnets , 2007 .

[22]  Kevin P. Murphy,et al.  Machine learning - a probabilistic perspective , 2012, Adaptive computation and machine learning series.

[23]  Wei-gang Wang,et al.  Electric-field-assisted switching in magnetic tunnel junctions. , 2012, Nature materials.

[24]  Ran El-Yaniv,et al.  Binarized Neural Networks , 2016, ArXiv.

[25]  P. Freitas,et al.  High power and low critical current density spin transfer torque nano-oscillators using MgO barriers with intermediate thickness , 2017, Scientific Reports.

[26]  Shihui Yin,et al.  Neuromorphic Hardware Accelerator for SNN Inference based on STT-RAM Crossbar Arrays , 2019, 2019 26th IEEE International Conference on Electronics, Circuits and Systems (ICECS).

[27]  V. Cros,et al.  Electrical measurement of magnetic-field-impeded polarity switching of a ferromagnetic vortex core , 2016, 1605.01826.

[28]  Meng-Fan Chang,et al.  Neuro-inspired computing chips , 2020, Nature Electronics.

[29]  Eitan Yaakobi,et al.  Information-Theoretic Sneak-Path Mitigation in Memristor Crossbar Arrays , 2016, IEEE Transactions on Information Theory.

[30]  Bernard Dieny,et al.  Ultra-fast sweep-tuned spectrum analyzer with temporal resolution based on a spin-torque nano-oscillator. , 2020, Nano letters.

[31]  Bin Gao,et al.  Fully hardware-implemented memristor convolutional neural network , 2020, Nature.

[32]  Hitoshi Kubota,et al.  Physical reservoir computing based on spin torque oscillator with forced synchronization , 2019, Applied Physics Letters.

[33]  Switching spin valves using rf currents , 2005, cond-mat/0511498.

[34]  Mohammad Zahedinejad,et al.  Reduced spin torque nano-oscillator linewidth using He + irradiation , 2020, Applied Physics Letters.

[35]  Gouhei Tanaka,et al.  2021 Roadmap on Neuromorphic Computing and Engineering , 2021, ArXiv.

[36]  Yansong Chua,et al.  HFNet: A CNN Architecture Co-designed for Neuromorphic Hardware With a Crossbar Array of Synapses , 2020, Frontiers in Neuroscience.

[37]  Xuan Li,et al.  Parallel convolutional processing using an integrated photonic tensor core , 2021, Nature.

[38]  John M. Acken,et al.  Sneak Path Characterization in Memristor Crossbar Circuits , 2020 .

[39]  P. Bortolotti,et al.  Reservoir computing with the frequency, phase, and amplitude of spin-torque nano-oscillators , 2018, Applied Physics Letters.

[40]  P. Alam ‘G’ , 2021, Composites Engineering: An A–Z Guide.

[41]  S. Urazhdin,et al.  Controlled nonlinear magnetic damping in spin-Hall nano-devices , 2019, Nature Communications.

[42]  Damien Querlioz,et al.  Synaptic metaplasticity in binarized neural networks , 2020, Nature Communications.

[43]  Chris Yakopcic,et al.  Memristor crossbar deep network implementation based on a Convolutional neural network , 2016, 2016 International Joint Conference on Neural Networks (IJCNN).

[44]  Jacques-Olivier Klein,et al.  In-Memory and Error-Immune Differential RRAM Implementation of Binarized Deep Neural Networks , 2018, 2018 IEEE International Electron Devices Meeting (IEDM).

[45]  Eitan Yaakobi,et al.  Sneak-path constraints in memristor crossbar arrays , 2013, 2013 IEEE International Symposium on Information Theory.

[46]  V. Tiberkevich,et al.  Nonlinear Auto-Oscillator Theory of Microwave Generation by Spin-Polarized Current , 2009, IEEE Transactions on Magnetics.

[47]  M. Kostylev,et al.  Implementing a Magnonic Reservoir Computer Model Based on Time-Delay Multiplexing , 2021, Physical Review Applied.

[48]  Damien Querlioz,et al.  Training Dynamical Binary Neural Networks with Equilibrium Propagation , 2021, 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW).

[49]  Robert A. Buhrman,et al.  Time-resolved measurement of spin-transfer-driven ferromagnetic resonance and spin torque in magnetic tunnel junctions , 2011 .

[50]  Yoshua Bengio,et al.  BinaryConnect: Training Deep Neural Networks with binary weights during propagations , 2015, NIPS.

[51]  Guigang Zhang,et al.  Deep Learning , 2016, Int. J. Semantic Comput..

[52]  Damien Querlioz,et al.  Vowel recognition with four coupled spin-torque nano-oscillators , 2017, Nature.

[53]  Mahdi Jamali,et al.  Scaling effect of spin-torque nano-oscillators , 2017 .

[54]  Timo Aila,et al.  A Style-Based Generator Architecture for Generative Adversarial Networks , 2020, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[55]  P. Alam ‘W’ , 2021, Composites Engineering.

[56]  Andrew S. Cassidy,et al.  Convolutional networks for fast, energy-efficient neuromorphic computing , 2016, Proceedings of the National Academy of Sciences.

[57]  Wolfgang Porod,et al.  Distance Computation Based on Coupled Spin-Torque Oscillators: Application to Image Processing , 2020, Physical Review Applied.

[58]  Hitoshi Kubota,et al.  Controlling the chirality and polarity of vortices in magnetic tunnel junctions , 2014 .

[59]  P. Kurahashi,et al.  Convolution Inference via Synchronization of a Coupled CMOS Oscillator Array , 2019, IEEE Journal on Exploratory Solid-State Computational Devices and Circuits.

[60]  Guanda Wang,et al.  Time-Domain Computing in Memory Using Spintronics for Energy-Efficient Convolutional Neural Network , 2021, IEEE Transactions on Circuits and Systems I: Regular Papers.

[61]  J. Åkerman,et al.  Two-dimensional mutually synchronized spin Hall nano-oscillator arrays for neuromorphic computing , 2019, Nature Nanotechnology.

[62]  Yoshua Bengio,et al.  Gradient-based learning applied to document recognition , 1998, Proc. IEEE.

[63]  Chris Yakopcic,et al.  Extremely parallel memristor crossbar architecture for convolutional neural network implementation , 2017, 2017 International Joint Conference on Neural Networks (IJCNN).

[64]  A quantum material spintronic resonator , 2021, Scientific reports.

[65]  CNRS,et al.  Radio-Frequency Multiply-and-Accumulate Operations with Spintronic Synapses , 2020, Physical Review Applied.

[66]  Ryan Hamerly,et al.  Large-Scale Optical Neural Networks based on Photoelectric Multiplication , 2018, Physical Review X.

[67]  Juan Trastoy,et al.  Hardware realization of the multiply and accumulate operation on radio-frequency signals with magnetic tunnel junctions , 2021, Neuromorph. Comput. Eng..

[68]  S. Yuasa,et al.  Spin–torque dynamics for noise reduction in vortex-based sensors , 2020, Applied Physics Letters.

[69]  P. Freitas,et al.  Non-volatile artificial synapse based on a vortex nano-oscillator , 2021, Scientific Reports.

[70]  Resonant spin-transfer-driven switching of magnetic devices assisted by microwave current pulses , 2008, 0803.2880.

[71]  H. Kappen,et al.  An atomic Boltzmann machine capable of self-adaption , 2020, Nature Nanotechnology.

[72]  M. Stiles,et al.  Neuromorphic spintronics , 2020, Nature Electronics.