GithubHelp home page GithubHelp logo

hell-to-heaven / supplementary-materials-for-lignn Goto Github PK

View Code? Open in Web Editor NEW

This project forked from doujzc/supplementary-materials-for-lignn

0.0 1.0 0.0 974 KB

C++ 14.79% Python 63.34% Cuda 14.80% Jupyter Notebook 7.07%

supplementary-materials-for-lignn's Introduction

Supplementary-materials-for-LIGNN

About the paper

We have updated the writing of the paper, including:

  • Clarifying notations in Section 2.
  • Added discussions about link prediction literature in Section 6.
  • Added discussions about 2-FWL in Appendix H.
  • Added discussions about limitations and social impacts in Section 8.

Other updates include

  • Fixed citations.
  • Title: link level $\rightarrow$ link-level
  • Section 2.2 renamed to be "Message passing for link prediction"
  • Added motivations for linearizing the layers in Section 3.2.
  • Added more citations of related works
  • ...

Major additions to the paper are colored blue.

About the additional experiments

As reviewers suggested, we provide two additional experiments, including NBFNet for knowledge graph completion and EdgeTransformer for PCQM-Contact. All the models are trained with their official implementations, and tested with the evaluation routine in the LIGNN codes.

NBFNet

NBFNet's evaluation details are different from here, e.g. when computing rankings, it samples negative triples without replacement. To run the evaluation, first follow the NBFNet's instructions to train a NBFNet model. Follow the scripts NBFNet/evaluation.ipynb to load and evaluate the model. The evaluation process in NBFNet/evaluation.ipynb is directly taken from the LIGNN code, which provides a fair comparison.

The results are as follows. Best results are bold, and secondary results are underlined.

Dataset Metric Ours NBFNet INDIGO
wn18rr-v1 e-hits@3 86.7 81.3 12.5
r-hits@3 99.5 72.7 98.4
ACC 88.8 84.0 85.7
AUC 97.5 89.9 91.2
nell995-v1 e-hits@3 54.3 51.5 39.5
r-hits@3 95.0 46.0 80.0
ACC 90.5 98.5 85.6
AUC 99.7 99.2 94.5
fb15k237-v1 e-hits@3 60.6 62.1 45.1
r-hits@3 57.4 39.0 53.1
ACC 77.8 62.2 84.3
AUC 81.9 82.5 93.4

We are sorry that we are only able to run these experiments due to the limited time.

EdgeTransformer

Run EdgeTransformer/train_ef.py to evaluate EdgeTransformer on PCQM-Contact. The codes of the EdgeTransformer model is taken from its official implementation, and the training & evaluation routine is taken from the LIGNN codes.

We have just made the codes work properly recently and are still running the experiments. We will release the results as soon as the results are out.

supplementary-materials-for-lignn's People

Contributors

doujzc avatar

Watchers

James Cloos avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.