用梯度提升决策树与时间感知图神经网络预测人工智能研究趋势
社会与信息网络
2022-01-19 v1
摘要
Science4cast 2021 竞赛聚焦于在演化的语义网络中预测未来边，其中每个顶点代表一个人工智能概念，一对顶点间的边表示这两个概念曾在一篇科学论文中被共同研究。本文描述我们针对该竞赛的解法。我们提出两种不同方法：基于树的梯度提升方法与深度学习方法，并展示两种方法均取得有竞争力的性能。我们基于两种方法融合的最终解法在所有参赛队伍中获第 1 名。本文源代码见 https://github.com/YichaoLu/Science4cast2021。
引用
@article{arxiv.2201.05743,
title = {Predicting Research Trends in Artificial Intelligence with Gradient Boosting Decision Trees and Time-aware Graph Neural Networks},
author = {Yichao Lu},
journal= {arXiv preprint arXiv:2201.05743},
year = {2022}
}
备注
1st place solution to the Science4cast 2021 competition. Published in: 2021 IEEE International Conference on Big Data (Big Data)