RBNets: A Reinforcement Learning Approach for Learning Bayesian Network Structure

Zuowu Zheng,Chao Wang,Xiaofeng Gao,Guihai Chen
DOI: https://doi.org/10.1007/978-3-031-43418-1_12
2023-01-01
Abstract:Bayesian networks are graphical models that are capable of encoding complex statistical and causal dependencies, thereby facilitating powerful probabilistic inferences. To apply these models to real-world problems, it is first necessary to determine the Bayesian network structure, which represents the dependencies. Classic methods for this problem typically employ score-based search techniques, which are often heuristic in nature and have limited running times and performances that do not scale well for larger problems. In this paper, we propose a novel technique called RBNets, which uses deep reinforcement learning along with an exploration strategy guided by Upper Confidence Bound for learning Bayesian Network structures. RBNets solves the highest-value path problem and progressively finds better solutions. We demonstrate the efficiency and effectiveness of our approach against several state-of-the-art methods in extensive experiments using both real-world and synthetic datasets.
What problem does this paper attempt to address?