LiRank: Industrial Large Scale Ranking Models at LinkedIn
Authors: name: Fedor Borisyuk; name: Mingzhou Zhou; name: Qingquan Song; name: Siyu Zhu; name: Birjodh Tiwana; name: Ganesh Parameswaran; name: Siddharth Dangi; name: Lars Hertel; name: Qiang Xiao; name: Xiaochen Hou; name: Yunbo Ouyang; name: Aman Gupta; name: Sheallika Singh; name: Dan Liu; name: Hailing Cheng; name: Lei Le; name: Jonathan Hung; name: Sathiya Keerthi; name: Ruoyan Wang; name: Fengyu Zhang; name: Mohit Kothari; name: Chen Zhu; name: Daqi Sun; name: Yun Dai; name: Xun Luan; name: Sirou Zhu; name: Zhiwei Wang; name: Neil Daftary; name: Qianqi Shen; name: Chengming Jiang; name: Haichao Wei; name: Maneesh Varshney; name: Amol Ghoting; name: Souvik Ghosh
Published: 2024-02-10 01:47:10

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.