KDD Cup 2021 与 OGB 大规模挑战赛图预测赛道第一名方案
机器学习
2021-06-22 v3
摘要
在本技术报告中，我们呈现了 KDD Cup 2021 OGB 大规模挑战赛 - PCQM4M-LSC 赛道的方案。我们采用 Graphormer 与 ExpC 作为基础模型。我们通过 8 折交叉验证训练每个模型，并额外以不同随机种子在训练集与验证集的并集中训练两个 Graphormer 模型。对于最终提交，我们对这 18 个模型使用朴素集成，取其输出平均值。使用我们的方法，我们的团队 MachineLearning 在测试集上取得了 0.1200 的 MAE，赢得了 KDD Cup 图预测赛道的第一名。
关键词
引用
@article{arxiv.2106.08279,
title = {First Place Solution of KDD Cup 2021 & OGB Large-Scale Challenge Graph Prediction Track},
author = {Chengxuan Ying and Mingqi Yang and Shuxin Zheng and Guolin Ke and Shengjie Luo and Tianle Cai and Chenglin Wu and Yuxin Wang and Yanming Shen and Di He},
journal= {arXiv preprint arXiv:2106.08279},
year = {2021}
}