English

Probabilistic Memory for Trustworthy Edge Intelligence

Hardware Architecture 2026-07-02 v1

Abstract

Probabilistic computation plays an important role in trustworthy edge intelligence to quantify uncertainty, enhance robustness, reconstruct data, and protect privacy, but its adoption is limited by the orders-of-magnitude data throughput gap between Gaussian random number generation (GRNG) and computation, as well as instruction overhead. This paper introduces probabilistic memory (p-MEM), a unified memory primitive that stores distribution parameters, such as mean and standard deviation, and samples directly at the native memory bandwidth, where deterministic data becomes the zero-variance special case. Using a layout-validated p-MEM simulator, we comprehensively explore device choices, memory specifications, and technology nodes, showing that p-MEM can achieve more than 1000 GSa/s/mm^2 GRNG throughput, including memory-array access. Integrated into CPU/GPU systems, p-MEM reduces instruction count by up to 2.19x/4.37x, sampling latency by 562x/3.45x, and energy by 295.5x/3.53x for Bayesian neural network workloads, providing a scalable hardware substrate for trustworthy probabilistic AI.

Cite

@article{arxiv.2607.02465,
  title  = {Probabilistic Memory for Trustworthy Edge Intelligence},
  author = {Likai Pei and Jiahao Zheng and Xueji Zhao and Emilie Ye and Jianbo Liu and Hanqing Tao and Ming-Yen Lee and Ruiyang Qin and Yiyu Shi and Shimeng Yu and X. Sharon Hu and Ningyuan Cao},
  journal= {arXiv preprint arXiv:2607.02465},
  year   = {2026}
}

Comments

This paper has been accepted for publication in the proceedings of the ACM/IEEE Design Automation Conference (DAC), 2026