ZiMM: a deep learning model for long term and blurry relapses with non-clinical claims data

TL;DR

This paper introduces ZiMM, a deep learning model designed to predict long-term, blurry relapses from non-clinical claims data, demonstrating improved accuracy over existing methods in a large population-based study.

Contribution

The paper presents ZiMM, a novel zero-inflated mixture model combined with an end-to-end deep learning architecture for modeling complex relapse patterns from claims data.

Findings

ZiMM ED outperforms baseline models in relapse prediction accuracy.

The approach effectively handles sparse, irregular, and heterogeneous health event data.

Minimal preprocessing is required for large-scale claims datasets.

Abstract

This paper considers the problems of modeling and predicting a long-term and ``blurry'' relapse that occurs after a medical act, such as a surgery. The relapse is observed only indirectly, in a ``blurry'' fashion, through longitudinal prescriptions of drugs over a long period of time after the medical act. We introduce a new model, called ZiMM (Zero-inflated Mixture of Multinomial distributions) in order to capture long-term and blurry relapses. On top of it, we build an end-to-end deep-learning architecture called ZiMM Encoder-Decoder (ZiMM ED) that can learn from the complex, irregular, highly heterogeneous and sparse patterns of health events that are observed through a claims-only database. ZiMM ED is applied on a ``non-clinical'' claims database, that contains only timestamped reimbursement codes for drug purchases, medical procedures and hospital diagnoses, the only available clinical feature being the age of the patient. This setting is more challenging than a setting where bedside clinical signals are available. Our motivation for using such a non-clinical claims database is its exhaustivity population-wise, compared to clinical electronic health records coming from a single or a small set of hospitals. Indeed, we consider a dataset containing the claims of almost \emph{all French citizens} who had surgery for prostatic problems, with a history between 1.5 and 5 years. We consider a long-term (18 months) relapse (urination problems still occur despite surgery), which is blurry since it is observed only through the reimbursement of a specific set of drugs for urination problems. Our experiments show that ZiMM ED improves several baselines, including non-deep learning and deep-learning approaches, and that it allows working on such a dataset with minimal preprocessing work.