Mining human preference via self-correction causal structure learning.

Journal: Scientific reports
PMID:

Abstract

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.

Authors

  • Jian Sun
    Department Of Computer Science, University of Denver, 2155 E Wesley Ave, Denver, Colorado, 80210, United States of America.
  • Chenye Wu
    School of Science and Engineering, The Chinese University of Hong Kong, Shenzhen, Shenzhen, Guangdong, China. chenyewu@yeah.net.
  • Weihua Peng
    Baidu, International Technology (Shenzhen) Co., Ltd, Shenzhen, China.
  • Jiayan Huang
    Baidu Inc., Shenzhen, Guangdong, China.
  • Cuiyun Han
    Baidu Inc., Shenzhen, Guangdong, China.
  • Yong Zhu
    Jiangsu Key Laboratory of Regional Specific Resource Pharmaceutical Transformation, Huaiyin Institute of Technology, Huai'an 223003, Jiangsu, P. R. China.
  • Yajuan Lyu
    Baidu Inc., Beijing, China.