Classification under Nuisance Parameters and Generalized Label Shift in Likelihood-Free InferenceShow others and affiliations
2024 (English)In: Proceedings of Machine Learning Research, 2024 / [ed] Ruslan Salakhutdinov; Zico Kolter; Katherine Heller; Adrian Weller; Nuria Oliver; Jonathan Scarlett; Felix Berkenkamp, ML Research Press , 2024, Vol. 235, p. 34987-35012, article id 201670Conference paper, Published paper (Refereed)
Abstract [en]
An open scientific challenge is how to classify events with reliable measures of uncertainty, when we have a mechanistic model of the data-generating process but the distribution over both labels and latent nuisance parameters is different between train and target data. We refer to this type of distributional shift as generalized label shift (GLS). Direct classification using observed data X as covariates leads to biased predictions and invalid uncertainty estimates of labels Y. We overcome these biases by proposing a new method for robust uncertainty quantification that casts classification as a hypothesis testing problem under nuisance parameters. The key idea is to estimate the classifier's receiver operating characteristic (ROC) across the entire nuisance parameter space, which allows us to devise cutoffs that are invariant under GLS. Our method effectively endows a pretrained classifier with domain adaptation capabilities and returns valid prediction sets while maintaining high power. We demonstrate its performance on two challenging scientific problems in biology and astroparticle physics with data from realistic mechanistic models.
Place, publisher, year, edition, pages
ML Research Press , 2024. Vol. 235, p. 34987-35012, article id 201670
Series
Proceedings of Machine Learning Research, ISSN 2640-3498
National Category
Mathematics
Identifiers
URN: urn:nbn:se:ltu:diva-110133Scopus ID: 2-s2.0-85203805703OAI: oai:DiVA.org:ltu-110133DiVA, id: diva2:1902469
Conference
41st International Conference on Machine Learning (ICML 2024), Vienna, Austria, July 21-27, 2024
2024-10-012024-10-012025-10-21Bibliographically approved