Institute for Interdisciplinary Information Sciences, Tsinghua University, Beijing, China.
School of Science and Engineering, The Chinese University of Hong Kong, Shenzhen, Shenzhen, Guangdong, China.
Sci Rep. 2022 Mar 23;12(1):5051. doi: 10.1038/s41598-022-08879-6.
Spurred by causal structure learning (CSL) ability to reveal the cause-effect connection, significant research efforts have been made to enhance the scalability of CSL algorithms in various artificial intelligence applications. However, less effort has been made regarding the stability and the interpretability of CSL algorithms. Thus, this work proposes a self-correction mechanism that embeds domain knowledge for CSL, improving the stability and accuracy even in low-dimensional but high-noise environments by guaranteeing a meaningful output. The suggested algorithm is challenged against multiple classic and influential CSL algorithms in synthesized and field datasets. Our algorithm achieves a superior accuracy on the synthesized dataset, while on the field dataset, our method interprets the learned causal structure as a human preference for investment, coinciding with domain expert analysis.
受因果结构学习(CSL)揭示因果关系的能力的推动,研究人员在各种人工智能应用中付出了大量努力来提高 CSL 算法的可扩展性。然而,在 CSL 算法的稳定性和可解释性方面的研究却相对较少。因此,本工作提出了一种自校正机制,将领域知识嵌入到 CSL 中,通过保证有意义的输出,即使在低维但噪声较高的环境中,也能提高稳定性和准确性。所提出的算法在合成数据集和现场数据集上与多个经典且有影响力的 CSL 算法进行了对比。我们的算法在合成数据集上取得了较高的准确性,而在现场数据集上,我们的方法将学习到的因果结构解释为人类对投资的偏好,与领域专家的分析一致。