#前沿科技|中科院科学家研究图神经网络框架获进展
图神经网络(GNNs)在生物医药、知识图谱和AI4S等领域具备应用潜力。随着真实应用场景中图规模扩展至亿级边及以上,GNN训练面临通信开销与计算等问题。
中国科学院科学家团队研发TAC框架,融合数据亲和缓存填充算法、稀疏性感知的混合矩阵存储格式、多层次细粒度训练流水线三大技术,可提升缓存局部性有效降低全局通信开销,并借助Tensor Core实现稀疏矩阵计算加速。实验证明,该框架端到端性能优于其他框架[强]『前沿科技 | 中科院科学家研究图神经网络框架获进展』http://t.cn/AXSrdVDs
发布于 北京
