arXiv · AI· Arun Vignesh Malarkkan, Wangyang Ying, Yanjie Fu·· 1 天前AI 评分28
CAFE：结合多智能体强化学习与因果引导的自动特征工程
CAFE: Causally-Guided Automated Feature Engineering with Multi-Agent Reinforcement Learning
AI 导读
论文提出 CAFE 框架，将自动特征工程（AFE）重构为因果引导的顺序决策过程，结合因果发现与强化学习。 该框架分两阶段：先学习稀疏有向无环图获取软因果先验，再用级联多智能体深度 Q 学习选择因果组及算子。 在 15 个基准上显著优于最强 AFE 基线（Wilcoxon p<0.001），在分布偏移下性能下降幅度最多减少 4 倍。
来源：arXiv · AI · arxiv.org