Learning the Finer Things: Bayesian Structure Learning at the Instantiation Level
This work addresses the challenge of balancing memorization and generalization in machine learning for domains like gene regulatory networks, though it appears incremental as it builds on existing Bayesian Knowledge Bases and DAG learning techniques.
The authors tackled the problem of learning probabilistic graphical models in under-determined domains where accuracy is unsatisfactory, by developing a novel structure learning approach at the random variable instantiation level, which demonstrated significant improvements over Bayesian Networks on 40 benchmark datasets.
Successful machine learning methods require a trade-off between memorization and generalization. Too much memorization and the model cannot generalize to unobserved examples. Too much over-generalization and we risk under-fitting the data. While we commonly measure their performance through cross validation and accuracy metrics, how should these algorithms cope in domains that are extremely under-determined where accuracy is always unsatisfactory? We present a novel probabilistic graphical model structure learning approach that can learn, generalize and explain in these elusive domains by operating at the random variable instantiation level. Using Minimum Description Length (MDL) analysis, we propose a new decomposition of the learning problem over all training exemplars, fusing together minimal entropy inferences to construct a final knowledge base. By leveraging Bayesian Knowledge Bases (BKBs), a framework that operates at the instantiation level and inherently subsumes Bayesian Networks (BNs), we develop both a theoretical MDL score and associated structure learning algorithm that demonstrates significant improvements over learned BNs on 40 benchmark datasets. Further, our algorithm incorporates recent off-the-shelf DAG learning techniques enabling tractable results even on large problems. We then demonstrate the utility of our approach in a significantly under-determined domain by learning gene regulatory networks on breast cancer gene mutational data available from The Cancer Genome Atlas (TCGA).