Bayesian Active Learning for Censored Regression
Abstract
Bayesian active learning is based on information theoretical approaches that focus on maximising the information that new observations provide to the model parameters. This is commonly done by maximizing the Bayesian Active Learning by Disagreement (BALD) acquisition function. However, it is challenging to estimate BALD when the new data points are subject to censorship, where only clipped values of the targets are observed. To address this, we derive the entropy and the mutual information for right-censored distributions and derive the BALD objective for active learning in censored regression ( $\mathcal {C}$ C -BALD). We propose a novel modeling approach to estimate the $\mathcal {C}$ C -BALD objective and use it for active learning in the censored setting. Across a wide range of datasets and models, we demonstrate that $\mathcal {C}$ C -BALD outperforms other Bayesian active learning methods in censored regression.
Cite
Text
Hüttel et al. "Bayesian Active Learning for Censored Regression." European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases, 2025. doi:10.1007/978-3-032-05981-9_3Markdown
[Hüttel et al. "Bayesian Active Learning for Censored Regression." European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases, 2025.](https://mlanthology.org/ecmlpkdd/2025/huttel2025ecmlpkdd-bayesian/) doi:10.1007/978-3-032-05981-9_3BibTeX
@inproceedings{huttel2025ecmlpkdd-bayesian,
title = {{Bayesian Active Learning for Censored Regression}},
author = {Hüttel, Frederik Boe and Riis, Christoffer and Rodrigues, Filipe and Pereira, Francisco C.},
booktitle = {European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases},
year = {2025},
pages = {36-51},
doi = {10.1007/978-3-032-05981-9_3},
url = {https://mlanthology.org/ecmlpkdd/2025/huttel2025ecmlpkdd-bayesian/}
}