Pages
← Pages

Basic ML Speed Round Prep

Comprehensive prep guide for Fin's Basics of ML rapid-fire round: a tiered concept checklist, worked 3-minute answers, follow-up patterns, and a rehearsal plan across classical ML, statistics, and LLM/RAG topics.

Fin's own interview guide describes this round as: "We will ask you a series of rapid-fire questions related to different aspects of ML. These won't be too deep: as a ballpark, we expect them to be around three minutes each." That's roughly 8-10 questions in 30 minutes, for a Staff/Senior ML Scientist role at an AI customer-support company.

This round tests breadth, not depth: working knowledge over academic theory, the ability to connect concepts into a system, and clarity under time pressure. Fin builds an AI support agent, so expect the panel to weight questions toward LLM and RAG system design more heavily than classical ML. This guide has five parts: a tiered concept checklist to scan the night before, worked 3-minute answers for the questions most likely to come up, the follow-up questions each worked answer tends to trigger, a rehearsal plan for practicing out loud, and a breakdown of how much prep time to spend where.

Section 1: Concept Checklist

Each pillar is split into tiers. Tier 1 is must-know for any ML basics round. Tier 2 is Fin-specific: topics an enterprise support AI company is likely to probe given its product. Tier 3 is nice-to-have depth that signals seniority but is less likely to come up directly. Under time pressure, drill Tier 1 first, then Tier 2, and treat Tier 3 as a bonus.

Pillar A: Classical ML Fundamentals (~30% of questions)

Tier 1 (must-know)

Tier 2 (Fin-specific depth)

Pillar B: Applied Statistics (~25% of questions)

Tier 1 (must-know)

Tier 2 (Fin-specific depth)

Pillar C: LLM / Modern AI Fundamentals (~45% of questions, for Fin)

Tier 1 (must-know)

Tier 2 (Fin-specific, heavily weighted)

Tier 3 (nice-to-have depth)

Section 2: Worked Examples

Eight worked answers, weighted toward the RAG, fine-tuning, and alignment topics Fin is most likely to ask about, each written to the 3-minute template: a crisp definition (15 seconds), why it matters with the core tradeoff (30 seconds), a mental model (60 seconds), a concrete example (30 seconds), then a pause, ready for follow-ups (15 seconds).

1. Can you explain DPO and why you'd use it over RLHF?

Definition. DPO is a one-stage supervised learning approach to preference alignment. Instead of training a separate reward model, it derives the optimal policy in closed form and reduces alignment to a binary classification loss.

Why it matters. Traditional RLHF requires three stages (SFT, reward model training, PPO), which is complex and requires tuning the reward model separately, and that reward model can overfit or drift. DPO solves this with a single unified objective, which is why it's become the default first alignment pass for teams without a large RL infrastructure investment.

Mental model. The optimal RLHF policy is pi* proportional to pi_ref * exp(reward / beta). Rearranging to express reward in terms of the policies, then substituting into the Bradley-Terry preference model, gives the DPO loss, which is just binary cross-entropy on the ratio of policy to reference-policy log-probabilities for the winning versus losing response. No reward model needed.

Example. Fine-tuning a 7B model on 5k preference pairs took about 4 hours on a single A100, versus PPO needing roughly 2 weeks with a critic network, for comparable quality.

Ready for follow-up. (Deeper dive: your ml-system-design-enterprise-support-ai and reinforcement-learning-in-llms posts.)

2. How would you design a retrieval system for a support knowledge base?

Definition. Retrieval for support means turning a heterogeneous knowledge base, tickets, Confluence pages, Slack threads, into a system that reliably surfaces the right passage for a user's question, fast enough for a live chat.

Why it matters. Get retrieval wrong and everything downstream, grounding, hallucination rate, resolution rate, degrades no matter how good the generation model is. The core tradeoff is exact match versus semantic match: users ask about specific error codes and policy names, but also paraphrase the same question a dozen different ways.

Mental model. Hybrid retrieval is the answer: BM25 catches exact-match tokens like error codes and product names that dense embeddings often miss, dense embeddings catch paraphrase and semantic similarity, and a cross-encoder reranker sits on top of both candidate sets to pick the final passages. This combination typically improves recall at 10 by 10-15% over dense retrieval alone. Chunking strategy matters just as much as the retrieval method: fixed-size chunks are fast to build but naive, semantic chunking respects topic boundaries, and hierarchical chunking (paragraph plus section) works best across a heterogeneous source mix. Tables inside support docs are a specific failure point, they tend to get garbled during text extraction, so table-aware parsing is worth the extra engineering.

Example. On a support KB mixing tickets and internal docs, moving from dense-only retrieval to BM25-plus-dense-plus-reranker is a realistic path to a double-digit recall at 10 improvement, with the reranker doing a lot of the heavy lifting on precision.

Ready for follow-up. (Deeper dive: your ml-system-design-enterprise-support-ai post.)

3. How do you stop a support agent from hallucinating policy details?

Definition. Hallucination here means the model states a policy, refund window, or procedure with confidence when it isn't actually grounded in a real, current source document.

Why it matters. For an AI that answers real customers, a confidently wrong policy statement is worse than a slow answer or an escalation. The tradeoff is coverage versus safety: the more the model is allowed to answer from parametric memory instead of retrieved evidence, the more fluent it sounds and the more it risks being wrong.

Mental model. The fix is layered, not a single trick. RAG grounding forces the model to cite retrieved passages rather than recall from weights. A faithfulness check, either a smaller model or a rule-based verifier, compares the draft answer against the retrieved source and flags unsupported claims before the answer ships. Confidence-based abstention lets the model say "I don't have a confident answer" and hand off, rather than guess. Underneath all of this, knowledge base rot is a root cause of hallucination that's easy to miss, if the policy the model is grounded in is six months stale, grounding doesn't help, so temporal decay scoring and contradiction detection on the KB itself matter as much as anything at generation time.

Example. A faithfulness classifier that flags claims not supported by the retrieved passage, run before the answer is shown to the customer, catches a meaningful share of hallucinated policy details that pure RAG grounding alone lets through, because retrieval being correct doesn't guarantee the generated text stayed faithful to it.

Ready for follow-up. (Deeper dive: your ml-system-design-enterprise-support-ai post.)

4. How would you fine-tune on years of support tickets without inheriting bad agent habits?

Definition. Historical support tickets are a huge, free source of SFT data, but they also contain years of shortcuts, wrong answers, and rude or unresolved conversations that a model would happily learn to imitate if trained on everything.

Why it matters. The naive approach, dump every ticket into SFT, actively hurts quality because the model learns the average behavior of past agents, not the best behavior. The tradeoff is data volume versus data quality: more data is usually better, except when a meaningful fraction of it is bad behavior you don't want reproduced.

Mental model. Filter before you fine-tune. Survival filtering removes tickets that were short, unresolved, or spam, which alone often removes 40-60% of raw volume. Quality scoring, using a reward model or heuristics like resolution time and follow-up rate, ranks the remainder and keeps only the top slice, training on the top 30% by quality typically beats training on everything. CSAT-weighted training goes a step further and up-weights examples tied to conversations the customer actually rated well, so the signal keeps improving as new, better-rated conversations come in. Where real data is thin for a specific scenario, synthetic data from a teacher model can fill the gap, but it should be down-weighted relative to real examples since it risks inheriting teacher bias and topic drift.

Example. A quality-scored subset of historical tickets, roughly 30% of the raw corpus, trained a model that outperformed one trained on the full unfiltered corpus, because the unfiltered version had learned to imitate agents' bad shortcuts along with their good ones.

Ready for follow-up. (Deeper dive: your ml-system-design-enterprise-support-ai post.)

5. How do you decide when a support agent should escalate to a human?

Definition. Escalation routing is the decision layer that sits above generation: given a user's message and the model's draft response, decide whether to send the answer, ask a clarifying question, or hand off to a human agent.

Why it matters. Every wrong escalation costs something: escalate too often and you lose the automation value the whole system exists for, escalate too rarely and low-confidence answers reach customers unchecked. The tradeoff is precision of automation versus safety net coverage, essentially the same precision-recall tradeoff from classical ML, just applied to a routing decision instead of a classifier.

Mental model. Think of it as multi-model routing with a confidence threshold. A cheap, fast model or a set of features (retrieval confidence, generation model's own log-probability, whether the query matches a known unresolved category) scores each conversation, and anything below the threshold routes to a human or to a bigger, slower model. The threshold isn't fixed, it should reflect the actual cost asymmetry: a wrong answer on a billing question costs more than a wrong answer on a documentation lookup, so thresholds can vary by topic. Latency budget matters here too, escalation decisions need to happen fast enough not to stall the conversation.

Example. A confidence threshold tuned per ticket category, rather than one global threshold, routes billing and account-security questions to humans more conservatively than general documentation questions, which better matches the real cost of getting each wrong.

Ready for follow-up. (Deeper dive: your ml-system-design-enterprise-support-ai post.)

6. How would you design this pipeline under a strict cost-per-query budget?

Definition. Cost optimization for a support AI pipeline means hitting a target cost-per-resolved-query without giving up so much quality that resolution rate or CSAT drops.

Why it matters. At Fin's scale, a few cents per query compounds fast, and the naive answer, always call the biggest, best model at every stage, is usually not affordable. The tradeoff is uniform quality versus targeted spend: not every step in the pipeline needs the frontier model.

Mental model. Spend compute where it actually buys quality, and cut it everywhere else. Query rewriting only needs a small 1-3B model, not the main generation model. Retrieval reranking should run on a shortlist, rerank only the top handful of candidates the first-stage retriever returns, not the whole corpus. Quantization (INT8, sometimes INT4) shrinks the inference footprint 2-4x for under 1% accuracy loss when calibrated well, which is close to free money on cost. Batching and prompt caching, caching the stable system-prompt prefix rather than the whole conversation, can cut cost up to 90% on the shared portion of every request. The overall pattern is a cascade: cheap components filter and narrow the problem before an expensive model ever sees it.

Example. Moving query rewriting from the main generation model to a dedicated small fine-tuned model, and reranking only the top-20 retrieved candidates instead of the full result set, are both changes that cut cost meaningfully without touching the final generation model at all.

Ready for follow-up. (Deeper dive: your ml-system-design-enterprise-support-ai and attention-everything-you-need posts.)

7. How do you evaluate the quality of an end-to-end support AI system?

Definition. End-to-end evaluation here means measuring both retrieval and generation, and tying both back to a business outcome, not just a model-level score.

Why it matters. No single metric captures whether a support agent is good. A model can score well on a generation metric while still failing customers because retrieval fed it the wrong document, or it can retrieve the right document and still fail because the generation step misrepresented it.

Mental model. Layer the evaluation by pipeline stage. Retrieval gets its own metrics: Recall at k for whether the gold document made the shortlist, MRR for how highly it ranked, NDCG for graded relevance, since a system can look fine on recall at 100 while precision at 5 is poor. Generation gets LLM-as-judge against a rubric (correctness, tone, completeness) plus periodic human review, being explicit about LLM-as-judge's known biases toward longer, more-confident, and same-family outputs, and calibrating it against human labels rather than trusting it blind. Underneath both, the metric that actually matters for the business is production behavior: resolution rate, escalation rate, CSAT, and how often a human has to correct the model after the fact.

Example. A system can post a strong LLM-as-judge score while its resolution rate stays flat, that gap is usually the tell that the judge is rewarding verbose, confident-sounding answers rather than answers that actually solved the customer's problem.

Ready for follow-up. (Deeper dive: your ml-system-design-enterprise-support-ai post.)

8. Can you explain the bias-variance tradeoff and how it shows up in a support model?

Definition. Bias is error from a model too simple to capture the real pattern (underfitting); variance is error from a model too sensitive to noise in its training data (overfitting). Total generalization error breaks down roughly into bias squared plus variance plus irreducible error.

Why it matters. It's the single tradeoff underneath almost every modeling decision: model size, regularization strength, how much data to collect, when to stop training. Getting it wrong in either direction has a concrete cost: too much bias means missing real cases, too much variance means memorizing noise that doesn't generalize.

Mental model. Picture two support models. A small model trained on too little ticket data underfits: it misses edge cases and gives generic answers because it never saw enough variety to learn the nuance. A large model fine-tuned too long on a narrow slice of tickets overfits: it starts reproducing very specific, sometimes bad, agent habits from its training set instead of generalizing to new questions. The fix set is the same one used everywhere else: regularization (LoRA rank control, weight decay), more or better data (quality-filtered tickets, not just more volume), ensembling, and validation on held-out, representative traffic to pick the right stopping point. Worth a sentence on the modern wrinkle: double descent shows that pushing capacity well past the point of fitting training data perfectly can sometimes lower test error again, so the classic U-shaped curve isn't universal in overparameterized regimes.

Example. A support model fine-tuned for too many epochs on a narrow set of billing tickets started reproducing specific phrasing quirks from a handful of training conversations, a variance failure, while a model with too little data defaulted to generic non-answers on anything outside common cases, a bias failure.

Ready for follow-up. (Deeper dive: your interview-preparation and fundamental-statistics posts.)

Section 3: Follow-Up Patterns

Every worked answer above tends to draw one of these follow-ups. Prepare the short version of each so a follow-up doesn't turn into rambling.

1. DPO

2. Retrieval system design

3. Hallucination mitigation

4. SFT from tickets

5. Routing and escalation

6. Cost optimization

7. Evaluation

8. Bias-variance

Section 4: Rehearsal Plan

The 3-minute structure. Crisp definition (15 seconds), why it matters with a concrete scenario and the core tradeoff (30 seconds), your mental model or intuitive explanation (60 seconds), a real example or benchmark number (30 seconds), then pause, ready for follow-ups (15 seconds). This is the template every worked example above follows.

Daily drill methodology.

Common failure modes

| Failure mode | Fix | | --- | --- | | Rambling without structure | Use the 5-part template every time, even under pressure | | Too mathematical, no intuition | Always pair the formula with a concrete example | | Forgetting to state assumptions | Say "assuming X" explicitly before diving in | | No answer ready for "when would you NOT use this" | Prepare the inverse case for every concept in advance | | Over-explaining basics the interviewer already knows | Assume the basics are known, go straight to the subtlety |

What "not too deep" actually means.

Section 5: Fin-Specific Weighting

Given Fin's product (a text-based AI support agent) and this being a Staff/Senior role, prep time is best spent where the questions are most likely to land.

Most likely, roughly 60% of questions

Possible, roughly 30% of questions

Unlikely, roughly 10% of questions

Read the checklist once the night before, run the daily drill on the Tier 1 and Tier 2 concepts, and go into the round ready to hit the 5-part structure on whatever comes up first.