← Openai Interview Insights

Openai·Machine Learning Engineer·Technical Phone Screen·Senior

Senior
May 2026

Summary

ML engineer screen at OpenAI, just the one question from what I can tell. Short and technical, felt more like a warmup than a deep dive.

Questions Asked (1)

Q1

If your feature set contains highly correlated features, how do you decide which ones to keep when building a model?

Technical Trade-offsData Modeling
Author's notes

I talked through a few angles: checking variance inflation, dropping one of a correlated pair based on domain knowledge, or just letting regularization handle it.

Create a free account to read the full note

AI HintsAI Generated

Suggested Approach

Start by explaining that correlation itself isn't inherently bad, but it can cause multicollinearity, making model interpretation and feature importance unreliable. Then outline a systematic process: assess correlation, consider model type, use feature selection techniques, and validate with domain knowledge. Emphasize that the decision should be driven by the model's goal and the trade-off between simplicity and performance.

Pro tip: Mention that for tree-based models, correlated features are less problematic, but for linear models, you might use regularization (Lasso) to automatically select features. Also, highlight that sometimes keeping both can improve performance if they capture different aspects, so always validate empirically.

1. Assess Correlation and Multicollinearity

Compute correlation matrix (Pearson, Spearman) and VIF scores to identify highly correlated features. Determine the severity and potential impact on the model.

2. Consider Model Type and Goal

For linear models, multicollinearity is a concern; for tree-based models, it's less so. Also, consider if interpretability or predictive performance is the priority.

3. Apply Feature Selection Techniques

Use methods like Lasso regularization, recursive feature elimination, or PCA to reduce dimensionality. Alternatively, keep the feature with higher variance or better domain relevance.

4. Validate with Domain Knowledge and Experiments

Consult domain experts to decide which feature is more meaningful. Run experiments with different feature subsets and compare model performance using cross-validation.

5. Monitor and Iterate

After deployment, monitor model performance and feature importance. Be prepared to revisit feature selection if data distribution changes.

Key Points to Mention

  • Multicollinearity and its impact on coefficient estimates and interpretability
  • Variance Inflation Factor (VIF) as a diagnostic tool
  • Regularization methods (Lasso, Ridge) for feature selection
  • Tree-based models' robustness to correlated features
  • Domain knowledge and business context in feature selection
  • Cross-validation to empirically compare feature subsets

AI-generated suggestions, not part of the candidate's original notes. May be inaccurate — verify before relying on them.