The Health Gym: synthetic health-related datasets for the development of reinforcement learning algorithms.
Nicholas I-Hsien KuoMark N PolizzottoSimon FinferFederico GarciaAnders SönnerborgMaurizio ZazziMichael BöhmRolf KaiserLouisa JormSebastiano BarbieriPublished in: Scientific data (2022)
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.
Keyphrases
- machine learning
- human immunodeficiency virus
- antiretroviral therapy
- healthcare
- rna seq
- hiv infected
- mental health
- hiv positive
- hepatitis c virus
- big data
- public health
- hiv aids
- artificial intelligence
- hiv infected patients
- deep learning
- health information
- liver failure
- south africa
- climate change
- risk assessment
- hiv testing
- extracorporeal membrane oxygenation
- men who have sex with men
- health promotion