Scalable Bayesian Network Structure Learning Using Tsetlin Machine to Constrain the Search Space

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Dumbre, Kunal, Jiao, Lei, Granmo, Ole-Christoffer
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917101409665024
author Dumbre, Kunal
Jiao, Lei
Granmo, Ole-Christoffer
author_facet Dumbre, Kunal
Jiao, Lei
Granmo, Ole-Christoffer
contents The PC algorithm is a widely used method in causal inference for learning the structure of Bayesian networks. Despite its popularity, the PC algorithm suffers from significant time complexity, particularly as the size of the dataset increases, which limits its applicability in large-scale real-world problems. In this study, we propose a novel approach that utilises the Tsetlin Machine (TM) to construct Bayesian structures more efficiently. Our method leverages the most significant literals extracted from the TM and performs conditional independence (CI) tests on these selected literals instead of the full set of variables, resulting in a considerable reduction in computational time. We implemented our approach and compared it with various state-of-the-art methods. Our evaluation includes categorical datasets from the bnlearn repository, such as Munin1, Hepar2. The findings indicate that the proposed TM-based method not only reduces computational complexity but also maintains competitive accuracy in causal discovery, making it a viable alternative to traditional PC algorithm implementations by offering improved efficiency without compromising performance.
format Preprint
id arxiv_https___arxiv_org_abs_2511_19273
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Scalable Bayesian Network Structure Learning Using Tsetlin Machine to Constrain the Search Space
Dumbre, Kunal
Jiao, Lei
Granmo, Ole-Christoffer
Machine Learning
The PC algorithm is a widely used method in causal inference for learning the structure of Bayesian networks. Despite its popularity, the PC algorithm suffers from significant time complexity, particularly as the size of the dataset increases, which limits its applicability in large-scale real-world problems. In this study, we propose a novel approach that utilises the Tsetlin Machine (TM) to construct Bayesian structures more efficiently. Our method leverages the most significant literals extracted from the TM and performs conditional independence (CI) tests on these selected literals instead of the full set of variables, resulting in a considerable reduction in computational time. We implemented our approach and compared it with various state-of-the-art methods. Our evaluation includes categorical datasets from the bnlearn repository, such as Munin1, Hepar2. The findings indicate that the proposed TM-based method not only reduces computational complexity but also maintains competitive accuracy in causal discovery, making it a viable alternative to traditional PC algorithm implementations by offering improved efficiency without compromising performance.
title Scalable Bayesian Network Structure Learning Using Tsetlin Machine to Constrain the Search Space
topic Machine Learning
url https://arxiv.org/abs/2511.19273