| 专利号 | 2025117833612 | 申请日 | 2025-12-01 | 专利名称 | 一种面向CPU采样的图神经网络训练动态缓存方法、装置和介质 |
| 授权日 | 2026-01-30 | 专利权人 | 山东科技大学 | 发明人 | 崔焕庆;李代辉;徐强;周升庆;吴庆龙;董柯桢;范菲 |
| 主分类号 | G06F9/50 | 关键词 | 应用领域 | ||
| 摘要 | 本发明提供了一种面向CPU采样的图神经网络训练动态缓存方法、装置和介质，属于图神经网络训练技术领域。所述方法包括：将图神经网络训练的图数据划分为与加速卡数量相等的分区；将每个加速卡的显存划分为T区、F区、L区和R区；主机CPU对各分区进行采样并生成采样子图，根据采样子图获取顶点标签分发至加速卡；在首轮训练中，从主机内存获取当前采样子图的顶点特征，在非首轮训练中，依次在当前加速卡F区、L区、R区中查找顶点特征，若未命中则通过跨加速卡或主机内存获取；定期根据顶点特征进入F区的轮次与预设阈值，将F区中轮次大于预设阈值的顶点特征迁移至L区；各加速卡更新模型参数。本发明方法能够显著提升图神经网络的整体训练效率。 | ||||
| 创新点 | |||||
| 技术分类 | 标 签 | 战兴产业 | 新一代信息技术 互联网与云计算、大数据服务 | ||
| 运营方式 | 合作方式 | ||||
| 联系人 | 联系电话 | 电子邮箱 | |||
| 详细说明 | |||||
