数据集Freebase-FB15k-237

数据集：

FB15k-237是Freebase的子集，包含237种关系和14k种实体。

https://download.csdn.net/download/qq_21097885/12340908

类别	数量
#Relation	237
#Entity	14,541
#Train	271,115
#Valid	17,535
#Test	20,466

相关论文：

• CP: The expression of a tensor or a polyadic as a sum of products
• DistMult: Embedding Entities and Relations for Learning and Inference in Knowledge Bases
• ComplEx: Complex Embeddings for Simple Link Prediction
• HolE: Holographic Embeddings of Knowledge Graphs
• TransE: Translating Embeddings for Modeling Multi-relational Data
• R-GCN: Modeling Relational Data with Graph Convolutional Networks
• ConvE: Convolutional 2D Knowledge Graph Embeddings

实验结果：

model	Raw MRR	Filter MRR	Filter [email protected]	Filter [email protected]	Filter [email protected]
CP [1]	0.080	0.182	0.101	0.197	0.357
DistMult [2]	0.100	0.191	0.106	0.207	0.376
ComplEx [3]	0.109	0.201	0.112	0.213	0.388
HolE [4]	0.124	0.222	0.133	0.253	0.391
TransE [5]	0.144	0.233	0.147	0.263	0.398
R-GCN [6]	0.158	0.248	0.153	0.258	0.414
ConvE [7]	-	0.316	0.239	0.350	0.491

参考文献：

[1] Hitchcock, F.L., The expression of a tensor or a polyadic as a sum of products, in Studies in Applied Mathematics 6, 1-4. 1927. p. 164–189.

[2] Yang, B., et al., Embedding Entities and Relations for Learning and Inference in Knowledge Bases, in Proceedings of the 53rd Annual Meeting of the Association for Computational Linguistics and the 7th International Joint Conference on Natural Language Processing of the Asian Federation of Natural Language Processing, ACL 2015, July 26-31, 2015, Beijing, China, Volume 1: Long Papers. 2015. p. 687–696.

[3] Trouillon, T., et al., Complex Embeddings for Simple Link Prediction, in Proceedings of the 33nd International Conference on Machine Learning, ICML 2016, New York City, NY, USA, June 19-24, 2016. 2016. p. 2071-2080.

[4] Nickel, M., L. Rosasco and T.A. Poggio, Holographic Embeddings of Knowledge Graphs, in Proceedings of the Thirtieth AAAI Conference on Artificial Intelligence, February 12-17, 2016, Phoenix, Arizona, USA. 2016. p. 1955-1961.

[5] Bordes, A., et al., Translating Embeddings for Modeling Multi-relational Data, in Advances in Neural Information Processing Systems 26: 27th Annual Conference on Neural Information Processing Systems 2013. Proceedings of a meeting held December 5-8, 2013, Lake Tahoe, Nevada, United States. 2013. p. 2787-2795.

[6] Schlichtkrull, M.S., et al., Modeling Relational Data with Graph Convolutional Networks, in The Semantic Web - 15th International Conference, ESWC 2018, Heraklion, Crete, Greece, June 3-7, 2018, Proceedings. 2018. p. 593-607.

[7] Dettmers, T., et al., Convolutional 2D Knowledge Graph Embeddings, in Proceedings of the Thirty-Second AAAI Conference on Artificial Intelligence, (AAAI-18), the 30th innovative Applications of Artificial Intelligence (IAAI-18), and the 8th AAAI Symposium on Educational Advances in Artificial Intelligence (EAAI-18), New Orleans, Louisiana, USA, February 2-7, 2018. 2018. p. 1811-1818.