图神经网络训练方法、装置、电子设备及存储介质

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
图神经网络训练方法、装置、电子设备及存储介质
申请号:CN202411553012
申请日期:2024-11-01
公开号:CN119647541A
公开日期:2025-03-18
类型:发明专利
摘要
本申请公开一种图神经网络训练方法、装置、电子设备及存储介质,属于图神经网络技术领域。该方法包括:通过对图神经网络训练任务进行解析,获取原始图神经网络的网络参数和硬件资源限制条件;将原始图神经网络的训练过程划分为至少一个训练优化阶段,确定各个训练优化阶段对应的训练参数;基于训练参数的初始参数组合值、原始图神经网络的网络参数和硬件资源限制条件,确定图神经网络的性能预测结果,并根据性能预测结果在训练参数的参数组合取值空间内搜索训练参数的目标参数组合值;基于训练参数的目标参数组合值,对原始图神经网络进行训练,得到目标图神经网络。通过该方式,可以提高图神经网络在多样化需求下的自适应能力。
技术关键词
参数 神经网络训练方法 仿真模型 阶段 神经网络训练装置 硬件资源配置 硬件资源信息 数据传输策略 电子设备 神经网络技术 可读存储介质 处理器 生成特征 复杂度 规模 程序 指令 模块