

ULTRA
简介 :
ULTRA是一个知识图谱推理的基础模型。单个预训练的ULTRA模型可以在任何多关系图谱上执行链接预测任务,并支持任意实体/关系词汇。性能优于许多专门针对每个图谱进行训练的SOTA模型。遵循基础模型的预训练-微调范式,可以在任何图谱上立即使用预训练的ULTRA检查点进行零样本推理,也可以进行进一步的微调。ULTRA为任何知识图谱提供了统一的、可学习的、可转移的表示。ULTRA使用图神经网络和NBFNet的修改版本。它不学习针对下游图谱的特定实体和关系嵌入,而是基于关系之间的交互获得相对关系表示。
需求人群 :
["知识图谱推理","链接预测","图神经网络","预训练模型"]
使用场景
使用预训练检查点ultra_4g.pth在WN18RR数据集上进行零样本推理:
python run.py -c config/transductive/inference.yaml --dataset WN18RR --epochs 0 --ckpt ckpts/ultra_4g.pth
在自定义知识图谱上微调预训练检查点ultra_4g.pth:
python run.py -c config/custom_graph.yaml --dataset MyGraph --epochs 20 --ckpt ckpts/ultra_4g.pth
产品特色
可以使用提供的预训练检查点对任何图谱(包括自定义图谱)进行零样本推理和微调
支持多GPU训练和推理
可以在自定义的图谱混合上预训练ULTRA
可以顺序评估多个数据集
可以使用预训练检查点对自定义知识图谱进行推理和微调
精选AI产品推荐

Deepmind Gemini
Gemini是谷歌DeepMind推出的新一代人工智能系统。它能够进行多模态推理,支持文本、图像、视频、音频和代码之间的无缝交互。Gemini在语言理解、推理、数学、编程等多个领域都超越了之前的状态,成为迄今为止最强大的AI系统之一。它有三个不同规模的版本,可满足从边缘计算到云计算的各种需求。Gemini可以广泛应用于创意设计、写作辅助、问题解答、代码生成等领域。
AI模型
11.4M
中文精选

Liblibai
LiblibAI是一个中国领先的AI创作平台,提供强大的AI创作能力,帮助创作者实现创意。平台提供海量免费AI创作模型,用户可以搜索使用模型进行图像、文字、音频等创作。平台还支持用户训练自己的AI模型。平台定位于广大创作者用户,致力于创造条件普惠,服务创意产业,让每个人都享有创作的乐趣。
AI模型
8.0M