Representation Learning for Patients in the Intensive Care Unit


The past decade has seen accelerating interest in Artificial Intelligence (AI) in Healthcare. Data is now being generated in the form of Electronic Health Records at a scale previously unimaginable. Not only does this create opportunities for the application of AI, but it also drives innovation in the machine learning sphere. This is because health problems can present unique challenges not encountered in other domains, and clinical decision making itself can provide ingenious approaches inspiring new learning methods. The work in this thesis sits in the space between medicine and machine learning and has contributions to both domains. The broad theme is representation learning for the patient in intensive care. The eventual aim is to promote better outcomes for patients and improve the efficiency of the healthcare system. I focus in particular on predicting patient deaths and estimated dates of discharge, because they lie at the heart of the resource allocation problem in hospitals. The efficient management of hospital beds is more important than ever in the wake of staff retention crises, post-pandemic budgets and ageing populations. Specifically, in Chapter 3, I use clinical knowledge of the medical time series (namely that they are periodic signals with particular systematic biases) to improve upon the state-of-the-art in length of stay prediction (with additional investigations into mortality prediction). In Chapter 4, I am again inspired by knowledge of the clinical decision making process to propose a method using graph neural networks to leverage data from similar patients when predicting outcomes, providing important context for the predictions and interpretability opportunities. In Chapter 5, I delve further into the representation space, exploring the effect of auxiliary tasks on the performance of patient outcome models for mechanically ventilated patients. I then cluster the learned representations with the aim of discovering hidden patient phenotypes. The vision is ultimately to create robust and holistic patient representations which are suitable for deployment in the real-world.

In University of Cambridge Thesis Repository
Emma Rocheteau
Emma Rocheteau
ML for Healthcare Researcher, Final Year Medical Student

My research interests include deep learning for patient outcome prediction, especially time series methods and graph neural networks.