Deep Multi-task Learning and Real-time Personalization for Closeup Recommendations

Pinterest Engineering
11 min readintermediate
--
View Original

Overview

This article discusses the implementation of deep multi-task learning and real-time personalization techniques for improving Closeup recommendations at Pinterest. It highlights the evolution of the ranking model, the introduction of Multi-gate Mixture of Experts (MMoE), and the development of a blending model to optimize user engagement.

What You'll Learn

1

How to implement Multi-gate Mixture of Experts (MMoE) in a multi-task learning environment

2

Why score regularization is essential for stabilizing model predictions

3

How to blend user and query features for real-time recommendation optimization

Prerequisites & Requirements

  • Understanding of deep learning concepts and multi-task learning
  • Experience with machine learning model implementation(optional)

Key Questions Answered

What is the role of Multi-gate Mixture of Experts (MMoE) in the ranking model?
MMoE allows the model to learn task relationships from data, enhancing prediction accuracy by using specialized MLP experts for different tasks. This structure reduces conflicts among tasks and improves the overall performance of the recommendation system.
How does score regularization improve model stability?
Score regularization distills knowledge from a teacher model to stabilize the predictions of the student model. By adding a regularization term to the loss function, it helps maintain consistent ranking distributions across similar models, enhancing user experience.
What are the key components of the Closeup ranking model?
The Closeup ranking model includes representation layers for feature preprocessing, a summarization layer to group similar features, a transformer mixer for self-attention, and an MMoE module that combines outputs from multiple experts for task predictions.
What improvements were observed from the online A/B experiments?
The online A/B experiments showed a 4% increase in repin volume and a 1% increase in closeup volume among users in P5 countries, confirming the effectiveness of the new ranking and blending models.

Key Statistics & Figures

Repin volume increase
4%
Observed during online A/B experiments for the Closeup recommendation system.
Closeup volume increase
1%
Also observed during online A/B experiments for the Closeup recommendation system.

Technologies & Tools

Machine Learning
Multi-gate Mixture Of Experts (mmoe)
Used to enhance task-specific learning and improve prediction accuracy.
Machine Learning
Deep Neural Networks (dnn)
Form the backbone of the ranking model for Closeup recommendations.
Machine Learning
Transformer Encoder
Preprocess user sequence signals and context features.

Key Actionable Insights

1
Implementing Multi-gate Mixture of Experts (MMoE) can significantly enhance the flexibility and accuracy of your multi-task learning models.
This approach allows for specialized learning across different tasks, which can lead to improved predictions and user engagement in recommendation systems.
2
Utilizing score regularization can help mitigate model instability and enhance user experience by ensuring consistent ranking outputs.
By distilling knowledge from a teacher model, you can stabilize predictions and reduce variability in user interactions.
3
Incorporating real-time user signals into your recommendation models can provide a more personalized experience for users.
This allows the model to adapt quickly to changing user preferences and improve engagement metrics.

Common Pitfalls

1
Failing to properly tune the blending weights can lead to suboptimal user engagement and recommendation performance.
Without robust tuning methods, the model may not adapt well to changes in user behavior or content distribution, resulting in a poor user experience.

Related Concepts

Multi-task Learning
Machine Learning Model Optimization
User Personalization Techniques