Saved in:
Bibliographic Details
Main Authors: Ku, Dohyun, Chong, Catherine D., Berisha, Visar, Schwedt, Todd J., Li, Jing
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2509.19577
Tags: Add Tag
No Tags, Be the first to tag this record!
_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