Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Conferences, Workshops, Symposiums, and Seminars
  4. Augmenting and Tuning Knowledge Graph Embeddings
 
conference paper

Augmenting and Tuning Knowledge Graph Embeddings

Bamler, Robert
•
Salehi, Farnood  
•
Mandt, Stephan
January 1, 2020
35Th Uncertainty In Artificial Intelligence Conference (Uai 2019)
35th Uncertainty in Artificial Intelligence (UAI) Conference

Knowledge graph embeddings rank among the most successful methods for link prediction in knowledge graphs, i.e., the task of completing an incomplete collection of relational facts. A downside of these models is their strong sensitivity to model hyperparameters, in particular regularizers, which have to be extensively tuned to reach good performance [Kadlec el al, 2017]. We propose an efficient method for large scale hyperparameter tuning by interpreting these models in a probabilistic framework. After a model augmentation that introduces perentity hyperparameters, we use a variational expectation-maximization approach to tune thousands of such hyperparameters with minimal additional cost. Our approach is agnostic to details of the model and results in a new state of the art in link prediction on standard benchmark data.

  • Details
  • Metrics
Type
conference paper
Web of Science ID

WOS:000722423500046

Author(s)
Bamler, Robert
Salehi, Farnood  
Mandt, Stephan
Date Issued

2020-01-01

Publisher

JMLR-JOURNAL MACHINE LEARNING RESEARCH

Publisher place

San Diego

Published in
35Th Uncertainty In Artificial Intelligence Conference (Uai 2019)
Series title/Series vol.

Proceedings of Machine Learning Research

Volume

115

Start page

508

End page

518

Subjects

Computer Science, Artificial Intelligence

•

Computer Science, Theory & Methods

•

Mathematics, Applied

•

Statistics & Probability

•

Computer Science

•

Mathematics

Editorial or Peer reviewed

REVIEWED

Written at

EPFL

EPFL units
INDY2  
Event nameEvent placeEvent date
35th Uncertainty in Artificial Intelligence (UAI) Conference

Tel Aviv, ISRAEL

Jul 22-25, 2019

Available on Infoscience
December 18, 2021
Use this identifier to reference this record
https://infoscience.epfl.ch/handle/20.500.14299/183888
Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés