Completing a member knowledge graph with Graph Neural Networks

Overview

The article discusses the completion of member knowledge graphs using Graph Neural Networks (GNNs), specifically introducing a novel model called Entity-BERT. This model aims to infer missing profile entities for LinkedIn members by leveraging existing data to enhance user recommendations and engagement.

What You'll Learn

1

How to leverage Graph Neural Networks for entity inference in knowledge graphs

2

Why using multi-layer bidirectional transformers can improve entity inference accuracy

3

How to implement self-supervised learning for predicting masked attributes in member profiles

Prerequisites & Requirements

  • Understanding of Graph Neural Networks and their applications
  • Familiarity with self-supervised learning techniques(optional)

Key Questions Answered

How does Entity-BERT improve the inference of missing entities in member profiles?
Entity-BERT enhances the inference of missing entities by using a multi-layer bidirectional transformer to capture complex interactions among existing entities. This approach allows for better aggregation of information from member profiles, leading to more accurate predictions of skills and other attributes.
What challenges exist in inferring missing entities from member profiles?
Inferring missing entities is challenging due to the reliance on explicit textual information and incomplete member inputs. Many current entity extraction methods fail to identify entities not explicitly mentioned, which can lead to gaps in the knowledge graph.
What is the training process for the Entity-BERT model?
The Entity-BERT model is trained using self-supervision, where a percentage of entities in member profiles are masked. The model learns to predict these masked attributes based on the remaining visible entities, enhancing its ability to infer missing information.
What applications have been developed using Entity-BERT?
Entity-BERT has been applied in a skills recommender system that suggests skills members might have but haven't listed, and in an ads audience expansion feature that targets users based on inferred attributes, improving ad revenue without negatively impacting user experience.

Key Statistics & Figures

Percentage of masked entities used during training
10%
During training, 10% of the entities in member profiles are masked to teach the model to predict these attributes.

Technologies & Tools

Machine Learning
Graph Neural Networks
Used for extracting information from graphs and predicting missing entities.
Machine Learning
Transformer
Utilized in the Entity-BERT model for aggregating entity interactions.

Key Actionable Insights

1
Implementing Entity-BERT can significantly enhance the accuracy of entity inference in knowledge graphs.
By adopting advanced GNN techniques and transformers, organizations can improve the relevance of recommendations made to users, leading to increased engagement and satisfaction.
2
Utilizing self-supervised learning techniques can streamline the training process for models like Entity-BERT.
This approach allows for efficient use of available data, enabling models to learn from incomplete profiles and predict missing attributes effectively.
3
Integrating advanced entity inference into user-facing applications can drive higher user engagement.
As seen with LinkedIn's skills recommender, accurately predicting and suggesting relevant skills can lead to more profile updates and interactions, benefiting both users and the platform.

Common Pitfalls

1
Relying solely on simple aggregation methods can lead to inaccurate entity predictions.
Many existing GNN models use basic averaging techniques, which fail to capture complex relationships among entities. This can result in less relevant or incorrect inferences.

Related Concepts

Graph Neural Networks
Self-supervised Learning
Entity Inference
Knowledge Graphs