arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2203.02471cs.LGcs.DMmath.OCstat.COstat.ML
收藏

基于玻尔兹曼机的图聚类

Graph clustering with Boltzmann machines

University of Toronto(多伦多大学)

Pierre Miasnikof, Mohammad Bagherbeik, Ali Sheikholeslami

AI总结:

本文提出基于玻尔兹曼机的图聚类方法,通过定制两种数学规划公式近似求解簇内密度最大化问题,实验表明其在解质量和计算效率上优于Gurobi精确求解器及Louvain算法。

摘要:

图聚类是将顶点分组为密集连接集合(称为簇)的过程。我们从文献中定制了两个数学规划公式用于该问题。通过这样做,我们获得了对簇内密度最大化问题的启发式近似。我们使用玻尔兹曼机启发式的两种变体来获得数值解。为了进行基准测试,我们将解质量和计算性能与使用商业求解器 Gurobi 获得的结果进行了比较。我们还将聚类质量与使用流行的 Louvain 模块度最大化方法获得的簇进行了比较。我们的初步结果清楚地证明了我们问题公式的优越性。它们还确立了玻尔兹曼机相对于传统精确求解器的优越性。在较小且不太复杂的图的情况下,玻尔兹曼机提供与 Gurobi 相同的解,但求解时间低几个数量级。在更大且更复杂的图的情况下,Gurobi 未能在合理的时间范围内返回有意义的结果。最后,我们还注意到,我们的两种聚类公式,即距离最小化和 $K$-medoids,都产生了优于 Louvain 算法所获得簇的质量。

英文摘要:

Graph clustering is the process of grouping vertices into densely connected sets called clusters. We tailor two mathematical programming formulations from the literature, to this problem. In doing so, we obtain a heuristic approximation to the intra-cluster density maximization problem. We use two variations of a Boltzmann machine heuristic to obtain numerical solutions. For benchmarking purposes, we compare solution quality and computational performances to those obtained using a commercial solver, Gurobi. We also compare clustering quality to the clusters obtained using the popular Louvain modularity maximization method. Our initial results clearly demonstrate the superiority of our problem formulations. They also establish the superiority of the Boltzmann machine over the traditional exact solver. In the case of smaller less complex graphs, Boltzmann machines provide the same solutions as Gurobi, but with solution times that are orders of magnitude lower. In the case of larger and more complex graphs, Gurobi fails to return meaningful results within a reasonable time frame. Finally, we also note that both our clustering formulations, the distance minimization and $K$-medoids, yield clusters of superior quality to those obtained with the Louvain algorithm.

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

↑