Design an Experiment to Evaluate New Recommendation Model
MetaData Scientist · OnsiteLast updated
This problem tests the ability to design an end-to-end A/B test framework for a machine learning model update, focusing on metric selection, causal validity, and post-experiment analysis.
Scenario
You are a Data Scientist at a social media company. The engineering team has developed a new recommendation model for the 'Feed' section that aims to increase long-term user engagement. You need to design an experiment to evaluate whether this model should be launched globally.
Requirements
- Metric Definition: Define both the primary evaluation metric and guardrail metrics to ensure the model does not negatively impact the ecosystem.
- Experiment Design: Describe the randomization unit and how you would handle potential network interference (spillover effects) between treatment and control groups.
- Analysis Plan: Explain the statistical approach to determine significance, specifically how you would account for novelty effects (where users click more simply because the interface changed).
- Validation: Propose a method to verify if the experiment results are internally consistent before making a go/no-go decision.
Constraints
- Assume the platform has 100 million daily active users.
- Assume your model influences both 'Time Spent' (a proxy for engagement) and 'Ad Revenue' (a core business metric).
- Consider the trade-off between short-term gains and long-term user retention.
Hints
Hint 1
Consider using Ratio metrics for engagement to normalize for differences in user activity levels across buckets.
Hint 2
Think about how user-level randomization might be insufficient if users interact heavily with shared content, and explore cluster or geographic randomization as alternatives.
Hint 3
To address novelty effects, analyze the treatment effect trend over time to see if the uplift decays as the new model becomes 'familiar' to the user.
Follow-up questions
- How would you adjust your analysis if the treatment effect was positive for power users but negative for casual users?
- If your primary metric shows a significant lift but guardrail metrics like 'report rate' or 'block rate' also increase, how do you decide whether to launch?
- How would you design a 'holdback' experiment to measure the long-term cumulative impact of the recommendation model on churn?
Solution walkthrough is for members
Get the full walkthrough with reference code and complexity analysis, plus Kitty Coach at 300 messages a day. Hints, the code runner and community answers stay free.