Neural Autoregressive Flows for Markov Boundary Learning

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Nguyen, Khoa, Duong, Bao, Huynh, Viet, Nguyen, Thin
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910062524497920
author Nguyen, Khoa
Duong, Bao
Huynh, Viet
Nguyen, Thin
author_facet Nguyen, Khoa
Duong, Bao
Huynh, Viet
Nguyen, Thin
contents Recovering Markov boundary -- the minimal set of variables that maximizes predictive performance for a response variable -- is crucial in many applications. While recent advances improve upon traditional constraint-based techniques by scoring local causal structures, they still rely on nonparametric estimators and heuristic searches, lacking theoretical guarantees for reliability. This paper investigates a framework for efficient Markov boundary discovery by integrating conditional entropy from information theory as a scoring criterion. We design a novel masked autoregressive network to capture complex dependencies. A parallelizable greedy search strategy in polynomial time is proposed, supported by analytical evidence. We also discuss how initializing a graph with learned Markov boundaries accelerates the convergence of causal discovery. Comprehensive evaluations on real-world and synthetic datasets demonstrate the scalability and superior performance of our method in both Markov boundary discovery and causal discovery tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2603_20791
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Neural Autoregressive Flows for Markov Boundary Learning
Nguyen, Khoa
Duong, Bao
Huynh, Viet
Nguyen, Thin
Machine Learning
Recovering Markov boundary -- the minimal set of variables that maximizes predictive performance for a response variable -- is crucial in many applications. While recent advances improve upon traditional constraint-based techniques by scoring local causal structures, they still rely on nonparametric estimators and heuristic searches, lacking theoretical guarantees for reliability. This paper investigates a framework for efficient Markov boundary discovery by integrating conditional entropy from information theory as a scoring criterion. We design a novel masked autoregressive network to capture complex dependencies. A parallelizable greedy search strategy in polynomial time is proposed, supported by analytical evidence. We also discuss how initializing a graph with learned Markov boundaries accelerates the convergence of causal discovery. Comprehensive evaluations on real-world and synthetic datasets demonstrate the scalability and superior performance of our method in both Markov boundary discovery and causal discovery tasks.
title Neural Autoregressive Flows for Markov Boundary Learning
topic Machine Learning
url https://arxiv.org/abs/2603.20791