Bayesian Inference on Binary Spiking Networks Leveraging Nanoscale Device Stochasticity
Fuente:
arXiv
Gespeichert in:
| Hauptverfasser: | , , , , |
|---|---|
| Format: | Preprint |
| Veröffentlicht: |
2023
|
| Schlagworte: | |
| Online-Zugang: | |
| Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
| _version_ | 1866929225472147456 |
|---|---|
| author | Katti, Prabodh Skatchkovsky, Nicolas Simeone, Osvaldo Rajendran, Bipin Al-Hashimi, Bashir M. |
| author_facet | Katti, Prabodh Skatchkovsky, Nicolas Simeone, Osvaldo Rajendran, Bipin Al-Hashimi, Bashir M. |
| contents | Bayesian Neural Networks (BNNs) can overcome the problem of overconfidence that plagues traditional frequentist deep neural networks, and are hence considered to be a key enabler for reliable AI systems. However, conventional hardware realizations of BNNs are resource intensive, requiring the implementation of random number generators for synaptic sampling. Owing to their inherent stochasticity during programming and read operations, nanoscale memristive devices can be directly leveraged for sampling, without the need for additional hardware resources. In this paper, we introduce a novel Phase Change Memory (PCM)-based hardware implementation for BNNs with binary synapses. The proposed architecture consists of separate weight and noise planes, in which PCM cells are configured and operated to represent the nominal values of weights and to generate the required noise for sampling, respectively. Using experimentally observed PCM noise characteristics, for the exemplary Breast Cancer Dataset classification problem, we obtain hardware accuracy and expected calibration error matching that of an 8-bit fixed-point (FxP8) implementation, with projected savings of over 9$\times$ in terms of core area transistor count. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2302_01302 |
| institution | arXiv |
| publishDate | 2023 |
| record_format | arxiv |
| spellingShingle | Bayesian Inference on Binary Spiking Networks Leveraging Nanoscale Device Stochasticity Katti, Prabodh Skatchkovsky, Nicolas Simeone, Osvaldo Rajendran, Bipin Al-Hashimi, Bashir M. Neural and Evolutionary Computing Hardware Architecture Emerging Technologies Machine Learning Bayesian Neural Networks (BNNs) can overcome the problem of overconfidence that plagues traditional frequentist deep neural networks, and are hence considered to be a key enabler for reliable AI systems. However, conventional hardware realizations of BNNs are resource intensive, requiring the implementation of random number generators for synaptic sampling. Owing to their inherent stochasticity during programming and read operations, nanoscale memristive devices can be directly leveraged for sampling, without the need for additional hardware resources. In this paper, we introduce a novel Phase Change Memory (PCM)-based hardware implementation for BNNs with binary synapses. The proposed architecture consists of separate weight and noise planes, in which PCM cells are configured and operated to represent the nominal values of weights and to generate the required noise for sampling, respectively. Using experimentally observed PCM noise characteristics, for the exemplary Breast Cancer Dataset classification problem, we obtain hardware accuracy and expected calibration error matching that of an 8-bit fixed-point (FxP8) implementation, with projected savings of over 9$\times$ in terms of core area transistor count. |
| title | Bayesian Inference on Binary Spiking Networks Leveraging Nanoscale Device Stochasticity |
| topic | Neural and Evolutionary Computing Hardware Architecture Emerging Technologies Machine Learning |
| url | https://arxiv.org/abs/2302.01302 |