基于边缘衰减采样的模型训练方法、装置和终端设备

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
基于边缘衰减采样的模型训练方法、装置和终端设备
申请号:CN202510293925
申请日期:2025-03-12
公开号:CN120373397A
公开日期:2025-07-25
类型:发明专利
摘要
本申请涉及一种基于边缘衰减采样的模型训练方法、装置和终端设备,应用于对连续时间动态图模型进行训练,该方法包括:将连续时间动态图划分为多个动态子图,其中,多个动态子图包括第一动态子图;对位于第一动态子图中的、与目标节点交互的第一邻居节点进行采样,以及按照预设采样概率,对位于除第一动态子图之外的动态子图中的、与目标节点交互的第二邻居节点进行采样,得到采样子图;获取第一邻居节点与第二邻居节点的特征数据;基于采样子图、以及第一邻居节点与第二邻居节点的特征数据,训练初始连续时间动态图模型,以获得训练完备的连续时间动态图模型。通过本申请,解决了相关技术中对连续时间动态图模型的并行训练效率较低的问题。
技术关键词
节点 邻居 训练器 模型训练方法 动态 热点 终端设备 模型训练装置 超参数 贪心策略 计算机程序产品 记忆单元 采样模块 处理器 数据存储 存储器 序列 时间段