Skip to main content

ORIGINAL RESEARCH article

Front. Cardiovasc. Med.
Sec. Clinical and Translational Cardiovascular Medicine
Volume 11 - 2024 | doi: 10.3389/fcvm.2024.1460354
This article is part of the Research Topic Novel Translational Advances in Artificial Intelligence for Diagnosis and Treatment of Cardiovascular Diseases View all 7 articles

Contrastive Learning with Transformer for Adverse Endpoint Prediction in Patients on DAPT Post-Coronary Stent Implantation

Provisionally accepted
Fang Li Fang Li 1Zenan Sun Zenan Sun 2*Ahmed Abdelhameed Ahmed Abdelhameed 1*Tiehang Duan Tiehang Duan 1Laila Rasmy Laila Rasmy 2Xinyue Hu Xinyue Hu 1*Jianping He Jianping He 2Yifang Dang Yifang Dang 2*Jingna Feng Jingna Feng 1*Jianfu Li Jianfu Li 1Yichen Wang Yichen Wang 3Tianchen Lyu Tianchen Lyu 4*Naomi Braun Naomi Braun 4*Si Pham Si Pham 5Michael Gharacholou Michael Gharacholou 6*DeLisa Fairweather DeLisa Fairweather 6Degui Zhi Degui Zhi 2*Jiang Bian Jiang Bian 4Cui Tao Cui Tao 1*
  • 1 Department of Artificial Intelligence and Informatics, Mayo Clinic, Jacksonville, Alabama, United States
  • 2 School of Biomedical Informatics, University of Texas Health Science Center at Houston, Houston, Texas, United States
  • 3 Perelman School of Medicine, University of Pennsylvania, Philadelphia, Pennsylvania, United States
  • 4 Department of Health Outcomes and Biomedical Informatics, College of Medicine, University of Florida, Gainesville, Florida, United States
  • 5 Department of Cardiothoracic Surgery, Mayo Clinic Florida, Jacksonville, Alabama, United States
  • 6 Department of Cardiovascular Medicine, Mayo Clinic, Jacksonville, United States

The final, formatted version of the article will be published soon.

    Background: Effective management of dual antiplatelet therapy (DAPT) following drug-eluting stent (DES) implantation is crucial for preventing adverse events. Traditional prognostic tools, such as rulebased methods or Cox regression, despite their widespread use and ease, tend to yield moderate predictive accuracy within predetermined timeframes. This study introduces a new contrastive learning-based approach to enhance prediction efficacy over multiple time intervals.We utilized retrospective, real-world data from the OneFlorida+ Clinical Research Consortium. Our study focused on two primary endpoints: ischemic and bleeding events, with prediction windows of 1, 2, 3, 6, and 12 months post-DES implantation. Our approach first utilized an auto-encoder to compress patient features into a more manageable, condensed representation.Following this, we integrated a Transformer architecture with multi-head attention mechanisms to focus on and amplify the most salient features, optimizing the representation for better predictive accuracy. Then, we applied contrastive learning to enable the model to further refine its predictive capabilities by maximizing intra-class similarities and distinguishing inter-class differences.Meanwhile, the model was holistically optimized using multiple loss functions, to ensure the predicted results closely align with the ground-truth values from various perspectives. We benchmarked model performance against three cutting-edge deep learning-based survival models, i.e., DeepSurv, DeepHit, and SurvTrace.The final cohort comprised 19,713 adult patients who underwent DES implantation with more than one month of records after coronary stenting. Our approach demonstrated superior predictive performance for both ischemic and bleeding events across prediction windows of 1, 2, 3, 6, and 12 months, with time-dependent concordance (C td ) index values ranging from 0.88 to 0.80 and 0.82 to 0.77, respectively. It consistently outperformed the baseline models, including DeepSurv, DeepHit, and SurvTrace, with statistically significant improvement in the C td -index values for most evaluated scenarios.The robust performance of our contrastive learning-based model underscores its potential to enhance DAPT management significantly. By delivering precise predictive insights at multiple time points, our method meets the current need for adaptive, personalized therapeutic strategies in cardiology, thereby offering substantial value in improving patient outcomes.

    Keywords: Dual antiplatelet therapy, Contrastive learning, transformer, Predictive Modeling, adverse endpoint, drug-eluting coronary artery stent implantation, survival analysis

    Received: 05 Jul 2024; Accepted: 19 Dec 2024.

    Copyright: © 2024 Li, Sun, Abdelhameed, Duan, Rasmy, Hu, He, Dang, Feng, Li, Wang, Lyu, Braun, Pham, Gharacholou, Fairweather, Zhi, Bian and Tao. This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) or licensor are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.

    * Correspondence:
    Zenan Sun, School of Biomedical Informatics, University of Texas Health Science Center at Houston, Houston, 77030, Texas, United States
    Ahmed Abdelhameed, Department of Artificial Intelligence and Informatics, Mayo Clinic, Jacksonville, Alabama, United States
    Xinyue Hu, Department of Artificial Intelligence and Informatics, Mayo Clinic, Jacksonville, Alabama, United States
    Yifang Dang, School of Biomedical Informatics, University of Texas Health Science Center at Houston, Houston, 77030, Texas, United States
    Jingna Feng, Department of Artificial Intelligence and Informatics, Mayo Clinic, Jacksonville, Alabama, United States
    Tianchen Lyu, Department of Health Outcomes and Biomedical Informatics, College of Medicine, University of Florida, Gainesville, 32610, Florida, United States
    Naomi Braun, Department of Health Outcomes and Biomedical Informatics, College of Medicine, University of Florida, Gainesville, 32610, Florida, United States
    Michael Gharacholou, Department of Cardiovascular Medicine, Mayo Clinic, Jacksonville, United States
    Degui Zhi, School of Biomedical Informatics, University of Texas Health Science Center at Houston, Houston, 77030, Texas, United States
    Cui Tao, Department of Artificial Intelligence and Informatics, Mayo Clinic, Jacksonville, Alabama, United States

    Disclaimer: All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.