一个十亿样本、约 1 万维稀疏特征的排序模型,要求每天全量重训并在 2 小时内产出新模型。团队里有人坚持 XGBoost(说更稳),有人主张 LightGBM(说更快)。你作为负责人怎么选、怎么验证这个选择?并说清 LightGBM 快在哪几个点上。
面试官在考什么
考察对 GBDT 各实现核心算法差异(预排序 vs 直方图、level-wise vs leaf-wise、GOSS/EFB)的理解,以及在大规模重训约束下做工程取舍的能力。
回答框架 · 前两步
- LightGBM 的核心加速:直方图分箱把特征扫描从按样本降到按箱;leaf-wise 生长在相同叶子数下损失下降更快;GOSS 对梯度大的样本保留、小的采样;EFB 捆绑稀疏互斥特征。
- 十亿样本、1 万维稀疏下,直方图 + EFB 是主导收益,预排序类方法的排序开销和内存会先崩。
还有 3 条答案要点、 3 层追问、 6 个常见坑, 以及针对你简历的追问路径,在定制报告里。
延伸阅读
想看这个方向的整体准备路线? 进入「机器学习基础」方向页
同一能力域的其他题
- 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 条能同时保住效果的替代方案。