Giter VIP home page Giter VIP logo

cor-gan's Introduction

COR-GAN: Correlation-Capturing Convolutional Neural Networks for Generating Synthetic Healthcare Records

Language grade: Python Total alerts PWC

This repository contains an implementation of "COR-GAN: Correlation-Capturing Convolutional Neural Networks for Generating Synthetic Healthcare Records". This is not an officially supported Google product.

For the following reason, this implementation does not contain the full implementation of the code:

  • The privacy restrictions regarding this work.

For a detailed description of the architecture please read our paper. Using the code of this repository is allowed with proper attribution: Please cite the paper if you use the code from this repository in your work.

Bibtex

@article{torfi2020cor,
title={COR-GAN: Correlation-Capturing Convolutional Neural Networks for Generating Synthetic Healthcare Records},
author={Torfi, Amirsina and Fox, Edward A},
journal={arXiv preprint arXiv:2001.09346},
year={2020}
}

Table of contents

Paper Summary

Abstract

Deep learning models have demonstrated high-quality performance in areas such as image classification and speech processing. However, creating a deep learning model using electronic health record (EHR) data, requires addressing particular privacy challenges that are unique to researchers in this domain. This matter focuses attention on generating realistic synthetic data while ensuring privacy. In this paper, we propose a novel framework called correlation-capturing Generative Adversarial Network (corGAN), to generate synthetic healthcare records. In corGAN we utilize Convolutional Neural Networks to capture the correlations between adjacent medical features in the data representation space by combining Convolutional Generative Adversarial Networks and Convolutional Autoencoders. To demonstrate the model fidelity, we show that corGAN generates synthetic data with performance similar to that of real data in various Machine Learning settings such as classification and prediction. We also give a privacy assessment and report on statistical analysis regarding realistic characteristics of the synthetic data.

Motivation
  • Synthetic records helping stakeholders to share and work on data without privacy hurdles
  • Despite advances in Synthetic Data Generation~(SDG). Research efforts mostly restricted to limited use cases
  • Lack of clarity regarding the synthetic data being realistic; and factors contributing to realism
  • Majority of methods for supervised settings
  • Lack of clarity in measuring privacy, and privacy guarantees
  • EHRs are discrete in nature. But most research is on continuous data
Contribution
  • We propose an efficient architecture to generate synthetic healthcare records using Convolutional GANs and Convolutional Autoencoders}~(CAs) which we call corGAN. We demonstrate that corGAN can effectively generate both discrete and continuous synthetic records.
  • We demonstrate the effectiveness of utilizing Convolutional Neural Networks~(CNNs) as opposed to Multilayer Perceptrons to capture inter-correlation between features.
  • We show that corGAN can generate realistic synthetic data that performs similarly to real data on classification tasks, according to our analysis and assessments.
  • We report on a privacy assessment of the model and demonstrate that corGAN provides an acceptable level of privacy, by varying the amount of synthetically generated data and amount of data known to an adversary.

Aspects of The Work

Synthetic Data Generation

Details

The discrete input X represents the source EHR data; z is the random distribution for the generator G; G is the employed neural network architecture; Dec(G(z))} refers to the decoding function which is used to transform the generator G continuous output to their equivalent discrete values. The discriminator D attempts to distinguish real input X from the discrete synthetic output Dec(G(z))}. For the generator and the discriminator, a 1-Dimensional Convolutional GAN architecture is utilized.

Privacy - Membership Inference Attack

Details

We utilize the Membership Inference (MI) attack as an approach to measure the privacy. Membership Inference~(MI) refers to determining whether a given record generated by a known machine learning model was used as part of the training data. The membership inference problem is basically the well-known problem of presence disclosure of an individual. If the adversary has complete access to the records of a particular patient and can recognize their employment in the model training, that is an indication of information leakage, as it can jeopardize the whole dataset privacy or at least the particular patient's private information. Here, we will assume the adversary has the synthetically generated data as well as a portion of the compromised real data.

Data Fidelity

Details

Binary Classification: We use this metric for our experiments with continuous data. To empirically verify the quality of the synthetic data, we consider two different settings. (A) Train and test the predictive models on the real data. (B)train the predictive model on synthetic data and test it on the real data. If the model evaluated in setting B, represents competitive results with the same model performed in setting (A), we can conclude the synthetic data has good predictive modeling similar to the real data.

Running the Code

Prerequisites

  • Pytorch 1.4
  • CUDA [strongly recommended]

NOTE: PyTorch does a pretty good job in installing required packages but you should have installed CUDA according to PyTorch requirements. Please refer to this link for further information.

Datasets

You need to download and process the following datasets as due to privacy restrictions we cannot provide the data here.

One good source code for processing MIMIC-III can be found here.

Training

To check the implementation refer to the folder Generative and you will see the following implementations:

  • corGAN: The implementation of the paper with concolutional autoencoders and regular multi-layer perceptions for discriminator and generator.
  • medGAN: The implementation of the medGAN paper as one of the baselines. This is a reimplemetation of the medGAN in PyTorch as we could not fully reproduce their results with their code. Furthermore, PyTorch is more flexible compare to TensorFlow for research!
  • VAE: One of the the baselines.

Collaborators


Amirsina Torfi

Mohammadreza Beyki

cor-gan's People

Contributors

astorfi avatar

Watchers

James Cloos avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.