ICLRMar, 2022
针对原子模拟的十亿参数图神经网络训练
Towards Training Billion Parameter Graph Neural Networks for Atomic Simulations
Anuroop Sriram, Abhishek Das, Brandon M. Wood, Siddharth Goyal, C. Lawrence Zitnick
TL;DR本文提出的图并行算法可在多个 GPU 上分布输入图,从而实现训练数十亿个参数的大型 GNN, 该方法在大规模开放铂族催化剂的数据集上取得了新的最先进结果。