中文

图扩散方程的快速局部求解器

机器学习 2024-12-24 v2 社会与信息网络

摘要

高效计算图扩散方程 (GDEs),如个性化 PageRank、Katz 中心性和热核,对于聚类、训练神经网络以及许多其他图相关问题至关重要。标准迭代方法每次迭代都需要访问整个图,这使得它们对于大规模图来说非常耗时。虽然现有的局部求解器通过启发式局部更新来近似扩散向量,但它们通常是串行操作的,并且是为特定扩散类型设计的,从而限制了它们的适用性。鉴于扩散向量具有高度可局部化的特性(通过参与率测量),本文引入了一种使用局部扩散过程近似求解 GDEs 的新框架。该框架揭示了现有局部求解器的次优性。此外,我们的方法通过设计简单且可证明的次线性时间算法,有效地将标准迭代求解器局部化。这些新的局部求解器高度可并行化,使其非常适合在 GPU 上实现。我们展示了我们的框架在快速获取近似扩散向量方面的有效性,实现了高达百倍的速度提升,并且适用于大规模动态图。我们的框架还可以促进 GNN 更高效的局部消息传递机制。

关键词

引用

@article{arxiv.2410.21634,
  title  = {Faster Local Solvers for Graph Diffusion Equations},
  author = {Jiahe Bai and Baojian Zhou and Deqing Yang and Yanghua Xiao},
  journal= {arXiv preprint arXiv:2410.21634},
  year   = {2024}
}

备注

NeurIPS 2024