Causal Bayesian Optimization via Exogenous Distribution Learning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Ren, Shaogang, Wang, Zihao, Chen, Yuzhou, Qian, Xiaoning
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917232988127232
author Ren, Shaogang
Wang, Zihao
Chen, Yuzhou
Qian, Xiaoning
author_facet Ren, Shaogang
Wang, Zihao
Chen, Yuzhou
Qian, Xiaoning
contents Maximizing a target variable as an operational objective within a structural causal model is a fundamental problem. Causal Bayesian Optimization (CBO) approaches typically achieve this either by performing interventions that modify the causal structure to increase the reward or by introducing action nodes to endogenous variables, thereby adjusting the data-generating mechanisms to meet the objective. In this paper, we propose a novel method that learns the distribution of exogenous variables-an aspect often ignored or marginalized through expectation in existing CBO frameworks. By modeling the exogenous distribution, we enhance the approximation fidelity of the data-generating structural causal models (SCMs) used in surrogate models, which are commonly trained on limited observational data. Furthermore, the ability to recover exogenous variables enables the application of our approach to more general causal structures beyond the confines of Additive Noise Models (ANMs) and single-mode Gaussian, allowing the use of more expressive priors for context noise. We incorporate the learned exogenous distribution into a new CBO method, demonstrating its advantages across diverse datasets and application scenarios.
format Preprint
id arxiv_https___arxiv_org_abs_2402_02277
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Causal Bayesian Optimization via Exogenous Distribution Learning
Ren, Shaogang
Wang, Zihao
Chen, Yuzhou
Qian, Xiaoning
Machine Learning
Maximizing a target variable as an operational objective within a structural causal model is a fundamental problem. Causal Bayesian Optimization (CBO) approaches typically achieve this either by performing interventions that modify the causal structure to increase the reward or by introducing action nodes to endogenous variables, thereby adjusting the data-generating mechanisms to meet the objective. In this paper, we propose a novel method that learns the distribution of exogenous variables-an aspect often ignored or marginalized through expectation in existing CBO frameworks. By modeling the exogenous distribution, we enhance the approximation fidelity of the data-generating structural causal models (SCMs) used in surrogate models, which are commonly trained on limited observational data. Furthermore, the ability to recover exogenous variables enables the application of our approach to more general causal structures beyond the confines of Additive Noise Models (ANMs) and single-mode Gaussian, allowing the use of more expressive priors for context noise. We incorporate the learned exogenous distribution into a new CBO method, demonstrating its advantages across diverse datasets and application scenarios.
title Causal Bayesian Optimization via Exogenous Distribution Learning
topic Machine Learning
url https://arxiv.org/abs/2402.02277