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
来源
IEEE TRANSACTIONS ON NETWORK SCIENCE AND ENGINEERING | 2024年 / 11卷 / 04期
基金
澳大利亚研究理事会;
关键词
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
相关论文
共 31 条
  • [1] Friends and neighbors on the Web
    Adamic, LA
    Adar, E
    [J]. SOCIAL NETWORKS, 2003, 25 (03) : 211 - 230
  • [2] Missing Link Prediction using Common Neighbor and Centrality based Parameterized Algorithm
    Ahmad, Iftikhar
    Akhtar, Muhammad Usman
    Noor, Salma
    Shahnaz, Ambreen
    [J]. SCIENTIFIC REPORTS, 2020, 10 (01)
  • [3] Ensemble methods in machine learning
    Dietterich, TG
    [J]. MULTIPLE CLASSIFIER SYSTEMS, 2000, 1857 : 1 - 15
  • [4] Fey J. E., 2019, P ICLR WORKSH REPR L
  • [5] Ensemble deep learning: A review
    Ganaie, M. A.
    Hu, Minghui
    Malik, A. K.
    Tanveer, M.
    Suganthan, P. N.
    [J]. ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2022, 115
  • [6] NEURAL NETWORK ENSEMBLES
    HANSEN, LK
    SALAMON, P
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 1990, 12 (10) : 993 - 1001
  • [7] Modern temporal network theory: a colloquium
    Holme, Petter
    [J]. EUROPEAN PHYSICAL JOURNAL B, 2015, 88 (09)
  • [8] Temporal networks
    Holme, Petter
    Saramaki, Jari
    [J]. PHYSICS REPORTS-REVIEW SECTION OF PHYSICS LETTERS, 2012, 519 (03): : 97 - 125
  • [9] Kazemi SM, 2020, J MACH LEARN RES, V21
  • [10] Kipf T. N., 2017, P INT C LEARN REPR, P1