English

LiRank: Industrial Large Scale Ranking Models at LinkedIn

Machine Learning 2024-08-08 v2 Artificial Intelligence Information Retrieval

Abstract

We present LiRank, a large-scale ranking framework at LinkedIn that brings to production state-of-the-art modeling architectures and optimization methods. We unveil several modeling improvements, including Residual DCN, which adds attention and residual connections to the famous DCNv2 architecture. We share insights into combining and tuning SOTA architectures to create a unified model, including Dense Gating, Transformers and Residual DCN. We also propose novel techniques for calibration and describe how we productionalized deep learning based explore/exploit methods. To enable effective, production-grade serving of large ranking models, we detail how to train and compress models using quantization and vocabulary compression. We provide details about the deployment setup for large-scale use cases of Feed ranking, Jobs Recommendations, and Ads click-through rate (CTR) prediction. We summarize our learnings from various A/B tests by elucidating the most effective technical approaches. These ideas have contributed to relative metrics improvements across the board at LinkedIn: +0.5% member sessions in the Feed, +1.76% qualified job applications for Jobs search and recommendations, and +4.3% for Ads CTR. We hope this work can provide practical insights and solutions for practitioners interested in leveraging large-scale deep ranking systems.

Keywords

Cite

@article{arxiv.2402.06859,
  title  = {LiRank: Industrial Large Scale Ranking Models at LinkedIn},
  author = {Fedor Borisyuk and Mingzhou Zhou and Qingquan Song and Siyu Zhu and Birjodh Tiwana and Ganesh Parameswaran and Siddharth Dangi and Lars Hertel and Qiang Xiao and Xiaochen Hou and Yunbo Ouyang and Aman Gupta and Sheallika Singh and Dan Liu and Hailing Cheng and Lei Le and Jonathan Hung and Sathiya Keerthi and Ruoyan Wang and Fengyu Zhang and Mohit Kothari and Chen Zhu and Daqi Sun and Yun Dai and Xun Luan and Sirou Zhu and Zhiwei Wang and Neil Daftary and Qianqi Shen and Chengming Jiang and Haichao Wei and Maneesh Varshney and Amol Ghoting and Souvik Ghosh},
  journal= {arXiv preprint arXiv:2402.06859},
  year   = {2024}
}
R2 v1 2026-06-28T14:44:45.827Z