DREAM: a Dual Variational Framework for Unsupervised Graph Domain Adaptation.

Journal: IEEE transactions on pattern analysis and machine intelligence
Published Date:

Abstract

Graph classification has been a prominent problem in graph machine learning fields. This problem has been investigated by leveraging message passing neural networks (MPNNs) to learn powerful graph representations. However, MPNNs extract topological semantics implicitly under label supervision, which could suffer from domain shift and label scarcity in unsupervised domain adaptation settings. In this paper, we propose an effective solution named Dual Variational Semantics Graph Mining (DREAM) for unsupervised graph domain adaptation by combining graph structural semantics from complementary perspectives. Besides a message passing branch to learn implicit semantics, our DREAM trains a path aggregation branch, which can provide explicit high-order structural semantics as a supplement. To train these two branches conjointly, we employ an expectation-maximization (EM) style variational framework for the maximization of likelihood. In the E-step, we fix the message passing branch and construct a graph-of-graph to indicate the geometric correlation between source and target domains, which would be adopted for the optimization of the other branch. In the M-step, we train the message passing branch and update the graph neural networks on the graph-of-graph with the other branch fixed. The alternative optimization improves the collaboration of knowledge from two branches. Extensive experiments on several benchmark datasets validate the superiority of the proposed DREAM compared with various baselines.

Authors

  • Nan Yin
    Department of Computer Science and Engineering, Hong Kong University of Science and Technology, Hong Kong, China. yinnan8911@gmail.com.
  • Li Shen
    Department of Clinical Pharmacy, Children's Hospital of Fudan University, National Children's Medical Center, Shanghai, China.
  • Mengzhu Wang
    College of Computer, National University of Defense Technology, Kaifu District, Changsha 410073, China. Electronic address: wangmengzhu19@nudt.edu.cn.
  • Xinwang Liu
  • Chong Chen
    Department of Orthorpaedic Surgery, Peking Union Medical College Hospital, No.1 Shuaifuyuan, Dongcheng district, Beijing, 100730, People's Republic of China.
  • Xian-Sheng Hua

Keywords

No keywords available for this article.