arXiv ID:
2606.26337
EMA-FS:通过增益感知特征筛选加速GBDT训练 / EMA-FS: Accelerating GBDT Training via Gain-Informed Feature Screening
1️⃣ 一句话总结
本文提出了一种名为EMA-FS的方法,通过追踪每个特征在训练过程中的历史贡献(增益),只保留最有用的特征来构建直方图,从而在不显著影响预测效果的前提下,将LightGBM等梯度提升决策树(GBDT)模型的训练速度提升1.3至2.6倍,尤其适用于中高维度的稠密数据。