On the Effectiveness of Heterogeneous Ensembles Combining Graph Neural Networks and Heuristics for Dynamic Link Prediction

被引:1
|
作者
Skarding, Joakim [1 ]
Gabrys, Bogdan [1 ]
Musial, Katarzyna [1 ]
机构
[1] Univ Technol Sydney, Data Sci Inst, Sydney, NSW 2007, Australia
基金
澳大利亚研究理事会;
关键词
Graph neural networks; Task analysis; Diversity reception; Training; Logistic regression; Predictive models; Deep learning; Link prediction; graph neural network; dynamic network; temporal graph; dynamic graph neural network; ensemble;
D O I
10.1109/TNSE.2023.3343927
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
With the ubiquity of complex networks in domains as diverse as social network analysis, recommender systems and epidemiology, improvements in link predictions have a potential for far-reaching impact. In machine learning, a well-known approach to improve classification performance is to combine several methods in an ensemble. It is also well established that, diversity or complementarity of the classifiers to be combined, is a prerequisite for a potential performance improvement of the formed ensemble. In this work, we combine link prediction heuristics (e.g. Common Neighbour, Adamic-Adar etc.), with static graph neural networks, discrete dynamic graph neural networks, and continuous dynamic graph neural networks. We analyze the importance of each of the methods and how complementary they are with respect to each other. We also use greedy searches to select better subsets of methods to combine and train the ensemble in a roll-forward manner to make it adapt to changes in the dynamic network over time. In the conducted experiments on a number of challenging link prediction tasks, the ensemble from the greedy search and the roll-forward ensemble shows on average a 55% and 61% improvement respectively, over the best single method.
引用
收藏
页码:3250 / 3259
页数:10
相关论文
共 50 条
  • [41] Combining Differentiable PDE Solvers and Graph Neural Networks for Fluid Flow Prediction
    Belbute-Peres, Filipe de Avila
    Economon, Thomas D.
    Kolter, J. Zico
    INTERNATIONAL CONFERENCE ON MACHINE LEARNING, VOL 119, 2020, 119
  • [42] Drug-target Interaction Prediction By Combining Transformer and Graph Neural Networks
    Liu, Junkai
    Lu, Yaoyao
    Guan, Shixuan
    Jiang, Tengsheng
    Ding, Yijie
    Fu, Qiming
    Cui, Zhiming
    Wu, Hongjie
    CURRENT BIOINFORMATICS, 2024, 19 (04) : 316 - 326
  • [43] Combining Differentiable PDE Solvers and Graph Neural Networks for Fluid Flow Prediction
    Belbute-Peres, Filipe de Avila
    Economon, Thomas D.
    Kolter, J. Zico
    25TH AMERICAS CONFERENCE ON INFORMATION SYSTEMS (AMCIS 2019), 2019,
  • [44] Link Prediction in Heterogeneous Social Networks
    Negi, Sumit
    Chaudhury, Santanu
    CIKM'16: PROCEEDINGS OF THE 2016 ACM CONFERENCE ON INFORMATION AND KNOWLEDGE MANAGEMENT, 2016, : 609 - 617
  • [45] Link Prediction in Aligned Heterogeneous Networks
    Liu, Fangbing
    Xia, Shu-Tao
    ADVANCES IN KNOWLEDGE DISCOVERY AND DATA MINING, PART I, 2015, 9077 : 33 - 44
  • [46] Link Prediction in Temporal Heterogeneous Networks
    Lakshmi, T. Jaya
    Bhavani, S. Durga
    INTELLIGENCE AND SECURITY INFORMATICS (PAISI 2017), 2017, 10241 : 83 - 98
  • [47] Evaluating Graph Neural Networks for Link Prediction: Current Pitfalls and New Benchmarking
    Li, Juanhui
    Shomer, Harry
    Mao, Haitao
    Zeng, Shenglai
    Ma, Yao
    Shah, Neil
    Tang, Jiliang
    Yin, Dawei
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 36 (NEURIPS 2023), 2023,
  • [48] Hybrid Structure Encoding Graph Neural Networks with Attention Mechanism for Link Prediction
    Hu, Man
    Sun, Dezhi
    You, Fucheng
    Xiao, Han
    2022 IEEE 34TH INTERNATIONAL CONFERENCE ON TOOLS WITH ARTIFICIAL INTELLIGENCE, ICTAI, 2022, : 417 - 424
  • [49] Graph Neural Networks in PyTorch for Link Prediction in Industry 4.0 Process Graphs
    Kafeza, Eleanna
    Drakpopoulos, Georgios
    Mylonas, Phivos
    ARTIFICIAL INTELLIGENCE APPLICATIONS AND INNOVATIONS, PT III, AIAI 2024, 2024, 713 : 220 - 234
  • [50] A Representation Learning Link Prediction Approach Using Line Graph Neural Networks
    Tai, Yu
    Yang, Hongwei
    He, Hui
    Wu, Xinglong
    Zhang, Weizhe
    PATTERN RECOGNITION AND COMPUTER VISION, PRCV 2023, PT IX, 2024, 14433 : 195 - 207