ICML2026

Exact and Approximate Algorithms for Polytree Learning

Juha Harviainen, Frank Sommer, Manuel Sorge

摘要

Polytrees are a subclass of Bayesian networks that seek to capture the conditional dependencies between a set of nn variables as a directed forest and are motivated by their more efficient inference and improved interpretability. Since the problem of learning the best polytree is NP-hard, we study which restrictions make it more tractable by considering for example in-degree bounds, properties of score functions measuring the quality of a polytree, and approximation algorithms. We devise an algorithm that finds the optimal polytree in time O((2+ϵ)n)\mathcal{O}((2+\epsilon)^n) for arbitrarily small ϵ>0\epsilon > 0 and any constant in-degree bound kk, improving over the fastest previously known algorithm of time complexity O(3n)\mathcal{O}(3^n). We further give polynomial-time algorithms for finding a polytree whose score is within a factor of kk from the optimal one for arbitrary scores and a factor of 22 for additive ones. Many of the results are complemented by (nearly) tight lower bounds for either the time complexity or the approximation factors.