一种特征切分的分布式图卷积神经网络实现方法

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
一种特征切分的分布式图卷积神经网络实现方法
申请号:CN202411137148
申请日期:2024-08-19
公开号:CN119106702B
公开日期:2025-06-06
类型:发明专利
摘要
本发明提供一种特征切分的分布式图卷积神经网络方法,首先将输入特征分割成多个部分并分配到不同的计算节点,每个计算节点上加载完整的图结构和一小部分节点特征,从而降低内存需求;然后,每个计算节点上的GCN分片模型对其特征片段进行前向传播,计算节点间的通信仅在输入和输出阶段进行;最后,所有计算节点的输出表征传输到主节点进行拼接,并通过切片编码统一调整各块GCN的输出表征,最终提高图神经网络在大规模图数据上的训练效率。
技术关键词
切片 分片 卷积神经网络方法 生成特征 策略 节点数 编码向量 节点特征 列表 分类器 矩阵 内存 因子 阶段 参数 数据