Informed Spectral Normalized Gaussian Processes for Trajectory Prediction

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Schlauch, Christian, Wirth, Christian, Klein, Nadja
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866912105063514112
author Schlauch, Christian
Wirth, Christian
Klein, Nadja
author_facet Schlauch, Christian
Wirth, Christian
Klein, Nadja
contents Prior parameter distributions provide an elegant way to represent prior expert and world knowledge for informed learning. Previous work has shown that using such informative priors to regularize probabilistic deep learning (DL) models increases their performance and data-efficiency. However, commonly used sampling-based approximations for probabilistic DL models can be computationally expensive, requiring multiple inference passes and longer training times. Promising alternatives are compute-efficient last layer kernel approximations like spectral normalized Gaussian processes (SNGPs). We propose a novel regularization-based continual learning method for SNGPs, which enables the use of informative priors that represent prior knowledge learned from previous tasks. Our proposal builds upon well-established methods and requires no rehearsal memory or parameter expansion. We apply our informed SNGP model to the trajectory prediction problem in autonomous driving by integrating prior drivability knowledge. On two public datasets, we investigate its performance under diminishing training data and across locations, and thereby demonstrate an increase in data-efficiency and robustness to location-transfers over non-informed and informed baselines.
format Preprint
id arxiv_https___arxiv_org_abs_2403_11966
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Informed Spectral Normalized Gaussian Processes for Trajectory Prediction
Schlauch, Christian
Wirth, Christian
Klein, Nadja
Machine Learning
Artificial Intelligence
Prior parameter distributions provide an elegant way to represent prior expert and world knowledge for informed learning. Previous work has shown that using such informative priors to regularize probabilistic deep learning (DL) models increases their performance and data-efficiency. However, commonly used sampling-based approximations for probabilistic DL models can be computationally expensive, requiring multiple inference passes and longer training times. Promising alternatives are compute-efficient last layer kernel approximations like spectral normalized Gaussian processes (SNGPs). We propose a novel regularization-based continual learning method for SNGPs, which enables the use of informative priors that represent prior knowledge learned from previous tasks. Our proposal builds upon well-established methods and requires no rehearsal memory or parameter expansion. We apply our informed SNGP model to the trajectory prediction problem in autonomous driving by integrating prior drivability knowledge. On two public datasets, we investigate its performance under diminishing training data and across locations, and thereby demonstrate an increase in data-efficiency and robustness to location-transfers over non-informed and informed baselines.
title Informed Spectral Normalized Gaussian Processes for Trajectory Prediction
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2403.11966