Revisiting Random Walks for Learning on Graphs

التفاصيل البيبلوغرافية
العنوان: Revisiting Random Walks for Learning on Graphs
المؤلفون: Kim, Jinwoo, Zaghen, Olga, Suleymanzade, Ayhan, Ryou, Youngmin, Hong, Seunghoon
سنة النشر: 2024
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Machine Learning, Computer Science - Artificial Intelligence
الوصف: We revisit a simple idea for machine learning on graphs, where a random walk on a graph produces a machine-readable record, and this record is processed by a deep neural network to directly make vertex-level or graph-level predictions. We refer to these stochastic machines as random walk neural networks, and show that we can design them to be isomorphism invariant while capable of universal approximation of graph functions in probability. A useful finding is that almost any kind of record of random walk guarantees probabilistic invariance as long as the vertices are anonymized. This enables us to record random walks in plain text and adopt a language model to read these text records to solve graph tasks. We further establish a parallelism to message passing neural networks using tools from Markov chain theory, and show that over-smoothing in message passing is alleviated by construction in random walk neural networks, while over-squashing manifests as probabilistic under-reaching. We show that random walk neural networks based on pre-trained language models can solve several hard problems on graphs, such as separating strongly regular graphs where the 3-WL test fails, counting substructures, and transductive classification on arXiv citation network without training. Code is available at https://github.com/jw9730/random-walk.
Comment: 41 pages, 11 figures
نوع الوثيقة: Working Paper
URL الوصول: http://arxiv.org/abs/2407.01214
رقم الأكسشن: edsarx.2407.01214
قاعدة البيانات: arXiv