Learning Representations for Hierarchies with Minimal Support
Benjamin Rozonoyer, Michael Boratko, Dhruvesh Patel, Wenlong Zhao, Shib Sankar Dasgupta, Hung Le, Andrew McCallum
Abstract
When training node embedding models to represent large directed graphs (digraphs), it is impossible to observe all entries of the adjacency matrix during training. As a consequence most methods employ sampling. For very large digraphs, however, this means many (most) entries may be unobserved during training. In general, observing every entry would be necessary to uniquely identify a graph, however if we know the graph has a certain property some entries can be omitted - for example, only half the entries would be required for a symmetric graph. In this work, we develop a novel framework to identify a subset of entries required to uniquely distinguish a graph among all transitively-closed DAGs. We give an explicit algorithm to compute the provably minimal set of entries, and demonstrate empirically that one can train node embedding models with greater efficiency and performance, provided the energy function has an appropriate inductive bias. We achieve robust performance on synthetic hierarchies and a larger real-world taxonomy, observing improved convergence rates in a resource-constrained setting while reducing the set of training examples by as much as 99%.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext d0d6fb89-61f0-49bc-9cbd-e1a1ca4cbc6fCited by top-tier papers1
Ask how each one uses itBuilds on10
- Iterative Deep Graph Learning for Graph Neural Networks: Better and Robust Node EmbeddingsYu Chen, Lingfei Wu, Mohammed J. ZakiNeurIPS 2020 · 559 citations
- Understanding Negative Sampling in Graph Representation LearningZhen Yang, Ming Ding, Chang Zhou, Hongxia Yang et al.KDD 2020 · 172 citations
- Tail-GNN: Tail-Node Graph Neural NetworksZemin Liu, Trung-Kien Nguyen, Yuan FangKDD 2021 · 105 citations
- Improving Local Identifiability in Probabilistic Box EmbeddingsShib Sankar Dasgupta, Michael Boratko, Dongxu Zhang, Luke Vilnis et al.NeurIPS 2020 · 75 citations
- Rot-Pro: Modeling Transitivity by Projection in Knowledge Graph EmbeddingTengwei Song, Jie Luo, Lei HuangNeurIPS 2021 · 46 citations
Related papers
- Faster Graph Embeddings via CoarseningMatthew Fahrbach, Gramoz Goranci, Richard Peng, Sushant Sachdeva et al.ICML 2020 · 32 citations
- Exploring Neural Scaling Law and Data Pruning Methods For Node Classification on Large-scale GraphsZhen Wang, Yaliang Li, Bolin Ding, Yule Li et al.WWW 2024 · 2 citations
- GraphZoom: A Multi-level Spectral Approach for Accurate and Scalable Graph EmbeddingChenhui Deng, Zhiqiang Zhao, Yongyu Wang, Zhiru Zhang et al.ICLR 2020 · 122 citations
- TT-GNN: Efficient On-Chip Graph Neural Network Training via Embedding Reformation and Hardware OptimizationZheng Qu, Dimin Niu, Shuangchen Li, Hongzhong Zheng et al.MICRO 2023 · 6 citations
- DeepWalking Backwards: From Embeddings Back to GraphsSudhanshu Chanpuriya, Cameron Musco, Konstantinos Sotiropoulos, Charalampos E. TsourakakisICML 2021 · 19 citations
