国内刊号:37-1389/N
国际刊号:1671-9352
发布日期:
作者:黄兴宇,赵明宇,吕子钰
单位:1. 中国科学院深圳先进技术研究院,广东 深圳 518055;2. 中山大学网络空间安全学院,广东 深圳 518107;3. 中国科学院大学,北京 100101
关键词:图神经网络,知识探针,模型评价,表征学习,
基金:国家自然科学基金资助项目(62002352);广东省基础与应用基础研究基金资助项目(2023A1515012848)
针对图神经网络(graph neural network, GNN)模型缺乏相应的探针这一问题,提出面向图神经网络表征学习的知识探测框架,基于不同领域数据的类别属性设计2种类别感知的知识探针,分别为聚类探针和对比聚类探针。2种探针分别探测不同模型的表征效果并给出相应的分数。在引用网络、社交网络和生物网络等3个邻域的8个数据集上,对7个经典的图神经网络模型的表征学习实现了系统性地知识探测和评估实验,归纳出探测和评估结论。
来源:2024年第7期
《山东大学学报(理学版)》期刊编辑部