Unsupervised Joint k-node Graph Representations with Compositional Energy-Based Models
Leonardo Cotta, Carlos H. C. Teixeira, Ananthram Swami, Bruno Ribeiro
Abstract
Existing Graph Neural Network (GNN) methods that learn inductive unsupervised graph representations focus on learning node and edge representations by predicting observed edges in the graph. Although such approaches have shown advances in downstream node classification tasks, they are ineffective in jointly representing larger k-node sets, k>2. We propose MHM-GNN, an inductive unsupervised graph representation approach that combines joint k-node representations with energy-based models (hypergraph Markov networks) and GNNs. To address the intractability of the loss that arises from this combination, we endow our optimization with a loss upper bound using a finite-sample unbiased Markov Chain Monte Carlo estimator. Our experiments show that the unsupervised joint k-node representations of MHM-GNN produce better unsupervised representations than existing approaches from the literature. * http://cottascience.github.io/ Preprint. Under review.
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 papers3
- Weisfeiler and Lehman Go Topological: Message Passing Simplicial NetworksCristian Bodnar, Fabrizio Frasca, Yuguang Wang, Nina Otter et al.ICML 2021 · 315 citations
- Regularized Molecular Conformation FieldsLihao Wang, Yi Zhou, Yiqun Wang, Xiaoqing Zheng et al.NeurIPS 2022 · 9 citations
- Investigating Out-of-Distribution Generalization of GNNs: An Architecture PerspectiveKai Guo, Hongzhi Wen, Wei Jin, Yaming Guo et al.KDD 2024 · 7 citations
Builds on3
- InfoGraph: Unsupervised and Semi-supervised Graph-Level Representation Learning via Mutual Information MaximizationFan-Yun Sun, Jordan Hoffmann, Vikas Verma, Jian TangICLR 2020 · 1,010 citations
- Weisfeiler and Leman go sparse: Towards scalable higher-order graph embeddingsChristopher Morris, Gaurav Rattan, Petra MutzelNeurIPS 2020 · 190 citations
- On the Equivalence between Positional Node Embeddings and Structural Graph RepresentationsBalasubramaniam Srinivasan, Bruno RibeiroICLR 2020 · 143 citations
Related papers
- Neural Structured Prediction for Inductive Node ClassificationMeng Qu, Huiyu Cai, Jian TangICLR 2022 · 23 citations
- INDIGO: GNN-Based Inductive Knowledge Graph Completion Using Pair-Wise EncodingShuwen Liu, Bernardo Cuenca Grau, Ian Horrocks, Egor V. KostylevNeurIPS 2021 · 128 citations
- Molecular Representation Learning via Heterogeneous Motif Graph Neural NetworksZhaoning Yu, Hongyang GaoICML 2022 · 56 citations
- Labeling Trick: A Theory of Using Graph Neural Networks for Multi-Node Representation LearningMuhan Zhang, Pan Li, Yinglong Xia, Kai Wang et al.NeurIPS 2021 · 255 citations
- A Variational Edge Partition Model for Supervised Graph Representation LearningYilin He, Chaojie Wang, Hao Zhang, Bo Chen et al.NeurIPS 2022 · 6 citations
