Personalized Heart Disease Detection via ECG Digital Twin Generation (2404.11171v2)
Abstract: Heart diseases rank among the leading causes of global mortality, demonstrating a crucial need for early diagnosis and intervention. Most traditional electrocardiogram (ECG) based automated diagnosis methods are trained at population level, neglecting the customization of personalized ECGs to enhance individual healthcare management. A potential solution to address this limitation is to employ digital twins to simulate symptoms of diseases in real patients. In this paper, we present an innovative prospective learning approach for personalized heart disease detection, which generates digital twins of healthy individuals' anomalous ECGs and enhances the model sensitivity to the personalized symptoms. In our approach, a vector quantized feature separator is proposed to locate and isolate the disease symptom and normal segments in ECG signals with ECG report guidance. Thus, the ECG digital twins can simulate specific heart diseases used to train a personalized heart disease detection model. Experiments demonstrate that our approach not only excels in generating high-fidelity ECG signals but also improves personalized heart disease detection. Moreover, our approach ensures robust privacy protection, safeguarding patient data in model development.
- Publicly available clinical BERT embeddings. In Proceedings of the 2nd Clinical Natural Language Processing Workshop, pages 72–78, Minneapolis, Minnesota, USA, June 2019. Association for Computational Linguistics.
- Wasserstein generative adversarial networks. In Doina Precup and Yee Whye Teh, editors, Proceedings of the 34th International Conference on Machine Learning, volume 70 of Proceedings of Machine Learning Research, pages 214–223. PMLR, 06–11 Aug 2017.
- Classification of myocardial infarction with multi-lead ECG signals and deep cnn. Pattern Recognition Letters, 122:23–30, 2019.
- Flow-Mixup: Classifying multi-labeled medical images with corrupted labels. In 2020 IEEE International Conference on Bioinformatics and Biomedicine, pages 534–541. IEEE, 2020.
- Electrocardio panorama: Synthesizing new ECG views with self-supervision. In Zhi-Hua Zhou, editor, Proceedings of the Thirtieth International Joint Conference on Artificial Intelligence, IJCAI-21, pages 3597–3605. International Joint Conferences on Artificial Intelligence Organization, 8 2021. Main Track.
- ME-GAN: Learning panoptic electrocardio representations for multi-view ECG synthesis conditioned on heart diseases. In Kamalika Chaudhuri, Stefanie Jegelka, Le Song, Csaba Szepesvari, Gang Niu, and Sivan Sabato, editors, Proceedings of the 39th International Conference on Machine Learning, volume 162 of Proceedings of Machine Learning Research, pages 3360–3370. PMLR, 17–23 Jul 2022.
- Congenital heart disease detection by pediatric electrocardiogram based deep learning integrated with human concepts. Nature Communications, 15(1):976, 2024.
- Text-to-ECG: 12-lead electrocardiogram synthesis conditioned on clinical text reports. In ICASSP 2023 - 2023 IEEE International Conference on Acoustics, Speech and Signal Processing, pages 1–5, 2023.
- TWIN: Personalized clinical trial digital twin generation. In Proceedings of the 29th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, pages 402–413, 2023.
- Taming transformers for high-resolution image synthesis. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 12873–12883, 2021.
- PGANs: personalized generative adversarial networks for ECG synthesis to improve patient-specific deep ECG classification. In Proceedings of the Thirty-Third AAAI Conference on Artificial Intelligence and Thirty-First Innovative Applications of Artificial Intelligence Conference and Ninth AAAI Symposium on Educational Advances in Artificial Intelligence, pages 557–564, 2019.
- SimGANs: Simulator-based generative adversarial networks for ECG synthesis to improve deep ECG classification. In Proceedings of the 37th International Conference on Machine Learning, ICML’20. JMLR.org, 2020.
- Generative adversarial nets. Advances in Neural Information Processing Systems, 27, 2014.
- Prompt-to-prompt image editing with cross attention control. arXiv preprint arXiv:2208.01626, 2022.
- GANs trained by a two time-scale update rule converge to a local nash equilibrium. In Proceedings of the 31st International Conference on Neural Information Processing Systems, NIPS’17, page 6629–6640, Red Hook, NY, USA, 2017. Curran Associates Inc.
- ECG-Adv-GAN: Detecting ECG adversarial examples with conditional generative adversarial networks. In 2021 20th IEEE International Conference on Machine Learning and Applications, pages 50–56, 2021.
- Cross attention network for few-shot classification. Advances in Neural Information Processing Systems, 32, 2019.
- A style-based generator architecture for generative adversarial networks. In IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 4396–4405, 2019.
- Derivative dynamic time warping. In SDM, 2001.
- Auto-encoding variational bayes. arXiv preprint arXiv:1312.6114, 2013.
- Frozen language model helps ECG zero-shot learning. arXiv preprint arXiv:2303.12311, 2023.
- Conditional generative adversarial nets. arXiv preprint arXiv:1411.1784, 2014.
- Semantic image synthesis with spatially-adaptive normalization. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 2337–2346, 2019.
- StyleCLIP: Text-driven manipulation of StyleGAN imagery. In Proceedings of the IEEE/CVF International Conference on Computer Vision, pages 2085–2094, 2021.
- Generating diverse high-fidelity images with VQ-VAE-2. Advances in Neural Information Processing Systems, 32, 2019.
- Expanding functional protein sequence spaces using generative adversarial networks. Nature Machine Intelligence, 3(4):324–333, 2021.
- Global, regional, and national age-sex-specific mortality for 282 causes of death in 195 countries and territories, 1980–2017: a systematic analysis for the global burden of disease study 2017. The Lancet, 392(10159):1736–1788, 2018.
- Improved techniques for training GANs. In Proceedings of the 30th International Conference on Neural Information Processing Systems, NIPS’16, page 2234–2242, Red Hook, NY, USA, 2016. Curran Associates Inc.
- Membership inference attacks against machine learning models. In 2017 IEEE Symposium on Security and Privacy, pages 3–18. IEEE, 2017.
- Rethinking the inception architecture for computer vision. In 2016 IEEE Conference on Computer Vision and Pattern Recognition, pages 2818–2826, 2016.
- An empirical study of example forgetting during deep neural network learning. In International Conference on Learning Representations, 2019.
- Neural discrete representation learning. Advances in Neural Information Processing Systems, 30, 2017.
- Attention is all you need. Advances in Neural Nnformation Processing Systems, 30, 2017.
- SeqGAN: sequence generative adversarial nets with policy gradient. In AAAI-17: Thirty-First AAAI Conference on Artificial Intelligence, volume 31, pages 2852–2858. Association for the Advancement of Artificial Intelligence, 2017.
- Yaojun Hu (7 papers)
- Jintai Chen (57 papers)
- Lianting Hu (2 papers)
- Dantong Li (6 papers)
- Jiahuan Yan (16 papers)
- Haochao Ying (21 papers)
- Huiying Liang (2 papers)
- Jian Wu (314 papers)