arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2301.04771stat.MLcs.LG

变分推断:后验阈值提升稀疏场景下的网络聚类精度

Variational Inference: Posterior Threshold Improves Network Clustering Accuracy in Sparse Regimes

University of California, Davis(加利福尼亚大学戴维斯分校)

Xuezhen Li, Can M. Le

AI总结:

针对现有变分推断社区检测方法仅适用于稠密网络、稀疏网络下因鞍点影响性能的问题,提出迭代后对社区分配后验做硬阈值的改进方法,理论证明其在平均节点度有界时可准确恢复社区标签,数值实验验证其优于经典方法与现有SOTA算法。

摘要:

变分推断已在机器学习文献中被广泛用于拟合各类贝叶斯模型。在网络分析中,该方法已被成功应用于解决社区检测问题。尽管这些研究结果颇具前景,但它们的理论支撑仅适用于相对稠密的网络,而这一假设在真实网络中可能并不成立。此外,近期研究表明变分损失曲面存在许多鞍点,这可能会严重影响其性能,尤其是在应用于稀疏网络时。本文提出了一种改进变分推断方法的简单思路:在每次迭代后对社区分配的后验概率进行硬阈值处理。通过使用与真实社区分配相关的随机初始化,我们证明了所提方法能够收敛,且即使在网络平均节点度有界的情况下,也能准确恢复真实的社区标签。大量数值研究进一步证实了所提方法相较于经典变分推断和另一种最先进算法的优势。

英文摘要:

Variational inference has been widely used in machine learning literature to fit various Bayesian models. In network analysis, this method has been successfully applied to solve the community detection problems. Although these results are promising, their theoretical support is only for relatively dense networks, an assumption that may not hold for real networks. In addition, it has been shown recently that the variational loss surface has many saddle points, which may severely affect its performance, especially when applied to sparse networks. This paper proposes a simple way to improve the variational inference method by hard thresholding the posterior of the community assignment after each iteration. Using a random initialization that correlates with the true community assignment, we show that the proposed method converges and can accurately recover the true community labels, even when the average node degree of the network is bounded. Extensive numerical study further confirms the advantage of the proposed method over the classical variational inference and another state-of-the-art algorithm.

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

↑