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

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

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