Cumulative Stay-time Representation for Electronic Health Records in Medical Event Time Prediction

Takayuki Katsuki, Kohei Miyaguchi, Akira Koseki, Toshiya Iwamori, Ryosuke Yanagiya, Atsushi Suzuki

Proceedings of the Thirty-First International Joint Conference on Artificial Intelligence

Main Track. Pages 3861-3867.

https://doi.org/10.24963/ijcai.2022/536

PDF BibTeX Video

Abstract

We address the problem of predicting when a disease will develop, i.e., medical event time (MET), from a patient's electronic health record (EHR).

The MET of non-communicable diseases like diabetes is highly correlated to cumulative health conditions, more specifically, how much time the patient spent with specific health conditions in the past.

The common time-series representation is indirect in extracting such information from EHR because it focuses on detailed dependencies between values in successive observations, not cumulative information.

We propose a novel data representation for EHR called cumulative stay-time representation (CTR), which directly models such cumulative health conditions.

We derive a trainable construction of CTR based on neural networks that has the flexibility to fit the target data and scalability to handle high-dimensional EHR.

Numerical experiments using synthetic and real-world datasets demonstrate that CTR alone achieves a high prediction performance, and it enhances the performance of existing models when combined with them.

Keywords

  • Multidisciplinary Topics and Applications: Health and Medicine
  • Machine Learning: Applications
  • Machine Learning: Feature Extraction, Selection and Dimensionality Reduction
  • Machine Learning: Representation learning
Welcome Future Conferences AI Journal Division IJCAI Proceedings Travel Grant Program IJCAI Awards