Proceedings

PALA: Class-imbalanced Graph Domain Adaptation via Prototype-anchored Learning and Alignment

Download BibTeX Cite

Proceedings of the Thirty-Fourth International Joint Conference on Artificial Intelligence. Main Track. Pages 3198-3207.

https://doi.org/10.24963/ijcai.2025/356

Abstract

Graph domain adaptation is a key subfield of graph transfer learning that aims to bridge domain gaps by transferring knowledge from a label-rich source graph to an unlabeled target graph. However, most existing methods assume balanced labels in the source graph, which often fails in practice and leads to biased knowledge transfer. To address this, in this paper, we propose a prototype-anchored learning and alignment framework for class-imbalanced graph domain adaptation. Specifically, we incorporate pointwise node mutual information into the graph encoder to capture high-order topological proximity and learn generalized node representations. Leveraging this, we then introduce categorical prototypes with adversarial proto-instances for prototype-anchored learning and recalibration to represent the source graph under an imbalanced class distribution. Finally, we introduce a weighted prototype contrastive adaptation strategy that aligns target pseudo-labels with source prototypes to handle class imbalance during adaptation. Extensive experiments show that our PALA outperforms the state-of-the-art methods. Our code is available at https://github.com/maxin88scu/PALA.

  1. [Arpit et al., 2017] Devansh Arpit, Stanisław Jastrzębski, Nicolas Ballas, David Krueger, Emmanuel Bengio, Maxinder S Kanwal, Tegan Maharaj, Asja Fischer, Aaron Courville, Yoshua Bengio, et al. A closer look at memo- rization in deep networks. In Proceedings of the Interna- tional Conference on Machine Learning, pages 233-242, 2017.
  2. [Cai et al., 2024] Ruichu Cai, Fengzhu Wu, Zijian Li, Pengfei Wei, Lingling Yi, and Kun Zhang. Graph do- main adaptation: A generative view. ACM Transactions on Knowledge Discovery from Data, 18(3):1-24, 2024.
  3. [Chen et al., 2019] Chaoqi Chen, Weiping Xie, Wenbing Huang, Yu Rong, Xinghao Ding, Yue Huang, Tingyang Xu, and Junzhou Huang. Progressive feature alignment for unsupervised domain adaptation. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pat- tern Recognition, pages 627-636, 2019.
  4. [Cui et al., 2019] Yin Cui, Menglin Jia, Tsung-Yi Lin, Yang Song, and Serge Belongie. Class-balanced loss based on effective number of samples. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 9268-9277, 2019.
  5. [Dai et al., 2022] Quanyu Dai, Xiao-Ming Wu, Jiaren Xiao, Xiao Shen, and Dan Wang. Graph transfer learning via adversarial domain adaptation with graph convolution. IEEE Transactions on Knowledge and Data Engineering, 35(5):4908-4922, 2022.
  6. [Graf et al., 2021] Florian Graf, Christoph Hofer, Marc Ni- ethammer, and Roland Kwitt. Dissecting supervised con- trastive learning. In Proceedings of the International Con- ference on Machine Learning, pages 3821-3830, 2021.
  7. [Gretton et al., 2012] Arthur Gretton, Karsten M Borgwardt, Malte J Rasch, Bernhard Schölkopf, and Alexander Smola. A kernel two-sample test. The Journal of Machine Learning Research, 13(1):723-773, 2012.
  8. [Guo et al., 2024] Xinyu Guo, Kai Wu, Xiaoyu Zhang, and Jing Liu. Automated loss function search for class- imbalanced node classification. In Forty-first International Conference on Machine Learning, 2024.
  9. [Hong et al., 2021] Youngkyu Hong, Seungju Han, Kwanghee Choi, Seokjun Seo, Beomsu Kim, and Buru Chang. Disentangling label distribution for long- tailed visual recognition. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 6626-6636, 2021.
  10. [Huang et al., 2016] Chen Huang, Yining Li, Chen Change Loy, and Xiaoou Tang. Learning deep representation for imbalanced classification. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 5375-5384, 2016.
  11. [Huang et al., 2023] Zhan Ao Huang, Yongsheng Sang, Yanan Sun, and Jiancheng Lv. Neural network with absent minority class samples and boundary shifting for imbal- anced data classification. Neural Computing and Applica- tions, 35(12):8937-8953, 2023.
  12. [Ju et al., 2024a] Wei Ju, Zhengyang Mao, Siyu Yi, Yi- fang Qin, Yiyang Gu, Zhiping Xiao, Yifan Wang, Xiao Luo, and Ming Zhang. Hypergraph-enhanced dual semi- supervised graph classification. In Proceedings of the In- ternational Conference on Machine Learning, 2024.
  13. [Ju et al., 2024b] Wei Ju, Siyu Yi, Yifan Wang, Qingqing Long, Junyu Luo, Zhiping Xiao, and Ming Zhang. A sur- vey of data-efficient graph learning. In Proceedings of the AAAI Conference on Artificial Intelligence, 2024. IJCAI
  14. [Ju et al., 2024c] Wei Ju, Siyu Yi, Yifan Wang, Zhiping Xiao, Zhengyang Mao, Hourun Li, Yiyang Gu, Yifang Qin, Nan Yin, Senzhang Wang, et al. A survey of graph neural networks in real world: Imbalance, noise, privacy and ood challenges. arXiv preprint arXiv:2403.04468, 2024.
  15. [Ju et al., 2025] Wei Ju, Zhengyang Mao, Siyu Yi, Yifang Qin, Yiyang Gu, Zhiping Xiao, Jianhao Shen, Ziyue Qiao, and Ming Zhang. Cluster-guided contrastive class- imbalanced graph classification. In Proceedings of the AAAI Conference on Artificial Intelligence, 2025.
  16. [Kang et al., 2020] Bingyi Kang, Saining Xie, Marcus Rohrbach, Zhicheng Yan, Albert Gordo, Jiashi Feng, and Yannis Kalantidis. Decoupling representation and classi- fier for long-tailed recognition. In International Confer- ence on Learning Representations, 2020.
  17. [Kipf and Welling, 2017] Thomas N Kipf and Max Welling. Semi-supervised classification with graph convolutional networks. In International Conference on Learning Rep- resentations, 2017.
  18. [Li et al., 2023] Wen-Zhi Li, Chang-Dong Wang, Hui Xiong, and Jian-Huang Lai. Graphsha: Synthesizing harder samples for class-imbalanced node classification. In Proceedings of the 29th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, pages 1328- 1340, 2023.
  19. [Li et al., 2025] Hourun Li, Yifan Wang, Zhiping Xiao, Jia Yang, Changling Zhou, Ming Zhang, and Wei Ju. Disco: graph-based disentangled contrastive learning for cold- start cross-domain recommendation. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 39, pages 12049-12057, 2025.
  20. [Liang et al., 2020] Jian Liang, Dapeng Hu, and Jiashi Feng. Do we really need to access the source data? source hy- pothesis transfer for unsupervised domain adaptation. In Proceedings of the International Conference on Machine Learning, pages 6028-6039, 2020.
  21. [Liu et al., 2020] Sheng Liu, Jonathan Niles-Weed, Narges Razavian, and Carlos Fernandez-Granda. Early-learning regularization prevents memorization of noisy labels. In Proceedings of the Conference on Neural Information Pro- cessing Systems, pages 20331-20342, 2020.
  22. [Liu et al., 2024a] Meihan Liu, Zeyu Fang, Zhen Zhang, Ming Gu, Sheng Zhou, Xin Wang, and Jiajun Bu. Re- thinking propagation for unsupervised graph domain adap- tation. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 38, pages 13963-13971, 2024.
  23. [Liu et al., 2024b] Shikun Liu, Deyu Zou, Han Zhao, and Pan Li. Pairwise alignment improves graph domain adap- tation. In Forty-first International Conference on Machine Learning, 2024.
  24. [Liu et al., 2024c] Zhining Liu, Ruizhong Qiu, Zhichen Zeng, Hyunsik Yoo, David Zhou, Zhe Xu, Yada Zhu, Kommy Weldemariam, Jingrui He, and Hanghang Tong. Class-imbalanced graph learning without class rebalanc- ing. In Forty-first International Conference on Machine Learning, 2024.
  25. [Luo et al., 2023] Xiao Luo, Yusheng Zhao, Zhengyang Mao, Yifang Qin, Wei Ju, Ming Zhang, and Yizhou Sun. Rignn: A rationale perspective for semi-supervised open-world graph classification. Transactions on Machine Learning Research, 2023.
  26. [Luo et al., 2024a] Junyu Luo, Yiyang Gu, Xiao Luo, Wei Ju, Zhiping Xiao, Yusheng Zhao, Jingyang Yuan, and Ming Zhang. Gala: Graph diffusion-based alignment with jigsaw for source-free domain adaptation. IEEE Transac- tions on Pattern Analysis and Machine Intelligence, 2024.
  27. [Luo et al., 2024b] Junyu Luo, Zhiping Xiao, Yifan Wang, Xiao Luo, Jingyang Yuan, Wei Ju, Langechuan Liu, and Ming Zhang. Rank and align: Towards effective source- free graph domain adaptation. In Proceedings of the Inter- national Joint Conference on Artificial Intelligence, 2024.
  28. [Mao et al., 2023] Zhengyang Mao, Wei Ju, Yifang Qin, Xiao Luo, and Ming Zhang. Rahnet: Retrieval augmented hybrid network for long-tailed graph classification. In Pro- ceedings of the 31st ACM international conference on mul- timedia, pages 3817-3826, 2023.
  29. [Mao et al., 2025] Zhengyang Mao, Wei Ju, Siyu Yi, Yifan Wang, Zhiping Xiao, Qingqing Long, Nan Yin, Xinwang Liu, and Ming Zhang. Learning knowledge-diverse ex- perts for long-tailed graph classification. ACM Trans- actions on Knowledge Discovery from Data, 19(2):1-24, 2025.
  30. [Menon et al., 2021] Aditya Krishna Menon, Sadeep Jaya- sumana, Ankit Singh Rawat, Himanshu Jain, Andreas Veit, and Sanjiv Kumar. Long-tail learning via logit ad- justment. In Proceedings of the International Conference on Learning Representations, 2021.
  31. [Park et al., 2021] Joonhyung Park, Jaeyun Song, and Eunho Yang. Graphens: Neighbor-aware ego network synthesis for class-imbalanced node classification. In International Conference on Learning Representations, 2021.
  32. [Qiao et al., 2023] Ziyue Qiao, Xiao Luo, Meng Xiao, Hao Dong, Yuanchun Zhou, and Hui Xiong. Semi-supervised domain adaptation in graph transfer learning. In Proceed- ings of the International Joint Conference on Artificial In- telligence, pages 2279-2287, 2023. IJCAI
  33. [Qiu et al., 2021] Zhen Qiu, Yifan Zhang, Hongbin Lin, Shuaicheng Niu, Yanxia Liu, Qing Du, and Mingkui Tan. Source-free domain adaptation via avatar prototype gener- ation and adaptation. In Proceedings of the International Joint Conference on Artificial Intelligence, 2021. IJCAI
  34. [Qu et al., 2021] Liang Qu, Huaisheng Zhu, Ruiqi Zheng, Yuhui Shi, and Hongzhi Yin. Imgagn: Imbalanced net- work embedding via generative adversarial graph net- works. In Proceedings of the 27th ACM SIGKDD Con- ference on Knowledge Discovery & Data mining, pages 1390-1398, 2021.
  35. [Song et al., 2022] Jaeyun Song, Joonhyung Park, and Eunho Yang. Tam: topology-aware margin loss for class- imbalanced node classification. In International Confer- ence on Machine Learning, pages 20369-20383. PMLR, 2022.
  36. [Tan et al., 2020] Jingru Tan, Changbao Wang, Buyu Li, Quanquan Li, Wanli Ouyang, Changqing Yin, and Jun- jie Yan. Equalization loss for long-tailed object recog- nition. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 11662- 11671, 2020.
  37. [Tang et al., 2008] Jie Tang, Jing Zhang, Limin Yao, Juanzi Li, Li Zhang, and Zhong Su. Arnetminer: extraction and mining of academic social networks. In Proceedings of the 14th ACM SIGKDD International Conference on Knowl- edge Discovery and Data Mining, pages 990-998, 2008.
  38. [Tzeng et al., 2017] Eric Tzeng, Judy Hoffman, Kate Saenko, and Trevor Darrell. Adversarial discriminative domain adaptation. In Proceedings of the IEEE Confer- ence on Computer Vision and Pattern Recognition, pages 7167-7176, 2017.
  39. [Veličković et al., 2018] Petar Veličković, Guillem Cucurull, Arantxa Casanova, Adriana Romero, Pietro Liò, and Yoshua Bengio. Graph attention networks. In Interna- tional Conference on Learning Representations, 2018.
  40. [Wang and Deng, 2018] Mei Wang and Weihong Deng. Deep visual domain adaptation: A survey. Neurocomput- ing, 312:135-153, 2018.
  41. [Wang et al., 2021] Jianfeng Wang, Thomas Lukasiewicz, Xiaolin Hu, Jianfei Cai, and Zhenghua Xu. Rsg: A sim- ple but effective module for learning imbalanced datasets. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 3784-3793, 2021.
  42. [Wang et al., 2024] Yifan Wang, Xiao Luo, Chong Chen, Xian-Sheng Hua, Ming Zhang, and Wei Ju. Disensemi: Semi-supervised graph classification via disentangled rep- resentation learning. IEEE Transactions on Neural Net- works and Learning Systems, 2024.
  43. [Wu et al., 2023] Jun Wu, Jingrui He, and Elizabeth Ainsworth. Non-iid transfer learning on graphs. In Pro- ceedings of the AAAI Conference on Artificial Intelligence, pages 10342-10350, 2023.
  44. [Xu et al., 2018] Keyulu Xu, Weihua Hu, Jure Leskovec, and Stefanie Jegelka. How powerful are graph neural net- works? In International Conference on Learning Rep- resentations, 2018.
  45. [Yi et al., 2023] Si-Yu Yi, Zhengyang Mao, Wei Ju, Yong- Dao Zhou, Luchen Liu, Xiao Luo, and Ming Zhang. To- wards long-tailed recognition for graph classification via collaborative experts. IEEE Transactions on Big Data, 9(6):1683-1696, 2023.
  46. [You et al., 2023] Yuning You, Tianlong Chen, Zhangyang Wang, and Yang Shen. Graph domain adaptation via theory-grounded spectral regularization. In The eleventh International Conference on Learning Representations, 2023.
  47. [Yuan et al., 2023] Jingyang Yuan, Xiao Luo, Yifang Qin, Zhengyang Mao, Wei Ju, and Ming Zhang. Alex: To- wards effective graph transfer learning with noisy labels. In Proceedings of the ACM International Conference on Multimedia, pages 3647-3656, 2023.
  48. [Zeng et al., 2023] Liang Zeng, Lanqing Li, Ziqi Gao, Peilin Zhao, and Jian Li. Imgcl: Revisiting graph contrastive learning on imbalanced node classification. In Proceed- ings of the AAAI Conference on Artificial Intelligence, vol- ume 37, pages 11138-11146, 2023.
  49. [Zhao et al., 2021] Tianxiang Zhao, Xiang Zhang, and Suhang Wang. Graphsmote: Imbalanced node classifica- tion on graphs with graph neural networks. In Proceedings of the 14th ACM International Conference on Web Search and Data Mining, pages 833-841, 2021.
  50. [Zhuang and Ma, 2018] Chenyi Zhuang and Qiang Ma. Dual graph convolutional networks for graph-based semi- supervised classification. In Proceedings of the Web Con- ference, pages 499-508, 2018.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards