Deep Bayesian active learning using in-memory computing hardware

Yudeng Lin

doi:10.1038/s43588-024-00744-y

Deep Bayesian active learning using in-memory computing hardware

2024-12-23

0

OA

AI

DOI：10.1038/s43588-024-00744-y

Original

Save

View PDF

Abstract

En 中文

Labeling data is a time-consuming, labor-intensive and costly procedure for many artificial intelligence tasks. Deep Bayesian active learning (DBAL) boosts labeling efficiency exponentially, substantially reducing costs. However, DBAL demands high-bandwidth data transfer and probabilistic computing, posing great challenges for conventional deterministic hardware. Here we propose a memristor stochastic gradient Langevin dynamics in situ learning method that uses the stochastic of memristor modulation to learn efficiency, enabling DBAL within the computation-in-memory (CIM) framework. To prove the feasibility and effectiveness of the proposed method, we implemented in-memory DBAL on a memristor-based stochastic CIM system and successfully demonstrated a robot's skill learning task. The inherent stochastic characteristics of memristors allow a four-layer memristor Bayesian deep neural network to efficiently identify and learn from uncertain samples. Compared with cutting-edge conventional complementary metal-oxide-semiconductor-based hardware implementation, the stochastic CIM system achieves a remarkable 44% boost in speed and could conserve 153 times more energy.

Keywords:

ELECTROCATALYSTS

DISCOVERY

Deep Bayesian active learning using in-memory computing hardware

Abstract

AI Summary

Journal

Organization