arXiv Machine Learning

Don't Get Your Kroneckers in a Twist: Gaussian Processes on High-Dimensional Incomplete Grids

The paper introduces CUTS-GPR, a method for exact Gaussian Process Regression (GPR) in high-dimensional settings. It achieves near-linear scaling in the number of training points and low-order polynomial scaling in dimensionality by using an additive kernel on an incomplete grid, enabling efficient kernel matrix-vector products. Benchmarks demonstrate scalability to billions of data points and thousands of dimensions, with full GPR runs on up to 4.5 million points in 500 dimensions and practical application to ten high-dimensional potential energy surfaces completed in hours.

arXiv Machine Learning
Jul 27

gp2Scale: A Class of Compactly Supported Non-Stationary Kernels and Distributed Computing for Exact Gaussian Processes on 10 Million Data Points

arXiv:2512. 06143v2 Announce Type: replace Abstract: Despite a large corpus of recent work on scaling up Gaussian processes, a stubborn trade-off between computational speed, prediction and uncertainty quantification accuracy, and customizability persists.

By Marcus M. Noack, Mark D. Risser, Hengrui Luo, Vardaan Tekriwal, Ronald J. Pandolfi
Hugging Face Trending Papers
Aug 11

Improving TensorSketch Using Complex Random Variables

\texttt{TensorSketch} by~\cite{pham2013fast,kar2012random} provides efficient sketching algorithms for high-dimensional polynomial kernels $\vec{x}^{\otimes p} \in \R^{d^p}$. \cite{kar2012random} uses dense Johnson-Lindenstrauss (JL)-type projections with computational cost $O(pDd)$, where $D$ denotes the sketch dimension, whereas~\cite{pham2013fast} extends the sparse \texttt{CountSketch}~\citep{count_sketch} algorithm, yielding a faster algorithm for high-dimensional sparse inputs with running time $O\big(p(\nnz{\vec{x}} + D \log D)\big)$.

arXiv Machine Learning
Sep 25

A Neural Hierarchical-Matrix Preconditioner for Real-Time GPU Solves

The paper presents a neural hierarchical‑matrix preconditioner designed for real‑time GPU solves of sparse symmetric positive‑definite systems that change each frame. By training a graph‑and‑attention network to predict an SPD approximate inverse in H²‑matrix format, the method achieves linear‑time inference and application, outperforming traditional multigrid setup times and local preconditioners. Experiments on 3D mesh diffusion problems show the preconditioner reduces conjugate‑gradient iterations from 116 to 33 and enables 120 fps real‑time performance for up to 3,647 unknowns.

By Carl Osborne, Minghao Guo, Crystal Owens, Wojciech Matusik