Dynamic point stochastic rounding algorithm for limited precision arithmetic in Deep Belief Network training
Mohaned Essam, Tong Boon Tang, Eric Tatt Wei Ho, Hsin Chen · 2017
This paper reports how to train a Deep Belief Network (DBN) using only 8-bit fixed-point parameters. We propose a dynamic-point stochastic rounding algorithm that provides enhanced results compared to the existing stochastic rounding. We show that by using a variable scaling factor, the fixed-point parameter updates are enhanced. To be more hardware amenable, the use of common scaling factor at each layer of DBN is further proposed. Using publicly available MNIST database, we show that the proposed algorithm can train a 3-layer DBN with an average accuracy of 98.49%, with a drop of 0.08% from the double floating-point average accuracy.