Bootstrap-Based Layerwise Refining for Causal Structure Learning

被引:0
作者
Xiang G. [1 ,2 ]
Wang H. [1 ,2 ]
Yu K. [1 ,2 ]
Guo X. [1 ,2 ]
Cao F. [3 ]
Song Y. [1 ,2 ]
机构
[1] Hefei University of Technology, Key Laboratory of Knowledge Engineering with the Big Data of Ministry of Education, Hefei
[2] Hefei University of Technology, School of Computer Science and Information Engineering, Hefei
[3] Shanxi University, School of Computer and Information Technology, Taiyuan
来源
IEEE Transactions on Artificial Intelligence | 2024年 / 5卷 / 06期
基金
中国国家自然科学基金;
关键词
Bootstrap sampling; causal structure learning; directed acyclic graph; layerwise refining;
D O I
10.1109/TAI.2023.3329786
中图分类号
学科分类号
摘要
Learning causal structures from observational data is critical for causal discovery and many machine learning tasks. Traditional constraint-based methods first adopt conditional independence (CI) tests to learn a global skeleton layer by layer and then orient the undirected edges to obtain a causal structure. However, the reliability of these statistical tests largely depends on the quality of data samples. In real-life scenarios, the presence of data noise or limited samples often makes many CI tests unreliable at each layer in the skeleton learning phase, leading to an inaccurate skeleton. As the number of layers increases, the inaccurate skeleton will continue to impair the skeleton construction of subsequent layers. Furthermore, an unreliable skeleton hampers the skeleton orientation procedure, resulting in an unsatisfactory causal structure. In this article, we propose a Bootstrap-based layerwise refining (BLR) algorithm for causal structure learning, which includes two new procedures to solve the above problems. First, BLR utilizes a novel layerwise skeleton refining procedure to construct the global skeleton layer by layer based on the bootstrap sampling. Second, BLR employs a collective skeleton orientation procedure that incorporates scoring techniques to collectively orient the global skeleton. The experimental results show that BLR outperforms the state-of-the-art methods on the benchmark Bayesian Network datasets. © 2020 IEEE.
引用
收藏
页码:2708 / 2722
页数:14
相关论文
共 43 条
[1]  
Spirtes P., Glymour C.N., Scheines R., Heckerman D., Causation, Prediction, and Search, (2000)
[2]  
Cai R., Qiao J., Zhang Z., Hao Z., Self: Structural equational likelihood framework for causal discovery, Proc. AAAI Conf. Artif. Intell., pp. 1787-1794, (2018)
[3]  
Guo X., Yu K., Liu L., Li P., Li J., Adaptive skeleton construction for accurate DAG learning, IEEE Trans. Knowl. Data Eng., 35, 10, pp. 10526-10539, (2023)
[4]  
Triantafillou S., Lagani V., Heinze-Deml C., Schmidt A., Tegner J., Tsamardinos I., Predicting causal relationships from biological data: Applying automated causal discovery on mass cytometry data of human immune cells, Sci. Rep., 7, 1, pp. 1-11, (2017)
[5]  
Castro D.C., Walker I., Glocker B., Causality matters in medical imaging, Nat. Commun., 11, 1, pp. 1-10, (2020)
[6]  
Chen Z., Tian Z., Zhu J., Li C., Du S., C-CAM: Causal CAM for weakly supervised semantic segmentation on medical image, Proc. IEEE/CVF Conf. Comput. Vis. Pattern Recognit., pp. 11676-11685, (2022)
[7]  
Kyono T., Schaar Der M.Van, Exploiting causal structure for robust model selection in unsupervised domain adaptation, IEEE Trans. Artif. Intell., 2, 6, pp. 494-507, (2021)
[8]  
Zhang K., Gong M., Stojanov P., Huang B., Liu Q., Glymour C., Domain adaptation as a problem of inference on graphical models, Proc. Int. Conf. Neural Inf. Process. Syst., pp. 4965-4976, (2020)
[9]  
Yu K., Liu L., Li J., Ding W., Le T.D., Multi-source causal feature selection, IEEE Trans. Pattern Anal. Mach. Intell., 42, 9, pp. 2240-2256, (2020)
[10]  
Yang S., Yu K., Cao F., Liu L., Wang H., Li J., Learning causal representations for robust domain adaptation, IEEE Trans. Knowl. Data Eng., 35, 3, pp. 2750-2764, (2023)