Publication:
The Health Gym: synthetic health-related datasets for the development of reinforcement learning algorithms.

Loading...
Thumbnail Image

Date

2022-11-11

Authors

Kuo, Nicholas I-Hsien
Polizzotto, Mark N
Finfer, Simon
Garcia, Federico
Sönnerborg, Anders
Zazzi, Maurizio
Böhm, Michael
Kaiser, Rolf
Jorm, Louisa
Barbieri, Sebastiano

Advisors

Journal Title

Journal ISSN

Volume Title

Publisher

Metrics
Google Scholar
Export

Research Projects

Organizational Units

Journal Issue

Abstract

In recent years, the machine learning research community has benefited tremendously from the availability of openly accessible benchmark datasets. Clinical data are usually not openly available due to their confidential nature. This has hampered the development of reproducible and generalisable machine learning applications in health care. Here we introduce the Health Gym - a growing collection of highly realistic synthetic medical datasets that can be freely accessed to prototype, evaluate, and compare machine learning algorithms, with a specific focus on reinforcement learning. The three synthetic datasets described in this paper present patient cohorts with acute hypotension and sepsis in the intensive care unit, and people with human immunodeficiency virus (HIV) receiving antiretroviral therapy. The datasets were created using a novel generative adversarial network (GAN). The distributions of variables, and correlations between variables and trends in variables over time in the synthetic datasets mirror those in the real datasets. Furthermore, the risk of sensitive information disclosure associated with the public distribution of the synthetic datasets is estimated to be very low.

Description

MeSH Terms

Humans
Algorithms
Machine Learning
Comprehensive Health Care

DeCS Terms

CIE Terms

Keywords

Citation