← Search

Keshav Balasubramanian

3 accepted papers

2022

SpreadGNN: Decentralized Multi-Task Federated Learning for Graph Neural Networks on Molecular Data

AAAI 2022technical

Graph Neural Networks (GNNs) are the first choice methods for graph machine learning problems thanks to their ability to learn state-of-the-art level representations from graph-structured data. However, centralizing a massive amount of real-world graph data for GNN training is prohibitive due to use…

Cited by 52SourcePDFScholar
2022

StATIK: Structure and Text for Inductive Knowledge Graph Completion

NAACL 2022findings

Knowledge graphs (KGs) often represent knowledge bases that are incomplete. Machine learning models can alleviate this by helping automate graph completion. Recently, there has been growing interest in completing knowledge bases that are dynamic, where previously unseen entities may be added to the…

2021

Graph Traversal with Tensor Functionals: A Meta-Algorithm for Scalable Learning

ICLR 2021poster

Graph Representation Learning (GRL) methods have impacted fields from chemistry to social science. However, their algorithmic implementations are specialized to specific use-cases e.g. "message passing" methods are run differently from "node embedding" ones. Despite their apparent differences, all t…