Timezone: »

Shift-Robust Node Classification via Graph Clustering Co-training
Qi Zhu · Chao Zhang · Chanyoung Park · Carl Yang · Jiawei Han
Event URL: https://openreview.net/forum?id=CXm7uzRlvxf »
It is widely known that machine learning models only achieve sub-optimal performance when testing data exhibit distribution shift against training \ie, $\Pr_\text{train}(X,Y) \neq \Pr_\text{test}(X,Y)$. Although Graph Neural Networks (GNNs) have become de facto models for semi-supervised learning tasks, they suffer even more from distribution shift because multiple types of shifts origin from not only node features but graph structures. Existing domain adaptation methods only work for specific type of shifts. In response, we propose Shift-Robust Node Classification (SRNC) - a unified domain adaptation framework for different kinds of distribution shifts on graph. Specifically, we co-train an unsupervised cluster GNN, which captures the data distribution by graph homophily on target graph. Then a shift-robust classifier is optimized on training graph and pseudo samples from target graph, which are provided by cluster GNN. Compared to the existing domain adaptation algorithms on graph, our approach works for both open-set and close-set shifts with convergence guarantees.In our experiments, the classification accuracy is improved at least $3\%$ against the second-best baseline under open-set shifts. On time-evolving graph with close-set shift, existing domain adaption algorithms can barely improve the generalization if not worse. SRNC is still able to mitigate the negative effect ($>2\%$ absolute improvements) of the shift across different testing-times.

Author Information

Qi Zhu (University of Illinois, Urbana Champaign)
Chao Zhang (Georgia Institute of Technology)
Chanyoung Park (Korea Advanced Institute of Science and Technology)
Carl Yang (Emory University)
Jiawei Han (University of Illinois at Urbana-Champaign)

More from the Same Authors