Scaling Gaussian Processes for Learning Curve Prediction via Latent Kronecker Structure
Not on the map. Why?
Abstract (arXiv)
A key task in AutoML is to model learning curves of machine learning models jointly as a function of model hyper-parameters and training progression. While Gaussian processes (GPs) are suitable for this task, na"ive GPs require O(n3m3) time and O(n2 m2) space for n hyper-parameter configurations and O(m) learning curve observations per hyper-parameter. Efficient inference via Kronecker structure is typically incompatible with early-stopping due to missing learning curve values. We impose textitlatent Kronecker structure to leverage efficient product kernels while handling missing values. In particular, we interpret the joint covariance matrix of observed values as the projection of a latent Kronecker product. Combined with iterative linear solvers and structured matrix-vector multiplication, our method only requires O(n3 + m3) time and O(n2 + m2) space. We show that our GP model can match the performance of a Transformer on a learning curve prediction task.
Key facts. Cited by 9 papers in the map (data through 7 Oct 2026). Builds on 15 of its 22 references that are in the map. First posted on arXiv 11 Oct 2024.
Builds on
15 of 22 references.
References · 15
- 2018GPyTorch: Blackbox Matrix-Matrix Gaussian Process Inference with GPU Accelerationcited by 603
- 2015Kernel Interpolation for Scalable Structured Gaussian Processes (KISS-GP)cited by 175
- 2020Efficiently Sampling Functions from Gaussian Process Posteriorscited by 125
- 2019Exact Gaussian Processes on a Million Data Pointscited by 118
- 2024Vanilla Bayesian Optimization Performs Great in High Dimensionscited by 87
- 2014Freeze-Thaw Bayesian Optimizationcited by 86
- 2020Auto-PyTorch Tabular: Multi-Fidelity MetaLearning for Efficient and Robust AutoDLcited by 67
- 2020Pathwise Conditioning of Gaussian Processescited by 63
- 2023Efficient Bayesian Learning Curve Extrapolation using Prior-Data Fitted Networkscited by 37
- 2021Bayesian Optimization with High-Dimensional Outputscited by 32
- 2023Sampling from Gaussian Process Posteriors using Stochastic Gradient Descentcited by 27
- 2024In-Context Freeze-Thaw Bayesian Optimization for Hyperparameter Optimizationcited by 26
Built on by
Citing papers per year: 2024: 1; 2025: 6; 2026: 2.
9 papers.
- 2025Completed Hyperparameter Transfer across Modules, Width, Depth, Batch and Durationcited by 22
- 2024Robust Gaussian Processes via Relevance Pursuitcited by 7
- 2025Graph Random Features for Scalable Gaussian Processescited by 4
- 2025Zero-Shot Performance Prediction for Probabilistic Scaling Lawscited by 2
- 2025Derivation of Output Correlation Inferences for Multi-Output (aka Multi-Task) Gaussian Processcited by 1
- 2025Large Language Model as Meta-Surrogate for Data-Driven Many-Task Optimization: A Proof-of-Principle Studycited by 1
- 2025Successive Halving with Learning Curve Prediction via Latent Kronecker Gaussian Processescited by 1
- 2026Spatio-Temporal Gaussian Process for Building Terrain-Incorporating Wind Power Curvescited by 1
- 2026Active Budget Allocation for Efficient Scaling Law Estimation via Surrogate-Guided Pruningcited by 0
- 2026Spatio-Temporal Gaussian Process for Building Terrain-Incorporating Wind Power Curves30 Jun 2026
- 2026Active Budget Allocation for Efficient Scaling Law Estimation via Surrogate-Guided Pruning17 May 2026
- 2025Completed Hyperparameter Transfer across Modules, Width, Depth, Batch and Duration26 Dec 2025
- 2025Zero-Shot Performance Prediction for Probabilistic Scaling Laws19 Oct 2025
- 2025Graph Random Features for Scalable Gaussian Processes3 Sep 2025
- 2025Successive Halving with Learning Curve Prediction via Latent Kronecker Gaussian Processes20 Aug 2025
- 2025Large Language Model as Meta-Surrogate for Data-Driven Many-Task Optimization: A Proof-of-Principle Study11 Mar 2025
- 2025Derivation of Output Correlation Inferences for Multi-Output (aka Multi-Task) Gaussian Process14 Jan 2025
- 2024Robust Gaussian Processes via Relevance Pursuit31 Oct 2024