Current browse context:
cs.AR
Change to browse by:
References & Citations
Computer Science > Hardware Architecture
Title: High-Performance FPGA-based Accelerator for Bayesian Neural Networks
(Submitted on 12 May 2021 (v1), last revised 30 Nov 2021 (this version, v3))
Abstract: Neural networks (NNs) have demonstrated their potential in a wide range of applications such as image recognition, decision making or recommendation systems. However, standard NNs are unable to capture their model uncertainty which is crucial for many safety-critical applications including healthcare and autonomous vehicles. In comparison, Bayesian neural networks (BNNs) are able to express uncertainty in their prediction via a mathematical grounding. Nevertheless, BNNs have not been as widely used in industrial practice, mainly because of their expensive computational cost and limited hardware performance. This work proposes a novel FPGA-based hardware architecture to accelerate BNNs inferred through Monte Carlo Dropout. Compared with other state-of-the-art BNN accelerators, the proposed accelerator can achieve up to 4 times higher energy efficiency and 9 times better compute efficiency. Considering partial Bayesian inference, an automatic framework is proposed, which explores the trade-off between hardware and algorithmic performance. Extensive experiments are conducted to demonstrate that our proposed framework can effectively find the optimal points in the design space.
Submission history
From: Hongxiang Fan [view email][v1] Wed, 12 May 2021 06:20:44 GMT (334kb,D)
[v2] Thu, 10 Jun 2021 20:01:12 GMT (334kb,D)
[v3] Tue, 30 Nov 2021 21:18:15 GMT (335kb,D)
Link back to: arXiv, form interface, contact.