Nov, 2023

通过因果发现和领域知识融合提升神经网络性能

TL;DR我们提出了一种名为因果感知神经网络(CINN)的通用方法,用于将观测变量之间的层次化因果结构编码到神经网络中,从而提高其预测性能。通过将因果性结构化知识映射到神经网络的逐层设计中,CINN通过连续优化问题将因果关系发现转化为定向无环图(DAG)学习来系统地从观测数据中发现因果关系,并通过专用架构和定制的损失函数将发现的层次化因果结构编码到神经网络中,实现了中间节点和叶节点的联合学习,显著提高了预测性能。