Cluster-Specific Predictions with Multi-Task Gaussian Processes - Inria - Institut national de recherche en sciences et technologies du numérique
Article Dans Une Revue Journal of Machine Learning Research Année : 2022

Cluster-Specific Predictions with Multi-Task Gaussian Processes

Résumé

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
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

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

Identifiants

Citer

Arthur Leroy, Pierre Latouche, Benjamin Guedj, Servane Gey. Cluster-Specific Predictions with Multi-Task Gaussian Processes. Journal of Machine Learning Research, In press. ⟨hal-03009276v3⟩
192 Consultations
288 Téléchargements

Altmetric

Partager

More