English

Laplace Approximation For Tensor Train Kernel Machines In System Identification

Machine Learning 2025-12-03 v1 Machine Learning

Abstract

To address the scalability limitations of Gaussian process (GP) regression, several approximation techniques have been proposed. One such method is based on tensor networks, which utilizes an exponential number of basis functions without incurring exponential computational cost. However, extending this model to a fully probabilistic formulation introduces several design challenges. In particular, for tensor train (TT) models, it is unclear which TT-core should be treated in a Bayesian manner. We introduce a Bayesian tensor train kernel machine that applies Laplace approximation to estimate the posterior distribution over a selected TT-core and employs variational inference (VI) for precision hyperparameters. Experiments show that core selection is largely independent of TT-ranks and feature structure, and that VI replaces cross-validation while offering up to 65x faster training. The method's effectiveness is demonstrated on an inverse dynamics problem.

Keywords

Cite

@article{arxiv.2512.02532,
  title  = {Laplace Approximation For Tensor Train Kernel Machines In System Identification},
  author = {Albert Saiapin and Kim Batselier},
  journal= {arXiv preprint arXiv:2512.02532},
  year   = {2025}
}

Comments

6 pages, 2 figures, 4 tables. Submitted to IFAC 2026. Code available at: https://github.com/AlbMLpy/laplace-ttkm

R2 v1 2026-07-01T08:05:17.879Z