English

Towards Foundation Models for Critical Care Time Series

Machine Learning 2024-11-26 v1 Machine Learning

Abstract

Notable progress has been made in generalist medical large language models across various healthcare areas. However, large-scale modeling of in-hospital time series data - such as vital signs, lab results, and treatments in critical care - remains underexplored. Existing datasets are relatively small, but combining them can enhance patient diversity and improve model robustness. To effectively utilize these combined datasets for large-scale modeling, it is essential to address the distribution shifts caused by varying treatment policies, necessitating the harmonization of treatment variables across the different datasets. This work aims to establish a foundation for training large-scale multi-variate time series models on critical care data and to provide a benchmark for machine learning models in transfer learning across hospitals to study and address distribution shift challenges. We introduce a harmonized dataset for sequence modeling and transfer learning research, representing the first large-scale collection to include core treatment variables. Future plans involve expanding this dataset to support further advancements in transfer learning and the development of scalable, generalizable models for critical healthcare applications.

Keywords

Cite

@article{arxiv.2411.16346,
  title  = {Towards Foundation Models for Critical Care Time Series},
  author = {Manuel Burger and Fedor Sergeev and Malte Londschien and Daphné Chopard and Hugo Yèche and Eike Gerdes and Polina Leshetkina and Alexander Morgenroth and Zeynep Babür and Jasmina Bogojeska and Martin Faltys and Rita Kuznetsova and Gunnar Rätsch},
  journal= {arXiv preprint arXiv:2411.16346},
  year   = {2024}
}

Comments

Accepted for Oral Presentation at AIM-FM Workshop at NeurIPS 2024

R2 v1 2026-06-28T20:11:22.892Z