← Back to Index

21. Interview Prep

AI system design interviews assess your ability to architect production-grade ML systems. The focus is on trade-offs, data pipelines, scale, and practical engineering decisions — not just model accuracy.

The AI System Design Interview Framework

  1. Clarify the problem: What is the ML task? What are the inputs/outputs? What are the constraints? (latency, throughput, cost, data volume)
  2. Define success metrics: Model metrics (AUC, F1) and business metrics (CTR, conversion, revenue).
  3. Data pipeline design: Data sources, ingestion, preprocessing, feature engineering, feature store.
  4. Model architecture: Model choice (DNN, GBDT, LLM), training strategy, evaluation.
  5. Serving infrastructure: API design, caching, batch vs real-time, autoscaling, GPU vs CPU.
  6. Monitoring & feedback: Drift detection, retraining triggers, A/B testing, explainability.
  7. Failure modes & trade-offs: What breaks? How do you degrade gracefully? What trade-offs did you make?

Common AI System Design Questions

QuestionKey Areas to Cover
Design YouTube recommendationsTwo-stage retrieval + ranking, user embeddings, real-time feedback, cold start, watch time prediction
Design a fraud detection systemReal-time inference, feature store, ensemble models, latency budget, imbalanced data, explainability for compliance
Design ChatGPT (a conversational AI system)LLM serving (vLLM, continuous batching), context management, prompt caching, safety guardrails, rate limiting
Design an image search systemEmbedding generation, vector DB (HNSW), hybrid search, indexing pipeline, real-time updates
Design a content moderation pipelineMulti-modal (text + image), cascade: fast filter → slow deep model, human review queue, feedback loop
Design a personalized news feedCandidate generation (follows, topics, trending), ranking (CTR model), real-time features, diversity, A/B testing

Scaling Estimation Practice

# Practice calculating these for any system you design
Users: MAU, DAU, concurrent users
Requests: QPS average and peak
Data storage: daily, monthly, yearly growth
Bandwidth: incoming and outgoing
Training data: volume, update frequency
Model size: parameters, memory footprint (FP32 vs FP16 vs INT8)
Inference cost: $ per prediction, $ per month

Must-Know Trade-offs for Interviews

How to Present Your Design

  1. Start with a high-level block diagram (5-7 boxes).
  2. Walk through a single request end-to-end.
  3. Deep-dive into the most interesting component (usually feature pipeline or model serving).
  4. Discuss failure modes ("What happens if Redis goes down?")
  5. Summarize trade-offs and what you would prioritize next.
Exercise: Prepare and practice answering this question: "Design an AI system that generates personalized workout and meal plans for 10M users." Use the 7-step framework. Record your answer (voice or written) and check: did you cover data pipeline, model architecture, serving, monitoring, failure modes, and trade-offs? Time yourself — aim for 35—40 minutes.