Skip to yearly menu bar Skip to main content

Workshop: Geometrical and Topological Representation Learning

Efficient Representation Learning of Subgraphs by Subgraph-To-Node Translation

Dongkwan Kim · Alice Oh

Keywords: [ graph neural networks ] [ efficiency ]


A subgraph is a data structure that can represent various real-world problems. We propose Subgraph-To-Node (S2N) translation, which is a novel formulation to efficiently learn representations of subgraphs. Specifically, given a set of subgraphs in the global graph, we construct a new graph by coarsely transforming subgraphs into nodes. We perform subgraph-level tasks as node-level tasks through this translation. By doing so, we can significantly reduce the memory and computational costs in both training and inference. We conduct experiments on four real-world datasets to evaluate performance and efficiency. Our experiments demonstrate that models with S2N translation are more efficient than state-of-the-art models without substantial performance decrease.

Chat is not available.