We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:


Current browse context:


Change to browse by:

References & Citations

DBLP - CS Bibliography


(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Computer Science > Machine Learning

Title: Models and Benchmarks for Representation Learning of Partially Observed Subgraphs

Abstract: Subgraphs are rich substructures in graphs, and their nodes and edges can be partially observed in real-world tasks. Under partial observation, existing node- or subgraph-level message-passing produces suboptimal representations. In this paper, we formulate a novel task of learning representations of partially observed subgraphs. To solve this problem, we propose Partial Subgraph InfoMax (PSI) framework and generalize existing InfoMax models, including DGI, InfoGraph, MVGRL, and GraphCL, into our framework. These models maximize the mutual information between the partial subgraph's summary and various substructures from nodes to full subgraphs. In addition, we suggest a novel two-stage model with $k$-hop PSI, which reconstructs the representation of the full subgraph and improves its expressiveness from different local-global structures. Under training and evaluation protocols designed for this problem, we conduct experiments on three real-world datasets and demonstrate that PSI models outperform baselines.
Comments: CIKM 2022 Short Paper (Camera-ready + Appendix)
Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Social and Information Networks (cs.SI)
Cite as: arXiv:2209.00508 [cs.LG]
  (or arXiv:2209.00508v2 [cs.LG] for this version)

Submission history

From: Dongkwan Kim [view email]
[v1] Thu, 1 Sep 2022 14:51:37 GMT (1135kb,D)
[v2] Mon, 19 Sep 2022 04:29:55 GMT (1023kb,D)

Link back to: arXiv, form interface, contact.