arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2307.00530cs.DScs.DC
收藏

随机块模型的大规模并行算法

Massively Parallel Algorithms for the Stochastic Block Model

Zelin Li, Pan Peng, Xianbin Zhu

AI总结:

本文研究MPC模型下SBM的精确社区恢复,提出基于r步邻域统计比较的并行算法,在更一般的低内存s-空间模型中改进轮数、空间及参数条件。

摘要:

学习大规模图的社区结构是机器学习、计算机科学和统计学中的一个基本问题。我们在大规模并行计算(MPC)模型中研究精确恢复由随机块模型(SBM)生成的图中社区的问题。具体而言,给定被划分为 k 个等大小簇(即每个簇大小为 n)的 kn 个顶点,在这些 kn 个顶点上随机生成一张图:若两个顶点位于同一簇中,则它们以概率 p 相连;若不在同一簇中,则以概率 q 相连,其中 p > q > 0。我们在(非常一般的)s-空间 MPC 模型中给出了用于 SBM 的 MPC 算法,其中每台机器具有内存 s=Ω(log n)。在对于任意整数 r∈[3,O(log n)] 满足 (p-q)/√p ≥ \u001bΩ(k^{1/2}n^{-1/2+1/(2(r-1))}) 的条件下,我们的第一个算法使用 \u001bO(m) 总空间在 O(kr log_s n) 轮内精确恢复所有 k 个簇,或者使用 \u001bO(km) 总空间在 O(r log_s n) 轮内精确恢复所有 k 个簇。如果 (p-q)/√p ≥ \u001bΩ(k^{3/4}n^{-1/4}),我们的第二个算法实现 O(log_s n) 轮和 \u001bO(m) 总空间复杂度。这两个算法都显著改进了 Cohen-Addad 等人最近的结果 [PODC'22],他们给出的算法只适用于亚线性空间 MPC 模型,其中每台机器对于某个常数 δ>0 具有本地内存 s=O(n^δ),并且对 p、q、k 有强得多的条件。我们的算法基于收集每个顶点的 r 步邻域,并比较每对顶点由局部邻域生成的某些统计信息的差异。为了并行实现聚类算法,我们提出了在 s-空间 MPC 模型中实现一些基本图操作的高效方法。

英文摘要:

Learning the community structure of a large-scale graph is a fundamental problem in machine learning, computer science and statistics. We study the problem of exactly recovering the communities in a graph generated from the Stochastic Block Model (SBM) in the Massively Parallel Computation (MPC) model. Specifically, given $kn$ vertices that are partitioned into $k$ equal-sized clusters (i.e., each has size $n$), a graph on these $kn$ vertices is randomly generated such that each pair of vertices is connected with probability~$p$ if they are in the same cluster and with probability $q$ if not, where $p > q > 0$. We give MPC algorithms for the SBM in the (very general) \emph{$s$-space MPC model}, where each machine has memory $s=Ω(\log n)$. Under the condition that $\frac{p-q}{\sqrt{p}}\geq \tildeΩ(k^{\frac12}n^{-\frac12+\frac{1}{2(r-1)}})$ for any integer $r\in [3,O(\log n)]$, our first algorithm exactly recovers all the $k$ clusters in $O(kr\log_s n)$ rounds using $\tilde{O}(m)$ total space, or in $O(r\log_s n)$ rounds using $\tilde{O}(km)$ total space. If $\frac{p-q}{\sqrt{p}}\geq \tildeΩ(k^{\frac34}n^{-\frac14})$, our second algorithm achieves $O(\log_s n)$ rounds and $\tilde{O}(m)$ total space complexity. Both algorithms significantly improve upon a recent result of Cohen-Addad et al. [PODC'22], who gave algorithms that only work in the \emph{sublinear space MPC model}, where each machine has local memory~$s=O(n^δ)$ for some constant $δ>0$, with a much stronger condition on $p,q,k$. Our algorithms are based on collecting the $r$-step neighborhood of each vertex and comparing the difference of some statistical information generated from the local neighborhoods for each pair of vertices. To implement the clustering algorithms in parallel, we present efficient approaches for implementing some basic graph operations in the $s$-space MPC model.

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

↑