The TWIML AI Podcast
The TWIML AI Podcast

Productive Machine Learning at LinkedIn with Bee-Chung Chen - TWiML Talk #200

In this episode of our AI Platforms series, we’re joined by Bee-Chung Chen, Principal Staff Engineer and Applied Researcher at LinkedIn. Bee-Chung and I caught up to discuss LinkedIn’s internal AI automation platform, Pro-ML. Bee-Chung breaks down some of the major pieces of the pipeline, LinkedIn’s

Featured Speakers

Bichung Chen Guest

Topics Discussed

Episode Summary

Executive Summary: LinkedIn’s Bichung Chen explains how the company is industrializing machine learning through Pro-ML, a platform designed to raise developer productivity across the full ML lifecycle—from shared feature access and model construction to deployment, monitoring, and experimentation. The conversation highlights LinkedIn’s scale, its open-source tooling like Photon ML and Tony, and the operational challenges of large personalized models in production.

Main Topics: Bichung Chen’s background and LinkedIn’s ML mission (Priority: 5/5): Chen traces his path from early neural networks and database-oriented ML research to recommender systems at Yahoo and later LinkedIn, where he now focuses on ensuring the company has the right ML/AI technology and that developers can use it productively. ML applications across LinkedIn products (Priority: 5/5): Machine learning powers feed ranking, sponsored content ranking, people recommendations, recruiter search, and sales lead discovery, using LinkedIn’s rich member, company, job, and interaction data. Evolution of ML tooling: Hadoop, Spark, Photon ML, and Tony (Priority: 5/5): LinkedIn moved from Hadoop-based ML to Spark, then built Photon ML as a replacement for Spark MLlib due to scaling limitations, and Tony for distributed TensorFlow cluster management on Hadoop/YARN. Pro-ML as an end-to-end productivity platform (Priority: 5/5): Pro-ML is positioned as a productivity initiative covering the full ML lifecycle: feature marketplace, model creation, deployment, inference, and health assurance, with the goal of doubling ML developer productivity. Feature marketplace and feature abstraction (Priority: 4/5): LinkedIn created a feature marketplace to reduce duplicated feature pipelines and simplify feature access via declarative definitions, source location abstraction, and reuse across teams, while limiting feature dependency complexity. Model deployment, inference, and operational complexity (Priority: 5/5): LinkedIn deals with very large models, including deep personalized models with tens of billions of parameters, requiring partitioned deployment across key-value stores, scoring services, and indexes plus controlled rollout through a centralized release process. Monitoring, experimentation, and future automation (Priority: 4/5): Health assurance includes feature/data quality monitoring, drift detection, explainability, and A/B testing. LinkedIn is also exploring automatic retraining and better offline experiment management.

Key Arguments: ML is embedded in nearly every LinkedIn product, so the platform must support many use cases rather than a single workflow. LinkedIn’s unique data—member profiles, connections, activity, jobs, companies, and content—creates strong opportunities for personalized ML. A shared feature abstraction is essential because many teams otherwise create duplicated, hard-to-manage pipelines. Large-scale ML at LinkedIn requires custom tooling beyond standard Spark MLlib because of data volume and scalability limits. Pro-ML should improve productivity by offering end-to-end lifecycle support rather than isolated point solutions. Semantic typing of tensors/features can help validate DAGs and make models easier to understand and debug. Deployment is difficult because large personalized models cannot fit in one place and must be split across multiple production systems. Monitoring must cover both feature consistency and distribution shift to keep models healthy in production. A/B testing remains the main mechanism for evaluating model changes and proving value in production. LinkedIn is still early in measuring productivity, using successful experiments per engineer as a key metric.

Data Points: LinkedIn member base: more than 500 million members - Chen cites LinkedIn’s scale as part of what makes its data and ML use cases unique. Company count: more than 30 million companies - Part of the platform’s rich structured data used in ML applications. AI Academy duration: 5 weeks - Training program to help software engineers learn machine learning and apply it to their own applications. Pro-ML goal: double productivity of ML developers - Chen describes Pro-ML as an initiative to significantly raise developer productivity. Model size: tens of billions of parameters - Large personalized models can require deployment across multiple systems rather than a single service. A/B testing usage: widely used in all products - LinkedIn uses A/B testing for ML and non-ML product changes alike. Productivity metric: number of successful experiments per engineer - Current internal measure for assessing ML tooling impact. Photon ML adoption: mainly used internally for model training - LinkedIn uses Photon ML instead of MLlib because MLlib did not scale to their data.

Pivotal Quotes: "Pro-ML is an initiative to double productivity of ML developers at LinkedIn." — Bichung Chen: Defines the purpose of LinkedIn’s machine learning platform. "There are different teams, they develop very similar features, there's very little leverage across the teams." — Bichung Chen: Explains why LinkedIn created the feature marketplace. "In order to be able to model each individual user's behavior, in the model training process, we generate model parameters for each of the individual users." — Bichung Chen: Illustrates why LinkedIn’s models become extremely large and hard to deploy.

Implications: LinkedIn is building a full-stack ML platform to make large-scale personalization easier, safer, and more reusable. For the industry, it shows that ML success increasingly depends on lifecycle tooling, governance, and operationalization—not just model quality.

🔓 Sign Up for Unlimited Episode Search

About The TWIML AI Podcast

View all episodes from The TWIML AI Podcast