研究
软件工程中的机器学习方法
← 返回到实验室
Large-scale pre-training of graph neural networks for ML4SE tasks
该项目研究图神经网络(GNNs)如何在源代码上进行预训练。 这是一个由几个部分组成的项目:
一个用于从不同语言的源代码中挖掘图形表示的工具。
为各种 ML4SE 任务和预培训目标实施 GNN。 我们在 PyTorch-Geometric 库的基础上实现并评估了 8 种类型的 GNN,并考虑到了它们的扩展性。
构建一个框架/可配置的管道,以方便 ML4SE 任务的实验。 该框架已经可用。
对 GNN 架构和培训目标提出新的改进建议。
参与者
Egor Bogomolov
Olga Petrova
Mikhail Evtikheev