ULTRA
U
ULTRA
Overview :
ULTRA is a basic model for knowledge graph reasoning. A single pre-trained ULTRA model can execute link prediction tasks on any multi-relational graph and support any entity/relationship vocabulary. Its performance surpasses many SOTA models that are trained specifically for each graph. Following the pre-training-fine-tuning paradigm of basic models, pre-trained ULTRA checkpoints can be immediately used for zero-shot reasoning on any graph, as well as further fine-tuning. ULTRA provides a unified, learnable, and transferable representation for any knowledge graph. ULTRA employs graph neural networks and a modified version of NBFNet. It does not learn specific entity and relationship embeddings for the downstream graph, but rather acquires relative relationship representations based on the interaction between relationships.
Target Users :
["Knowledge Graph Reasoning","Link Prediction","Graph Neural Networks","Pretrained Models"]
Total Visits: 474.6M
Top Region: US(19.34%)
Website Views : 58.0K
Use Cases
Perform zero-shot reasoning on the WN18RR dataset using the pre-trained checkpoint ultra_4g.pth:
python run.py -c config/transductive/inference.yaml --dataset WN18RR --epochs 0 --ckpt ckpts/ultra_4g.pth
Fine-tune the pre-trained checkpoint ultra_4g.pth on a custom knowledge graph:
python run.py -c config/custom_graph.yaml --dataset MyGraph --epochs 20 --ckpt ckpts/ultra_4g.pth
Features
Supports zero-shot reasoning and fine-tuning on any graph (including custom graphs) using provided pre-trained checkpoints
Supports multi-GPU training and inference
Can be pre-trained on custom graph mixes
Can sequentially evaluate multiple datasets
Can perform reasoning and fine-tuning on custom knowledge graphs using pre-trained checkpoints
AIbase
Empowering the Future, Your AI Solution Knowledge Base
© 2025AIbase