Categorical EHR imputation with generative adversarial nets
Yang Y, Wu Z, Tresp V, Fasching P (2019)
Publication Type: Conference contribution
Publication year: 2019
Publisher: Institute of Electrical and Electronics Engineers Inc.
Conference Proceedings Title: 2019 IEEE International Conference on Healthcare Informatics, ICHI 2019
Event location: Xi'an, CHN
ISBN: 9781538691380
DOI: 10.1109/ICHI.2019.8904717
Abstract
Electronic Health Records often suffer from missing data, which poses a major problem in clinical practice and clinical studies. A novel approach for dealing with missing data are Generative Adversarial Nets (GANs), which have been generating huge research interest in image generation and transformation. Recently, researchers have attempted to apply GANs to missing data generation and imputation for EHR data: a major challenge here is the categorical nature of the data. State-of-the-art solutions to the GAN-based generation of categorical data involve either reinforcement learning, or learning a bidirectional mapping between the categorical and the a real latent feature space, so that the GANs only need to generate real-valued features. However, these methods are designed to generate complete feature vectors instead of imputing only the subsets of missing features. In this paper we propose a simple and yet effective approach that is based on previous work on GANs for data imputation. We first motivate our solution by discussing the reason why adversarial training often fails in case of categorical features. Then we derive a novel way to re-code the categorical features to stabilize the adversarial training. Based on experiments on two real-world EHR data with multiple settings, we show that our imputation approach largely improves the prediction accuracy, compared to more traditional data imputation approaches.
Authors with CRIS profile
Involved external institutions
Germany (DE)How to cite
APA:
Yang, Y., Wu, Z., Tresp, V., & Fasching, P. (2019). Categorical EHR imputation with generative adversarial nets. In 2019 IEEE International Conference on Healthcare Informatics, ICHI 2019. Xi'an, CHN: Institute of Electrical and Electronics Engineers Inc..
MLA:
Yang, Yinchong, et al. "Categorical EHR imputation with generative adversarial nets." Proceedings of the 7th IEEE International Conference on Healthcare Informatics, ICHI 2019, Xi'an, CHN Institute of Electrical and Electronics Engineers Inc., 2019.
BibTeX: Download