Search papers, labs, and topics across Lattice.
We introduce Conditional Kernel Stein Discrepancy (CKSD) to assess goodness-of-fit for conditional distributions when only the conditional target model is known, without access to covariate distributions. This is achieved by constructing an operator-valued kernel on the covariate space, enabling the lifting of unconditional KSDs to the conditional setting beyond Euclidean domains. The proposed statistic provably vanishes if and only if the conditional model matches the true conditional distribution across covariates, demonstrating empirical viability on smooth manifolds and discrete spaces.
Unconditional KSDs are lifted to the conditional setting using operator-valued kernels, enabling goodness-of-fit testing for conditional models without knowing covariate distributions.
Kernel Stein discrepancies (KSDs) provide a versatile tool for comparing distributions. One of their main applications is in quantifying the goodness-of-fit (GoF) between a data-generating distribution and a prescribed target distribution. In this work, we study the related problem of conditional GoF quantification: given only a (possibly non-normalized) conditional target model, without information on the distribution of its covariates, and samples from a joint distribution, the goal is to assess how well the conditional distribution of the samples matches the target. To tackle this setting, we present a framework that allows lifting unconditional KSDs to the conditional setting through an operator-valued kernel on the covariate space, going beyond the known Euclidean case. We establish that our suggested statistic vanishes if and only if the conditional model and the true conditional distribution agree for almost all covariates and deploy it to test conditional GoF on smooth manifolds and on discrete spaces. Our experiments on level, power, and runtime demonstrate the viability of testing on these domains using the proposed statistic.