Gradient-based optimization of spintronic devices

Abstract

The optimization of physical parameters serves various purposes, such as system identification and efficiency in developing devices. Spin-torque oscillators have been applied to neuromorphic computing experimentally and theoretically, but the optimization of their physical parameters has usually been done by grid search. In this paper, we propose a scheme to optimize the parameters of the dynamics of macrospin-type spin-torque oscillators using the gradient descent method with automatic differentiation. First, we prepared numerically created dynamics as teacher data and successfully tuned the parameters to reproduce the dynamics. This can be applied to obtain the correspondence between the simulation and experiment of the spin-torque oscillators. Next, we successfully solved the image recognition task with high accuracy by connecting the coupled system of spin-torque oscillators to the input and output layers and training all of them through gradient descent. This approach allowed us to estimate how to control the experimental setup and design the physical systems so that the task could be solved with a high accuracy using spin-torque oscillators.