ICLR2026
Revisiting Nonstationary Kernel Design for Multi-Output Gaussian Processes
Qiaochu Xu, Zi Yang, Ying Li, Michael Minyi Zhang, Pablo M. Olmos
摘要
Multi-output Gaussian processes (MOGPs) provide a Bayesian framework for modeling non-linear functions with multiple outputs, in which nonstationary kernels are essential for capturing input-dependent variations in observations. However, from a spectral (dual) perspective, existing nonstationary kernels inherit the inflexibility and over-parameterization of their spectral densities due to the restrictive spectral-kernel duality. To overcome this, we establish a generalized spectral-kernel duality that enables fully flexible matrix-valued spectral densities -albeit at the cost of quadratic parameter growth in the number of outputs. To achieve linear scaling while retaining sufficient expressiveness, we propose the multi-output lowrank nonstationary (MO-LRN) kernel: by modeling the spectral density through a low-rank matrix whose rows are independently parameterized by bivariate Gaussian mixtures. Experiments on synthetic and real-world datasets demonstrate that MO-LRN consistently outperforms existing MOGP kernels in regression, missing-data interpolation, and imputation tasks. Code is publicly available at https://github.com/KrnteXu/MO-LRN .