Natural-Parameter Networks: A Class of Probabilistic Neural Networks
Table 5: AUC on Three Datasets
['Method', 'SAE', 'SDAE', 'VAE', 'gamma NPN', 'Gaussian NPN', 'Poisson NPN']
In this section, we show the AUC for different models on the link prediction task. NPN is able to achieve much higher AUC than SAE, SDAE, and VAE. Among different variants of NPN, the Gaussian NPN seems to perform better in datasets with fewer words like Citeulike-t (18.8 words per document). The Poisson NPN, as a more natural choice to model text, achieves the best performance in datasets with more words (Citeulike-a with 66.6 words per document and arXiv with 88.8 words per document).