Generating Multi-label Discrete Patient Records using Generative Adversarial Networks

Edward Choi, Siddharth Biswal, Bradley Malin, Jon Duke, Walter F. Stewart +1 more
3/19/2017
cs.LGcs.NE

Official code: mp2893/medgan (287 stars, BSD-3-Clause, last commit 2019-08-19).

Abstract

Access to electronic health record (EHR) data has motivated computational advances in medical research. However, various concerns, particularly over privacy, can limit access to and collaborative use of EHR data. Sharing synthetic EHR data could mitigate risk. In this paper, we propose a new approach, medical Generative Adversarial Network (medGAN), to generate realistic synthetic patient records. Based on input real patient records, medGAN can generate high-dimensional discrete variables (e.g., binary and count features) via a combination of an autoencoder and generative adversarial networks. We also propose minibatch averaging to efficiently avoid mode collapse, and increase the learning efficiency with batch normalization and shortcut connections. To demonstrate feasibility, we showed that medGAN generates synthetic patient records that achieve comparable performance to real data on many experiments including distribution statistics, predictive modeling tasks and a medical expert review. We also empirically observe a limited privacy risk in both identity and attribute disclosure using medGAN.

View on arXivView PDF

Code Implementations(1)

mp2893/medganOfficial100%
BSD-3-Clause

Generative adversarial network for generating electronic health records.

28795Python, Jupyter NotebookMar 19, 20177 years agoBSD-3-Clause

Similar Papers

Cite this paper

@article{choi2017generating,
  title  = {Generating Multi-label Discrete Patient Records using Generative Adversarial Networks},
  author = {Edward Choi and Siddharth Biswal and Bradley Malin and Jon Duke and Walter F. Stewart and Jimeng Sun},
  year   = {2017},
  eprint = {1703.06490},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/1703.06490}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/1703.06490.svg)](https://codeofpaper.com/paper/1703.06490)

Discussion