4 个月前

双曲蕴含锥用于学习层次嵌入

双曲蕴含锥用于学习层次嵌入

摘要

学习能够保留相关网络属性的低维嵌入图表示是机器学习中一个重要的问题类别。本文提出了一种新的方法来嵌入有向无环图。借鉴先前的研究,我们首先主张使用双曲空间,因为双曲空间在建模树状结构方面比欧几里得几何更为有效。其次,我们将层次关系视为由一系列嵌套的测地凸锥定义的部分顺序。我们证明了这些蕴含锥在欧几里得空间和双曲空间中都具有最优形状,并且该形状可以用闭式表达式表示,它们自然地定义了嵌入学习过程。实验结果表明,我们的方法在表示能力和泛化性能方面相对于近期的强基线模型有显著改进。

代码仓库

dalab/hyperbolic_cones
官方
GitHub 中提及
iesl/geometric_graph_embedding
pytorch
GitHub 中提及
dinobby/hypemo
pytorch
GitHub 中提及

基准测试

基准方法指标
link-prediction-on-wordnetHyperbolic Entailment Cones
Accuracy: 94.4

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
双曲蕴含锥用于学习层次嵌入 | 论文 | HyperAI超神经