跳转到主要内容
ToolPotion
AI 顾问
提交(免费)

CompGCN

AI GitHub 仓库AI 模型与大语言模型开源 · Linux, macOS, Windows

CompGCN 是一个基于组合的多关系图卷积网络的 PyTorch 实现。它能够实现先进的图表示学习,用于知识图谱链接预测等任务,提供灵活的组合操作和与各种评分函数的兼容性。

查看仓库
分享
查看仓库
分享

CompGCN(Composition-Based Multi-Relational Graph Convolutional Networks,基于组合的多关系图卷积网络)是一种专为多关系图表示学习设计的 AI 模型。该 PyTorch 实现由 malllabiisc 开发,并在 ICLR 2020 上发表,其重点在于增强图卷积网络在知识图谱链接预测等任务中的性能。

阅读更多

CompGCN 的核心创新在于其组合操作,用 φ(·) 表示。给定节点和关系嵌入,CompGCN 在节点邻域的每个边上应用此组合函数。然后,将生成的组合嵌入与原始关系和逆关系的特定滤波器进行卷积。该模型聚合来自邻居的消息来更新中心节点的嵌入,并使用专用的权重矩阵转换关系嵌入。这种方法能够为复杂的关系数据提供更丰富的表示。

CompGCN 与 PyTorch 1.0 和 Python 3.x 兼容。该项目提供了 FB15k-237 和 WN18RR 等数据集,这些数据集常用于知识图谱链接预测,并包含在 data 目录中。依赖项的安装通过 `requirements.txt` 文件进行管理。该存储库包含用于重现报告结果的脚本和命令,展示了其在 TransE、DistMult 和 ConvE 等不同评分函数以及减法、乘法和圆周相关等各种组合操作下的应用。

该模型对于处理大规模知识图谱和复杂关系结构的研究人员和实践者尤其有价值。它有效处理多关系数据的能力使其适用于需要深入理解数据中连接和关系的应用。该项目在 GitHub 上的开源性质有助于图表示学习领域的协作和进一步发展。

  • 类别AI 模型与大语言模型
  • 定价模式开源
  • 部署云端, 自托管
  • 平台Linux, macOS, Windows

  • 基于组合的多关系图卷积网络

  • 支持多种组合操作:减法、乘法、圆周相关

  • 兼容 PyTorch 1.0 和 Python 3.x

  • 包含用于知识图谱链接预测的数据集(FB15k-237, WN18RR)

  • 链接预测任务的可复现结果

  • 为原始关系和逆关系使用特定滤波器

  • 聚合邻居消息以更新节点嵌入

  • 使用专用权重矩阵转换关系嵌入

  • GitHub 上提供开源实现

  1. 设置环境:使用 `requirements.txt` 安装依赖项。

  2. 准备数据:执行 `./setup.sh` 进行数据集提取和文件夹设置。

  3. 配置训练:选择评分函数、组合操作和超参数。

  4. 运行实验:使用指定的参数执行 `run.py` 进行链接预测。

  5. 重现结果:使用提供的命令复制报告的性能。

  6. 集成模型:为自定义应用调整模型架构和训练脚本。

  • 知识图谱链接预测
  • 图表示学习
  • 多关系数据分析
  • 推荐系统
  • 语义网络分析

CompGCN 评价

加载中...

与 CompGCN 类似的热门AI工具

关系图卷积网络

AI GitHub 仓库

该项目提供了关系图卷积网络 (R-GCNs) 的 Keras 实现。它专为有向关系图上的半监督节点分类而设计。该代码有助于复现论文“使用图卷积网络对关系数据进行建模”中的实体分类结果。

AI 模型与大语言模型
查看工具

AI GitHub 仓库

该项目提供了关系图卷积网络 (R-GCNs) 的 Keras 实现。它专为有向关系图上的半监督节点分类而设计。该代码有助于复现论文“使用图卷积网络对关系数据进行建模”中的实体分类结果。

AI 模型与大语言模型
查看 AI GitHub 仓库 Top 250

LightGCN

AI GitHub 仓库

LightGCN 是一种简化的图卷积网络,专为推荐系统设计。它专注于协同过滤的核心邻域聚合组件,提供了一种从交互数据中学习用户和物品表示的高效方法。

AI 模型与大语言模型
查看工具

AI GitHub 仓库

简化的图卷积网络,专为推荐系统设计。它专注于协同过滤的核心邻域聚合组件,提供了一种从交互数据中学习用户和物品表示的高效方法。LightGCN 是一种简化的图卷积网络,专为推荐系统设计。它专注于协同过滤的核心邻域聚合组件,提供了一种从交互数据中学习用户和物品表示的高效方法。

AI 模型与大语言模型
查看 AI GitHub 仓库 Top 250

简化图卷积网络

AI GitHub 仓库

此存储库提供了“简化图卷积网络”论文的官方实现。它提供了一个简化的图卷积网络 (GCN) 模型,该模型消除了非线性并折叠了权重矩阵,从而得到了一个线性模型,该模型在显著缩短的训练时间内实现了具有竞争力的性能。

AI 模型与大语言模型
查看工具

AI GitHub 仓库

此存储库提供了“简化图卷积网络”论文的官方实现。它提供了一个简化的图卷积网络 (GCN) 模型,该模型消除了非线性并折叠了权重矩阵,从而得到了一个线性模型,该模型在显著缩短的训练时间内实现了具有竞争力的性能。

AI 模型与大语言模型
查看 AI GitHub 仓库 Top 250

图卷积网络

AI GitHub 仓库

图卷积网络(GCN)是一种专门用于处理图结构数据的神经网络。它将卷积神经网络(CNN)的概念推广到图结构数据,从而在半监督学习和图嵌入等领域实现强大的应用。该模型为分析复杂网络数据提供了一种新颖的方法。

其他 AI 工具
查看工具

AI GitHub 仓库

图卷积网络(GCN)是一种专门用于处理图结构数据的神经网络。它将卷积神经网络(CNN)的概念推广到图结构数据,从而在半监督学习和图嵌入等领域实现强大的应用。该模型为分析复杂网络数据提供了一种新颖的方法。

其他 AI 工具
查看 AI GitHub 仓库 Top 250

SEAL 链接预测

AI GitHub 仓库

SEAL 是一个新颖的链接预测框架,它将问题转化为子图分类任务。它提取 h-hop 包络子图和节点信息矩阵,并将它们输入图神经网络。SEAL 同时从图结构和潜在/显式节点特征中学习,以实现准确的链接预测。

AI 模型与大语言模型
查看工具

AI GitHub 仓库

新颖的链接预测框架,它将问题转化为子图分类任务。它提取 h-hop 包络子图和节点信息矩阵,并将它们输入图神经网络。SEAL 同时从图结构和潜在/显式节点特征中学习,以实现准确的链接预测。SEAL 是一个新颖的链接预测框架,它将问题转化为子图分类任务。它提取 h-hop 包络子图和节点信息矩阵,并将它们输入图神经网络。SEAL 同时从图结构和潜在/显式节点特征中学习,以实现准确的链接预测。

AI 模型与大语言模型
查看 AI GitHub 仓库 Top 250

Temporal Graph Networks

AI GitHub 仓库

Temporal Graph Networks (TGNs) 提供了一个通用的、高效的动态图深度学习框架。这个基于 Python 的库利用新颖的内存模块和基于图的运算符,在计算效率和演进图结构上的预测准确性方面优于现有方法。

AI 模型与大语言模型
查看工具

AI GitHub 仓库

提供了一个通用的、高效的动态图深度学习框架。这个基于 Python 的库利用新颖的内存模块和基于图的运算符,在计算效率和演进图结构上的预测准确性方面优于现有方法。Temporal Graph Networks (TGNs) 提供了一个通用的、高效的动态图深度学习框架。这个基于 Python 的库利用新颖的内存模块和基于图的运算符,在计算效率和演进图结构上的预测准确性方面优于现有方法。

AI 模型与大语言模型
查看 AI GitHub 仓库 Top 250

Neural Graph Collaborative Filtering (NGCF) 是一个利用图神经网络的推荐框架。它通过在用户-物品二部图中进行高阶连通性嵌入传播,显式地编码协同信号。此实现基于 SIGIR 2019 的论文。

AI 模型与大语言模型
查看工具

利用图神经网络的推荐框架。它通过在用户-物品二部图中进行高阶连通性嵌入传播,显式地编码协同信号。此实现基于 SIGIR 2019 的论文。Neural Graph Collaborative Filtering (NGCF) 是一个利用图神经网络的推荐框架。它通过在用户-物品二部图中进行高阶连通性嵌入传播,显式地编码协同信号。此实现基于 SIGIR 2019 的论文。

AI 模型与大语言模型
查看 AI GitHub 仓库 Top 250

Vision GNN (ViG)

AI GitHub 仓库

Vision GNN (ViG) 是华为诺亚方舟实验室开发的 Vision Graph Neural Networks 的 PyTorch 实现。它为图像处理任务提供了高效的 AI 主干网络,包括 ViG 和 Pyramid ViG 架构的预训练模型和训练脚本。

AI 模型与大语言模型
查看工具

AI GitHub 仓库

华为诺亚方舟实验室开发的 Vision Graph Neural Networks 的 PyTorch 实现。它为图像处理任务提供了高效的 AI 主干网络,包括 ViG 和 Pyramid ViG 架构的预训练模型和训练脚本。Vision GNN (ViG) 是华为诺亚方舟实验室开发的 Vision Graph Neural Networks 的 PyTorch 实现。它为图像处理任务提供了高效的 AI 主干网络,包括 ViG 和 Pyramid ViG 架构的预训练模型和训练脚本。

AI 模型与大语言模型
查看 AI GitHub 仓库 Top 250
1/8