AutoMaster: Differentiable Graph Neural Network Architecture Search for Collaborative Filtering Recommendation

被引:0
作者
Mu, Caihong [1 ]
Yu, Haikun [1 ]
Zhang, Keyang [1 ]
Tian, Qiang [1 ]
Liu, Yi [2 ]
机构
[1] Xidian Univ, Xian 710071, Peoples R China
[2] Xidian Univ, Sch Elect Engn, Xian 710071, Peoples R China
来源
WEB ENGINEERING, ICWE 2024 | 2024年 / 14629卷
基金
中国国家自然科学基金;
关键词
Collaborative filtering; Graph neural network; Neural architecture search;
D O I
10.1007/978-3-031-62362-2_6
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Graph Neural Networks (GNNs) have been widely applied in Collaborative Filtering (CF) and have demonstrated powerful capabilities in recommender systems (RSs). In recent years, there has been a heated debate on whether the non-linear propagation mechanism in Graph Convolutional Networks (GCNs) is suitable for CF tasks, and the performance of linear propagation is believed to be superior to non-linear propagationmainly in the field of RSs. Therefore, it is necessary to reexamine this issue: (1) whether linear propagation generally outperforms non-linear propagation, and (2) whether a combination of linear and non-linear propagation can be applied to CF tasks to achieve better accuracy. Furthermore, most existing studies design a single model architecture tailored to specific data or scenarios, and there remains a challenging and worthwhile problem to obtain the best-performing model in new recommendation data. To address the above issues, we propose a model called AutoMaster, which implements differentiable graph neural network architecture search for CF recommendation and automatically designs GNN architectures specific to different datasets. We design a compact and representative search space that includes various linear and non-linear graph convolutional layers, and employ a differentiable search strategy to search for the best-performing hybrid architecture in different recommendation datasets. Experimental results on five real-world datasets demonstrate that the GNN automatically achieved by the proposed AutoMaster contains both linear and nonlinear propagation, and outperforms several advanced GNN based CF models designed by the experienced human designers.
引用
收藏
页码:82 / 98
页数:17
相关论文
共 32 条
  • [1] [Anonymous], 2013, P 12 PYTHON SCI C
  • [2] Brody S., 2022, INT C LEARN REPR
  • [3] AutoGSR: Neural Architecture Search for Graph-based Session Recommendation
    Chen, Jingfan
    Zhu, Guanghui
    Hou, Haojun
    Yuan, Chunfeng
    Huang, Yihua
    [J]. PROCEEDINGS OF THE 45TH INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL (SIGIR '22), 2022, : 1694 - 1704
  • [4] Chen L, 2020, AAAI CONF ARTIF INTE, V34, P27
  • [5] Deep Neural Networks for YouTube Recommendations
    Covington, Paul
    Adams, Jay
    Sargin, Emre
    [J]. PROCEEDINGS OF THE 10TH ACM CONFERENCE ON RECOMMENDER SYSTEMS (RECSYS'16), 2016, : 191 - 198
  • [6] Elsken T, 2019, J MACH LEARN RES, V20
  • [7] Fey M, 2019, Arxiv, DOI [arXiv:1903.02428, DOI 10.48550/ARXIV.1903.02428]
  • [8] Gao C., 2023, ACM T RECOMMENDER SY, V1, P1, DOI 10.1145/3568022
  • [9] Gao Y, 2019, Arxiv, DOI arXiv:1904.09981
  • [10] Gao Y, 2020, PROCEEDINGS OF THE TWENTY-NINTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P1403