arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.03085cs.LGcs.AIstat.ML
收藏

Graph Mixture Density Networks

University of Pisa(比萨大学)

Federico Errica, Davide Bacciu, Alessio Micheli

英文摘要:

We introduce the Graph Mixture Density Networks, a new family of machine learning models that can fit multimodal output distributions conditioned on graphs of arbitrary topology. By combining ideas from mixture models and graph representation learning, we address a broader class of challenging conditional density estimation problems that rely on structured data. In this respect, we evaluate our method on a new benchmark application that leverages random graphs for stochastic epidemic simulations. We show a significant improvement in the likelihood of epidemic outcomes when taking into account both multimodality and structure. The empirical analysis is complemented by two real-world regression tasks showing the effectiveness of our approach in modeling the output prediction uncertainty. Graph Mixture Density Networks open appealing research opportunities in the study of structure-dependent phenomena that exhibit non-trivial conditional output distributions.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑