跳到正文
Hugging Face Daily Papers·· 21 天前AI 评分32

TREVIS:用 Transformer 变分自编码器学习稀疏决策树

Learning Sparse Decision Trees via Transformer Variational Auto-Encoders

阅读原文

本站未展示全文,请前往来源网站阅读。

AI 导读

TREVIS 通过 Tree Transformer 变分自编码器(TTVAE)的隐空间探索来学习决策树,将离散搜索空间映射为连续空间,借助可微代理模型实现基于梯度的优化。实验显示,TREVIS 在预测性能上匹配现有近最优算法,同时提升了决策树的结构稀疏性。该工作已被 ICDM 2026 接收。

来源:Hugging Face Daily Papers · arxiv.org