Bayesian Neural Network Realization by Exploiting Random Noise and Mitigating the Impact of Conductance Drift of PCM
Abstract Bayesian neural networks (BNNs) provide inference with uncertainty prediction, which is critical for safety‐sensitive applications such as autonomous driving and medical diagnosis. However, conventional implementations relying on Gaussian random number generators incur significant area and power overheads. In this work, the inherent random noise of phase change memory (PCM) devices are exploited to realize a compact, energy‐efficient BNN hardware architecture. The fabricated 4‐Mb indium‐doped Ge 2 Sb 2 Te 5 (In ‐ GST) PCM chip achieves a resistance ratio of over 190 times and can be programmed to 32 different conductance states. To mitigate the impact of conductance drift on inference accuracy, two compensation methods are proposed, with the measured minus fit method demonstrating superior reduction of epistemic uncertainty. The proposed PCM‐based BNN architecture achieves a MNIST recognition accuracy of 98.08% on LeNet‐5 and demonstrates an energy efficiency of 33.3 TOPS/W. This work establishes a unified memory‐computation‐randomness framework for probabilistic neural network hardware, enabling low‐power and reliable inference with uncertainty quantification.
Authors
- Zhuojun Chen (ORCID: https://orcid.org/0000-0003-0431-8852)
- Houpeng Chen (ORCID: https://orcid.org/0000-0002-5766-5746)
- Qian Wang (ORCID: https://orcid.org/0000-0001-9549-3140)
- Yu Lei (ORCID: https://orcid.org/0000-0003-4321-0385)
- Li Xie (ORCID: https://orcid.org/0000-0002-3303-312X)
- Yonghui Zheng
- Zouhao Song (ORCID: https://orcid.org/0009-0004-2258-3515)
- Jifan Tan (ORCID: https://orcid.org/0000-0002-5014-5393)
- Xinyu Yang (ORCID: https://orcid.org/0009-0009-1758-5221)
Institutions
- Hunan University (CN)
- Chinese Academy of Sciences (CN)
- Changsha University (CN)
- Shanghai Institute of Microsystem and Information Technology (CN)
- University of Chinese Academy of Sciences (CN)
- East China Normal University (CN)
Publication Details
- Journal
- Advanced Materials Technologies
- Published
- 2025-11-08
- DOI
- https://doi.org/10.1002/admt.202501278
- Primary Topic
- Advanced Memory and Neural Computing
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China
- Chinese Academy of Sciences
- Youth Innovation Promotion Association of the Chinese Academy of Sciences
- Shanghai Rising-Star Program
- National Key Research and Development Program of China
- Youth Innovation Promotion Association