arXiv cs.LG· Ziyuan Wang, Fredrik D. Johansson·· 25 天前AI 评分39
MotherTree:通过合成数据元学习改进决策树训练
MotherTree: Meta-learning on synthetic data improves decision tree training
AI 导读
研究人员提出表格 Transformer 模型 MotherTree,通过在合成先验上元学习决策树归纳,仅需单次前向传递即可直接输出硬轴对齐决策树。该模型预训练无需参考树监督,在基准测试中性能可媲美递归划分、全局最优树等算法生成的大小匹配树。此外,MotherTree 可作为强初始化器,经任务调优后在所有基准和样本规模上均优于从头学习模型。
来源:arXiv cs.LG · arxiv.org