arXiv · Machine Learning· Arun Vignesh Malarkkan, Wangyang Ying, Yanjie Fu·2026-10-28 12:00· 19 小时前AI 评分44

CAFE：以因果引导的多智能体强化学习实现自动特征工程

CAFE: Causally-Guided Automated Feature Engineering with Multi-Agent Reinforcement Learning

AI 导读

CAFE 将自动特征工程建模为因果引导的序贯决策过程，并通过多智能体强化学习构造特征。在15个公开基准上，该方法显著优于最强自动特征工程基线（Wilcoxon p<0.001）。在受控协变量偏移和自然分布偏移下，相比非因果多智能体基线，CAFE 将分布外性能下降最多降低4倍。