MAGIC: Multi-task Gaussian process for joint imputation and classification in healthcare time series

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Ku, Dohyun, Chong, Catherine D., Berisha, Visar, Schwedt, Todd J., Li, Jing
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866911173093359616
author Ku, Dohyun
Chong, Catherine D.
Berisha, Visar
Schwedt, Todd J.
Li, Jing
author_facet Ku, Dohyun
Chong, Catherine D.
Berisha, Visar
Schwedt, Todd J.
Li, Jing
contents Time series analysis has emerged as an important tool for improving patient diagnosis and management in healthcare applications. However, these applications commonly face two critical challenges: time misalignment and data sparsity. Traditional approaches address these issues through a two-step process of imputation followed by prediction. We propose MAGIC (Multi-tAsk Gaussian Process for Imputation and Classification), a novel unified framework that simultaneously performs class-informed missing value imputation and label prediction within a hierarchical multi-task Gaussian process coupled with functional logistic regression. To handle intractable likelihood components, MAGIC employs Taylor expansion approximations with bounded error analysis, and parameter estimation is performed using EM algorithm with block coordinate optimization supported by convergence analysis. We validate MAGIC through two healthcare applications: prediction of post-traumatic headache improvement following mild traumatic brain injury and prediction of in-hospital mortality within 48 hours after ICU admission. In both applications, MAGIC achieves superior predictive accuracy compared to existing methods. The ability to generate real-time and accurate predictions with limited samples facilitates early clinical assessment and treatment planning, enabling healthcare providers to make more informed treatment decisions.
format Preprint
id arxiv_https___arxiv_org_abs_2509_19577
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle MAGIC: Multi-task Gaussian process for joint imputation and classification in healthcare time series
Ku, Dohyun
Chong, Catherine D.
Berisha, Visar
Schwedt, Todd J.
Li, Jing
Machine Learning
Time series analysis has emerged as an important tool for improving patient diagnosis and management in healthcare applications. However, these applications commonly face two critical challenges: time misalignment and data sparsity. Traditional approaches address these issues through a two-step process of imputation followed by prediction. We propose MAGIC (Multi-tAsk Gaussian Process for Imputation and Classification), a novel unified framework that simultaneously performs class-informed missing value imputation and label prediction within a hierarchical multi-task Gaussian process coupled with functional logistic regression. To handle intractable likelihood components, MAGIC employs Taylor expansion approximations with bounded error analysis, and parameter estimation is performed using EM algorithm with block coordinate optimization supported by convergence analysis. We validate MAGIC through two healthcare applications: prediction of post-traumatic headache improvement following mild traumatic brain injury and prediction of in-hospital mortality within 48 hours after ICU admission. In both applications, MAGIC achieves superior predictive accuracy compared to existing methods. The ability to generate real-time and accurate predictions with limited samples facilitates early clinical assessment and treatment planning, enabling healthcare providers to make more informed treatment decisions.
title MAGIC: Multi-task Gaussian process for joint imputation and classification in healthcare time series
topic Machine Learning
url https://arxiv.org/abs/2509.19577