Graph Positional and Structural Encoder · pyg-team/pytorch_geometric#8310

(21 comments) (0 reactions) (0 assignees)Python (19,985 stars) (3,514 forks)batch import

featuregood first issuehelp wantednn

説明

🚀 The feature, motivation and pitch

It would be great to add the "GPSE" (Graph Positional and Structural Encoder) module to PyG, which acts as a trained encoder for Graph Transformers. Most importantly, we should support loading pre-trained encoders and provide them to the user, e.g., via

encoder = GPSE.from_pretrained("molpcba")

Code: https://github.com/G-Taxonomy-Workgroup/GPSE Paper: https://arxiv.org/abs/2307.07107 Twitter: https://twitter.com/RemyLau3/status/1717194030986190899

Any help on this is highly appreciated :)

Alternatives

No response

Additional context

No response

コントリビューターガイド

技術スタック: pythonpytorch
領域: machine learning
Issue 種別: feature
難度: 5
推定時間: over 1 week
活動状況: active
明確さ: clear
前提条件: PyTorchPyG basicsgraph neural networks
初心者向け度: 15
調査方針: Review the GPSE repository at https://github.com/G Taxonomy Workgroup/GPSE and the paper for architecture details. Look at existing encoder implementations in PyG (e.g., torch geometric.nn.models) for structure. Implement the GPSE module with a from pretrained method that loads pre trained weights. Ensure integration with PyG's data flow.