Deep Model Compression and Inference Speedup of Sum-Product Networks on Tensor Trains.

Ko, Ching-Yun; Chen, Cong; He, Zhuolun; Zhang, Yuke; Batselier, Kim; Wong, Ngai · IEEE Trans Neural Netw Learn Syst · 2020

basic_science · Level V

Where this comes from

Abstract

Sum-product networks (SPNs) constitute an emerging class of neural networks with clear probabilistic semantics and superior inference speed over other graphical models. This brief reveals an important connection between SPNs and tensor trains (TTs), leading to a new canonical form which we call tensor SPNs (tSPNs). Specifically, we demonstrate the intimate relationship between a valid SPN and a TT. For the first time, through mapping an SPN onto a tSPN and employing specially customized optimization techniques, we demonstrate improvements up to a factor of 100 on both model compression and inference speedup for various data sets with negligible loss in accuracy.