Convergent Graph Solvers
Junyoung Park, Jinhyun Choo, Jinkyoo Park
Abstract
We propose the convergent graph solver (CGS), a deep learning method that learns iterative mappings to predict the properties of a graph system at its stationary state (fixed point) with guaranteed convergence. CGS systematically computes the fixed points of a target graph system and decodes them to estimate the stationary properties of the system without the prior knowledge of existing solvers or intermediate solutions. The forward propagation of CGS proceeds in three steps: (1) constructing the input dependent linear contracting iterative maps, (2) computing the fixed-points of the linear maps, and (3) decoding the fixed-points to estimate the properties. The contractivity of the constructed linear maps guarantees the existence and uniqueness of the fixed points following the Banach fixed point theorem. To train CGS efficiently, we also derive a tractable analytical expression for its gradient by leveraging the implicit function theorem. We evaluate the performance of CGS by applying it to various network-analytic and graph benchmark problems. The results indicate that CGS has competitive capabilities for predicting the stationary properties of graph systems, irrespective of whether the target systems are linear or non-linear. CGS also shows high performance for graph classification problems where the existence or the meaning of a fixed point is hard to be clearly defined, which highlights the potential of CGS as a general graph neural network architecture.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers10
- Deep Equilibrium Approaches to Diffusion ModelsAshwini Pokle, Zhengyang Geng, J. Zico KolterNeurIPS 2022 · 61 citations
- Deep Equilibrium Optical Flow EstimationShaojie Bai, Zhengyang Geng, Yash Savani, J. Zico KolterCVPR 2022 · 45 citations
- Optimization-Induced Graph Implicit Nonlinear DiffusionQi Chen, Yifei Wang, Yisen Wang, Jiansheng Yang et al.ICML 2022 · 44 citations
- MGNNI: Multiscale Graph Neural Networks with Implicit LayersJuncheng Liu, Bryan Hooi, Kenji Kawaguchi, Xiaokui XiaoNeurIPS 2022 · 36 citations
- Scalable and Effective Implicit Graph Neural Networks on Large GraphsJuncheng Liu, Bryan Hooi, Kenji Kawaguchi, Yiwei Wang et al.ICLR 2024 · 13 citations
Builds on8
- Efficient and Modular Implicit DifferentiationMathieu Blondel, Quentin Berthet, Marco Cuturi, Roy Frostig et al.NeurIPS 2022 · 386 citations
- Dissecting Neural ODEsStefano Massaroli, Michael Poli, Jinkyoo Park, Atsushi Yamashita et al.NeurIPS 2020 · 261 citations
- GraphNorm: A Principled Approach to Accelerating Graph Neural Network TrainingTianle Cai, Shengjie Luo, Keyulu Xu, Di He et al.ICML 2021 · 224 citations
- Implicit Graph Neural NetworksFangda Gu, Heng Chang, Wenwu Zhu, Somayeh Sojoudi et al.NeurIPS 2020 · 188 citations
- Monotone operator equilibrium networksEzra Winston, J. Zico KolterNeurIPS 2020 · 177 citations
Related papers
- A Convergence Analysis of Gradient Descent on Graph Neural NetworksPranjal Awasthi, Abhimanyu Das, Sreenivas GollapudiNeurIPS 2021 · 15 citations
- Deep Statistical SolversBalthazar Donon, Zhengying Liu, Wenzhuo Liu, Isabelle Guyon et al.NeurIPS 2020 · 22 citations
- Fast and Deep Graph Neural NetworksClaudio Gallicchio, Alessio MicheliAAAI 2020
- Implicit Graph Neural Networks: A Monotone Operator ViewpointJustin M. Baker, Qingsong Wang, Cory D. Hauck, Bao WangICML 2023 · 10 citations
- Neural operators meet conjugate gradients: The FCG-NO method for efficient PDE solvingAlexander Rudikov, Vladimir Fanaskov, Ekaterina A. Muravleva, Yuri M. Laevsky et al.ICML 2024 · 14 citations
