你有 100 万条 128 维向量构成的向量库,要在线查询 top-10 最近邻,延迟预算 10ms。你用 sklearn 的 KDTree 建索引,实测单次查询大约 200ms。请说明 KDTree 为什么在这里失效,并给出一个能落进 10ms 预算的方案。
面试官在考什么
考察对维度灾难的机理理解,以及在延迟/内存/召回三者之间做工程取舍的能力,而不是只会说“用 faiss”。
回答框架 · 前两步
- KDTree 失效的机理是维度灾难,不是内存不够:它靠坐标轴对齐的划分做剪枝,维度升高后几乎所有点到查询点的距离都趋于接近,超球与节点的相交判断几乎排除不掉任何子树,查询退化成近似全量扫描;经验上维度超过 20 左右树索引的优势基本消失。
- 先算清楚底数:100 万 × 128 float32 约 512MB,一次暴力扫描约 1.28e8 次乘加,CPU 上就是几十毫秒量级。所以问题的关键是减少实际比较的候选数,而不是换一个更“聪明”的树结构。
还有 3 条答案要点、 3 层追问、 4 个常见坑, 以及针对你简历的追问路径,在定制报告里。
延伸阅读
想看这个方向的整体准备路线? 进入「机器学习基础」方向页
同一能力域的其他题
- GBDT 和 XGBoost 的关系是什么?XGBoost 相比 GBDT 做了哪些工程与数学上的改进?
- 特征重要性你一般怎么看?树模型的 gain 重要性和 permutation importance 各有什么局限?
- 你在做一个点击率预估模型,训练集 2000 万条,真实正样本率是 1%。你用带截距的逻辑回归训练完之后,验证集 AUC=0.78(看起来正常),但把验证集上的预测概率求平均得到 0.29。请说出你会按什么顺序排查,每一步的判据是什么。
- 某推荐场景有 3 亿条训练样本、2 万维特征(约 95% 的取值为 0,以稀疏 one-hot 为主)。业务要求:全量重训能在 2 小时内跑完,线上单次预测 p99 < 10ms,资源只有 CPU 集群、没有 GPU。现在有三个方案:(A) 逻辑回归 + FTRL 在线学习,边服务边更新;(B) 逻辑回归 + L-BFGS 离线全量重训,每天一次;(C) 直接换成 2 层 MLP。请给出你的选型,并说清楚在什么监控信号出现时你会推翻它。
- 线上一个 LR 二分类模型,每天全量重训,离线训练 AUC 稳定在 0.790,但近 7 天线上 AUC 从 0.780 单调掉到 0.762;模型容量、特征列表、样本量都没有改动。PM 要求本周内把模型换成 XGBoost 来提效果。你作为模型负责人,第一周会做什么?请给出你的归因顺序和判断依据。
- 一个 5 万样本、50 维的二分类任务,你直接用 SVC(kernel='rbf') 训练,跑了一小时没跑完、内存也吃紧。换成 LinearSVC 后 3 分钟训完,但验证集 F1 掉了 4 个点。请说明 RBF 核在这里慢在哪,并给出 2~3 条能同时保住效果的替代方案。