Cluster-Specific Predictions with Multi-Task Gaussian Processes - Inria CWI Access content directly
Journal Articles Journal of Machine Learning Research Year : 2022

Cluster-Specific Predictions with Multi-Task Gaussian Processes

Abstract

A model involving Gaussian processes (GPs) is introduced to simultaneously handle multi-task learning, clustering, and prediction for multiple functional data. This procedure acts as a model-based clustering method for functional data as well as a learning step for subsequent predictions for new tasks. The model is instantiated as a mixture of multi-task GPs with common mean processes. A variational EM algorithm is derived for dealing with the optimisation of the hyper-parameters along with the hyper-posteriors' estimation of latent variables and processes. We establish explicit formulas for integrating the mean processes and the latent clustering variables within a predictive distribution, accounting for uncertainty in both aspects. This distribution is defined as a mixture of cluster-specific GP predictions, which enhances the performance when dealing with group-structured data. The model handles irregular grids of observations and offers different hypotheses on the covariance structure for sharing additional information across tasks. The performances on both clustering and prediction tasks are assessed through various simulated scenarios and real datasets. The overall algorithm, called MagmaClust, is publicly available as an R package.
Fichier principal
Vignette du fichier
pdf_HAL.pdf (4.98 Mo) Télécharger le fichier
Origin : Files produced by the author(s)

Dates and versions

hal-03009276 , version 1 (17-11-2020)
hal-03009276 , version 2 (29-07-2022)
hal-03009276 , version 3 (01-12-2022)

Identifiers

Cite

Arthur Leroy, Pierre Latouche, Benjamin Guedj, Servane Gey. Cluster-Specific Predictions with Multi-Task Gaussian Processes. Journal of Machine Learning Research, inPress. ⟨hal-03009276v3⟩
139 View
234 Download

Altmetric

Share

Gmail Facebook X LinkedIn More