基于因果反馈的缺失数据集因果关系发现

摘要

因果关系发现是因果推断的重要部分，其目标是揭示数据内在的生成机制，并以有向无环图的形式表示.然而关于因果关系发现，现有方法很少考虑到观察数据存在缺失值的情况.在实际场景中，大量数据集存在缺失值，因此估计缺失数据集中的因果关系成为一个亟需解决的问题.本文提出了一种新的基于因果反馈的算法实现关于缺失数据集的因果关系发现，其中生成对抗网络被用于估计缺失数据集的分布，并利用基于Actor-Critic的因果关系发现模块搜索最优因果图，设计了一个基于扩展贝叶斯信息准则的自定义奖励函数，引入分类误差引导模型加速探索过程，提升模型稳定性.在模拟数据和真实数据上进行的大量实验结果表明，本文提出的方法在不同数据缺失率下优于现有方法 .

关键词

深度学习 / 缺失数据补全 / 因果关系发现 / 有向无环图

基金资助

四川大学博士后交叉学科项目（10822041A2137）; 四川省重点实验室开放课题（SCITLAB-20001）; 四川省科技厅项目（2023YFG0112,2022YFG0034)