akirato / hyperbolic-embeddings

Code for hyperboloid embeddings for knowledge graph entities

Implementation for the papers:

Self-Supervised Hyperboloid Representations from Logical Queries over Knowledge Graphs, Nurendra Choudhary, Nikhil Rao, Sumeet Katariya, Karthik Subbian and Chandan Reddy, WWW 2021.

Code directory: hype_kg/codes/

ANTHEM: Attentive Hyperbolic Entity Model for Product Search, Nurendra Choudhary, Nikhil Rao, Sumeet Katariya, Karthik Subbian and Chandan Reddy, WSDM 2022.

Code directory: product_matching/

Please refer these works if you find the code useful:

@inproceedings{10.1145/3442381.3449974,
author = {Choudhary, Nurendra and Rao, Nikhil and Katariya, Sumeet and Subbian, Karthik and Reddy, Chandan K.},
title = {Self-Supervised Hyperboloid Representations from Logical Queries over Knowledge Graphs},
year = {2021},
isbn = {9781450383127},
publisher = {Association for Computing Machinery},
address = {New York, NY, USA},
url = {https://doi.org/10.1145/3442381.3449974},
doi = {10.1145/3442381.3449974},
booktitle = {Proceedings of the Web Conference 2021},
pages = {1373–1384},
numpages = {12},
keywords = {knowledge graphs, hyperbolic space, Representation learning, reasoning queries},
location = {Ljubljana, Slovenia},
series = {WWW '21}
}

@inproceedings{choudhary2022anthem,
author = {Choudhary, Nurendra and Rao, Nikhil and Katariya, Sumeet and Subbian, Karthik and Reddy, Chandan K.},
title = {ANTHEM: Attentive Hyperbolic Entity Model for Product Search},
year = {2022},
publisher = {Association for Computing Machinery},
address = {New York, NY, USA},
booktitle = {{WSDM} '22: The Fifteenth {ACM} International Conference on Web Search
               and Data Mining, Phoenix, AZ, USA, February 21-25, 2022},
location = {Phoenix, AZ, USA},
series = {WSDM '22}
}