DICS: Data-Informed Centroid Splitting for Decision Tree Classifiers
DICS:面向决策树分类器的数据驱动质心分裂方法
🧭 决策树分裂搜索加速 · 训练效率优化
Department of Mathematical Sciences, University of Texas at El Paso
用聚类构建类感知的紧凑候选分裂集,大幅缩小分裂搜索空间,加速决策树训练且精度不降
📌 工作参考价值(4 点)
- 在电商/广告排序常用的 GBDT(LightGBM/XGBoost)训练中,可借鉴 DICS 思路:对高基数数值或类别特征先做聚类,用聚类质心作为候选分裂点,替代逐值扫描或直方图分桶的固定候选,有潜力进一步降低分裂增益计算量。
- 类感知的质心分裂能保留判别信息,适合处理类别不平衡或特征与标签强相关的场景,例如 CTR 预估中的用户行为计数特征、商品价格等连续特征。
- 该方法可作为特征预处理步骤集成到现有树模型训练框架中,对训练数据采样聚类得到候选分裂集,再交给树模型搜索,工程实现上只需增加一次聚类开销,换来后续所有节点分裂搜索空间减小。
- 注意 DICS 的理论保证依赖假设,实际落地时需在小规模数据上验证精度损失;若精度敏感,可结合已有直方图方法做混合候选集。