TREVIS 用树 Transformer VAE 学习稀疏且高精度的决策树

Giacomo Fidone · hf · 2026-09-15

Giacomo Fidone 等发布 TREVIS,一种通过 Transformer 变分自编码器学习稀疏决策树的新方法。

方法核心:决策树本质上是离散结构,直接优化困难。TREVIS 将其映射到连续隐空间,用树结构 transformer 变分自编码器对连续隐表示做优化,再解码为决策树,从而联合优化预测准确率与结构稀疏性——在保持树模型可解释性的同时提升性能。

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →