研究人员开发了CAFE,一个新颖的自动化特征工程(AFE)框架,它将因果发现与多智能体强化学习相结合。与依赖统计启发式方法的传统AFE方法不同,CAFE利用因果结构来指导特征构建过程,旨在创建在分布变化下表现更好的更鲁棒的特征。该框架首先学习一个有向无环图来识别特征与目标变量之间的因果关系,然后采用多智能体深度Q学习架构来选择因果组和转换。在15个基准测试和Folktables数据集上的实验表明,与现有的AFE基线相比,CAFE显著提高了性能,缩短了收敛时间,并增强了对分布变化的鲁棒性。 AI
影响 这项研究通过改进AI模型从原始数据中学习的方式,尤其是在数据分布变化的场景下,可能带来更鲁棒、更高效的AI模型。
排序理由 这是一篇详细介绍自动化特征工程新方法的学术论文。[lever_c_demoted from research: ic=1 ai=1.0]
- Arun Vignesh Malarkkan
- CAFE
- causal discovery
- directed acyclic graph
- Folktables
- multi-agent reinforcement learning
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →