The paper has been published in the January 2021 edition of Nature Electronics and is titled: 'In-situ learning using intrinsic memristor variability via Markov chain Monte Carlo sampling.'
Resistive random access memories (RRAMs), sometimes referred to as memristors, can be used for machine learning by using Kirchoff's current law to implement the dot-product or multiply-accumulate operation used in analog neural networks.
However, performing training or learning at edge is rarely done because of the energy cost of using von Neumann computing or – if using in-memory machine learning – the precision and accuracy required by back propagation algorithms. This has been thwarted by such as nonlinear conductance modulation, lack of multi-level conductance and device variability.
To get around that problem, the team developed a method that actively exploits the randomness, implementing a Markov Chain Monte Carlo (MCMC) sampling learning algorithm in a fabricated chip that acts as a Bayesian machine-learning model.
Bayesian probability is a refinement of conventional probability theory which assigns a probability to a hypothesis, which can be updated in the light of subsequent results and is therefore particularly relevant to machine learning.
In Leti's example, the in-situ learning is realized using nanosecond pulses and compared to a CMOS logic implementation requires five orders of magnitude less energy. As a result, this approach is capable of bringing learning to edge-computing systems.
As a test of learning at the edge, the researchers experimentally applied RRAM-based MCMC to train a multilayer Bayesian neural network to detect heart arrhythmias from electrocardiogram recordings – reporting a better detection rate than a standard neural network based on a von Neumann computing system.
"This highlights that, beyond being RRAM-compatible, Bayesian machine learning offers an alternative modelling method that appears well suited to the characteristics of edge learning," the article states. The team also applied their experimental system to solve further classification tasks including the diagnosis of malignant breast-tissue samples.
"Our system could be used as the foundation for the design and fabrication of a standalone and fully integrated RRAM-based MCMC sampling chip, for applications outside the laboratory," the article concludes.
Related links and articles: