JourneyFormer: Encoding Airbnb Guest Journey with Sequence Modeling
Summary
JourneyFormer is a sequence modeling solution developed and deployed at Airbnb for search ranking, addressing the complexities of long, exploratory guest sequences and sparse booking labels in a production environment. This work details crucial design considerations, including guest event selection, ID embeddings, model architecture, and label attribution, which are vital for balancing effectiveness and scalability. The authors also describe tailored strategies to accelerate model training and inference. JourneyFormer has been successfully integrated into Airbnb's production, demonstrating its impact through improved offline ranking metrics and significant gains in key business metrics via online A/B testing across two production surfaces. This highlights the practical deployment of sequence models despite inherent challenges.
Key takeaway
For MLOps Engineers deploying sequence models in complex, sparse-label environments like recommendation systems, you should prioritize robust design decisions for data handling and model architecture. Focus on optimizing guest event selection and ID embeddings to manage long, exploratory sequences effectively. Implement tailored training and inference acceleration strategies to ensure scalability. Validate your model's impact through both offline ranking metrics and online A/B testing to confirm significant business metric gains.
Key insights
Successfully deploying sequence models in production requires careful design decisions for data, architecture, and optimization.
Principles
- Balance effectiveness with scalability in production.
- Tailored strategies accelerate model training and inference.
- Offline metrics and online A/B tests validate impact.
Method
The method involves selecting guest events, designing ID embeddings, defining model architecture, and attributing labels, alongside specific training/inference acceleration strategies.
In practice
- Optimize guest event selection for long, sparse sequences.
- Implement ID embeddings for complex user behaviors.
- Use A/B testing to validate business metric gains.
Topics
- Sequence Modeling
- Recommendation Systems
- Search Ranking
- Airbnb
- Transformer Architecture
- A/B Testing
Best for: AI Engineer, Research Scientist, AI Scientist, Machine Learning Engineer, MLOps Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Takara TLDR - Daily AI Papers.