提升图神经网络训练效率的方法、装置、设备和介质

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
提升图神经网络训练效率的方法、装置、设备和介质
申请号:CN202510845554
申请日期:2025-06-23
公开号:CN120387493B
公开日期:2025-10-28
类型:发明专利
摘要
本发明提供一种提升图神经网络训练效率的方法、装置、设备和介质,所述方法的步骤包括构建论文共引关系网络,将论文共引关系建模为图结构数据作为图神经网络的输入;基于蒙特卡洛树搜索次模优化算法设计标注策略,完成关系网络的节点标注;选取图神经网络的训练集和测试集;将训练集输入分类模型,对图神经网络进行训练和分类,通过聚合节点的特征,学习节点的特征表示,得到训练好的图神经网络;将测试集输入训练好的图神经网络,进行图神经网络模型测试。实验结果表明,本发明的方法通过优化标注策略快速获取训练集的同时在多个典型数据集上显著提高了分类准确性,从而实现了图神经网络训练效率的提升,具有良好的应用前景。
技术关键词
节点 蒙特卡洛树搜索 神经网络训练 关系网络 标注策略 论文 关系建模 搜索算法 优化算法设计 分类特征 启发式策略 邻居 训练集 阶段 邻域