LLM Machine Learning for Predicting Cardiovascular Mortality in Patients
DOI:
https://doi.org/10.5281/zenodo.14015999Keywords:
cardiovascular mortality, chronic kidney disease, machine learning, interpretable model, shapAbstract
Patients with chronic kidney disease (CKD) face a high risk of cardiovascular death, yet accurately predicting this risk remains challenging. This study aims to develop an interpretable machine learning (ML) model to predict 10-year cardiovascular mortality in CKD patients using SHAP explainers. [1]Six ML models were created and tested, with the best model selected for prediction and patient categorization. Survival rates were analyzed using log-rank tests on Kaplan-Meier curves, and Cox regression was employed to explore the relationship between ML-predicted risk scores and mortality. The chosen autoencoder (AE) model demonstrated superior performance, with higher ML scores[2] significantly correlating with increased cardiovascular mortality risk. Key determinants such as age, high blood pressure, C-reactive protein, and serum creatinine were identified. The ML-driven tool showcased high accuracy in determining the 10-year cardiovascular mortality risk for CKD patients, offering valuable insights for individual risk assessments.
Downloads
References
Shimizu, Shosei et al. (2021). Proton beam therapy for a giant hepatic hemangioma: A case report and literature review. Clinical and Translational Radiation Oncology, 27, 152-156. doi:10.1016/j.ctro.2021.01.014.
Shimizu, Shosei et al. (2023). Boron neutron capture therapy for recurrent glioblastoma multiforme: Imaging evaluation of a case with long-term local control and survival. Cureus, 15(1), e33898. doi:10.7759/cureus.33898.
Li, Yinuo et al. (2023). A retrospective study of renal growth changes after proton beam therapy for pediatric malignant tumor. Current Oncology (Toronto, Ont.), 30(2), 1560-1570. doi:10.3390/curroncol30020120.
Nakamura, Masatoshi et al. (2024). A systematic review and meta-analysis of radiotherapy and particle beam therapy for skull base chondrosarcoma: TRP-chondrosarcoma 2024. Frontiers in Oncology, 14, 1380716. doi:10.3389/fonc.2024.1380716.
Nitta, Hazuki et al. (2024). An analysis of muscle growth after proton beam therapy for pediatric cancer. Journal of Radiation Research, 65(2), 251-255. doi:10.1093/jrr/rrad105.
Jin, Yonglong et al. (2023). Proton therapy (PT) combined with concurrent chemotherapy for locally advanced non-small cell lung cancer with negative driver genes. Radiation Oncology (London, England), 18(1), 189. doi:10.1186/s13014-023-02372-8.
Li, Yinuo et al. (2023). Smart nanofiber mesh with locally sustained drug release enabled synergistic combination therapy for glioblastoma. Nanomaterials (Basel, Switzerland), 13(3), 414. doi:10.3390/nano13030414.
Saito, Takashi et al. (2024). Systematic review and meta-analysis of particle beam therapy versus photon radiotherapy for skull base chordoma: TRP-chordoma 2024. Cancers, 16(14), 2569. doi:10.3390/cancers16142569.
Li, Yinuo et al. (2024). Late changes in renal volume and function after proton beam therapy in pediatric and adult patients: Children show significant renal atrophy but deterioration of renal function is minimal in the long-term in both groups. Cancers, 16(9), 1634. doi:10.3390/cancers16091634.
Niitsu, Hikaru et al. (2024). Tumor response on diagnostic imaging after proton beam therapy for hepatocellular carcinoma. Cancers, 16(2), 357. doi:10.3390/cancers16020357.
Li, Yinuo et al. (2022). Proton beam therapy for multifocal hepatocellular carcinoma (HCC) showing complete response in pathological anatomy after liver transplantation. Cureus, 14(6), e25744. doi:10.7759/cureus.25744.
Kumada, Hiroaki et al. (2022). Current development status of iBNCT001, demonstrator of a LINAC-based neutron source for BNCT, 347–358. doi:10.3233/JNR-220029.
Wang, Randi, & Morad Behandish. (2022). Surrogate modeling for physical systems with preserved properties and adjustable tradeoffs. arXiv preprint arXiv:2202.01139.
Wang, Randi, & Vadim Shapiro. (2019). Topological semantics for lumped parameter systems modeling. Advanced Engineering Informatics, 42, 100958.
Wang, Randi, Vadim Shapiro, & Morad Mehandish. (2024). Model consistency for mechanical design: Bridging lumped and distributed parameter models with a priori guarantees. Journal of Mechanical Design, 146(5).
Chen, M., Chen, Y., & Zhang, Q. (2021). A review of energy consumption in the acquisition of bio-feedstock for microalgae biofuel production. Sustainability, 13(16), 8873.
Chen, M. (2023). Investigating the influence of interannual precipitation variability on terrestrial ecosystem productivity. Doctoral Dissertation, Massachusetts Institute of Technology.
Chen, M. (2021, December). Annual precipitation forecast of Guangzhou based on genetic algorithm and backpropagation neural network (GA-BP). in International Conference on Algorithms, High Performance Computing, and Artificial Intelligence (AHPCAI 2021), 12156, pp. 182-186. SPIE.
Dong, S., Xu, T., & Chen, M. (2022, October). Solar radiation characteristics in Shanghai. in Journal of Physics: Conference Series, 2351(1), pp. 012016. IOP Publishing.
Zhang, X., Soe, A. N., Dong, S., Chen, M., Wu, M., & Htwe, T. (2024). Urban resilience through green roofing: A literature review on dual environmental benefits. in E3S Web of Conferences, 536, pp. 01023. EDP Sciences.
Chen, M., Chen, Y., & Zhang, Q. (2024). Assessing global carbon sequestration and bioenergy potential from microalgae cultivation on marginal lands leveraging machine learning. Science of the Total Environment, 948, 174462.
Song, Y., Arora, P., Varadharajan, S. T., Singh, R., Haynes, M., & Starner, T. (2024, April). Looking from a different angle: Placing head-worn displays near the nose. in Proceedings of the Augmented Humans International Conference, pp. 28-45.
Song, Y., Arora, P., Singh, R., Varadharajan, S. T., Haynes, M., & Starner, T. (2023, October). Going blank comfortably: Positioning monocular head-worn displays when they are inactive. in Proceedings of the 2023 ACM International Symposium on Wearable Computers, pp. 114-118.
Qian, C., Guo, Y., Mo, Y., & Li, W. (2024). WeatherDG: LLM-assisted procedural weather generation for domain-generalized semantic segmentation. arXiv [Cs.CV]. Retrieved from: http://arxiv.org/abs/2410.12075.
Wang, Yang, Yojiro Mori, & Hiroshi Hasegawa. (2021). Dynamic routing and spectrum allocation based on actor-critic learning for multi-fiber elastic optical networks. Photonics in Switching and Computing 2021, Optica Publishing Group, pp. W1B.3. doi:10.1364/PSC.2021.W1B.3.
Wang, Yang, et al. (2018). Optimizing multi-criteria k-shortest paths in graph by a natural routing genotype-based genetic algorithm. 13th IEEE Conference on Industrial Electronics and Applications (ICIEA), pp. 341-345. doi:10.1109/ICIEA.2018.8397739.
Wang, Yang, Yojiro Mori, & Hiroshi Hasegawa. (2020). Resource assignment based on core-state value evaluation to handle crosstalk and spectrum fragments in sdm elastic optical networks. Opto-Electronics and Communications Conference (OECC), pp. 1-3. doi:10.1109/OECC48412.2020.9273621.
Zhao, Yuwen, Baojun Hu, & Sizhe Wang. (2024). Prediction of brent crude oil price based on LSTM model under the background of low-carbon transition. arXiv preprint arXiv:2409.12376.
Yan, Hao, et al. (2024). Research on image generation optimization based deep learning.
Tang, Xirui, et al. (2024). Research on heterogeneous computation resource allocation based on data-driven method. arXiv preprint arXiv:2408.05671.
Han, Yi, & Thomas CM Lee. (2022). Uncertainty quantification for sparse estimation of spectral lines. IEEE Transactions on Signal Processing, 70, 6243-6256.
Yao, Jiawei, et al. (2023). Ndc-scene: Boost monocular 3d semantic scene completion in normalized device coordinates space. IEEE/CVF International Conference on Computer Vision (ICCV). IEEE Computer Society.
Yao, Jiawei, et al. (2024). QE-BEV: Query evolution for bird's eye view object detection in varied contexts. ACM Multimedia.
Pan, Xiaochao, et al. (2024). HarmonicNeRF: Geometry-informed synthetic view augmentation for 3d scene reconstruction in driving scenarios. ACM Multimedia.
Ma, B., Ma, B., Gao, M., Wang, Z., Ban, X., Huang, H., & Wu, W. (2021). Deep learning based automatic inpainting for material microscopic images. Journal of Microscopy, 281(3), 177-189.
Wang, Y., Ban, X., Wang, H., Li, X., Wang, Z., Wu, D., ... & Liu, S. (2019). Particle filter vehicles tracking by fusing multiple features. IEEE Access, 7, 133694-133706.
Liu, Jiabei, et al. (2024). Application of deep learning-based natural language processing in multilingual sentiment analysis. Mediterranean Journal of Basic and Applied Sciences (MJBAS), 8(2), 243-260.
Xu, Qiming, et al. (2024). Applications of explainable ai in natural language processing. Global Academic Frontiers, 2(3), 51-64.
Zhong, Yihao, et al. (2024). Deep learning solutions for pneumonia detection: Performance comparison of custom and transfer learning models. medRxiv, 2024-06.
Zhu, Armando, et al. (2024). Exploiting diffusion prior for out-of-distribution detection. arXiv preprint arXiv:2406.11105.
Li, Keqin, et al. (2024). Exploring the impact of quantum computing on machine learning performance.
Gu, Wenjun, et al. (2024). Predicting stock prices with FinBERT-LSTM: Integrating news sentiment analysis. arXiv preprint arXiv:2407.16150.
Wang, Zixiang, et al. (2024). Research on autonomous driving decision-making strategies based deep reinforcement learning. arXiv preprint arXiv:2408.03084.
Bo, Shi, et al. (2024). Attention mechanism and context modeling system for text mining machine translation. arXiv preprint arXiv:2408.04216.
Qian, Yang, et al. (2020). Heterogeneous optoelectronic characteristics of Si micropillar arrays fabricated by metal-assisted chemical etching. Scientific Reports, 10(1), 16349.
Li, Wei, et al. (2018). An intelligent electronic lock for remote-control system based on the internet of things. Journal of Physics: Conference Series, 1069(1). IOP Publishing.
Gao, Haoqi, et al. (2016). A novel texture extraction method for the sedimentary structures’ classification of petroleum imaging logging. Pattern Recognition: 7th Chinese Conference, Chengdu, China, Proceedings, Part II 7. Springer Singapore.
Yan, Hao, et al. (2024). Research on image generation optimization based deep learning.
Tang, Xirui, et al. (2024). Research on heterogeneous computation resource allocation based on data-driven method. arXiv preprint arXiv:2408.05671.
Su, Pei-Chiang, et al. (2022). A mixed-heuristic quantum-inspired simplified swarm optimization algorithm for scheduling of real-time tasks in the multiprocessor system. Applied Soft Computing, 131, 109807.
Zhao, Yuwen, Baojun Hu, & Sizhe Wang. (2024). Prediction of brent crude oil price based on LSTM model under the background of low-carbon transition. arXiv preprint arXiv:2409.12376.
Diao, Su, et al. (2024). Ventilator pressure prediction using recurrent neural network. arXiv preprint arXiv:2410.06552.
Zhao, Qinghe, Yue Hao, & Xuechen Li. (2024). Stock price prediction based on hybrid CNN-LSTM model.
Yin, Ziqing, Baojun Hu, & Shuhan Chen. (2024). Predicting employee turnover in the financial company: A comparative study of CatBoost and XGBoost Models.
Diao, Su, et al. (2024). Ventilator pressure prediction using recurrent neural network. arXiv preprint arXiv:2410.06552.
Qian, Chenghao, et al. (2024). WeatherDG: LLM-assisted procedural weather generation for domain-generalized semantic segmentation. arXiv preprint arXiv:2410.12075.
Li, Zhenglin, et al. (2023). Stock market analysis and prediction using LSTM: A case study on technology stocks. Innovations in Applied Engineering and Technology, 1-6.
Mo, Yuhong, et al. (2024). Large Language Model (LLM) AI text generation detection based on transformer deep learning algorithm. International Journal of Engineering and Management Research, 14(2), 154-159.
Li, Shaojie, Yuhong Mo, & Zhenglin Li. (2022). Automated pneumonia detection in chest x-ray images using deep learning model. Innovations in Applied Engineering and Technology, 1-6.
Mo, Yuhong, et al. (2024). Password complexity prediction based on roberta algorithm. Applied Science and Engineering Journal for Advanced Research, 3(3), 1-5.
Liu, Tianrui, et al. (2024). Spam detection and classification based on distilbert deep learning algorithm. Applied Science and Engineering Journal for Advanced Research, 3(3), 6-10.
Dai, Shuying, et al. (2024). The cloud-based design of unmanned constant temperature food delivery trolley in the context of artificial intelligence. Journal of Computer Technology and Applied Mathematics, 1(1), 6-12.
Mo, Yuhong, et al. (2024). Make scale invariant feature transform “Fly” with CUDA. International Journal of Engineering and Management Research, 14(3), 38-45.
He, Shuyao, et al. (2024). Lidar and monocular sensor fusion depth estimation. Applied Science and Engineering Journal for Advanced Research, 3(3), 20-26.
Liu, Jihang, et al. (2024). Unraveling large language models: From evolution to ethical implications-introduction to large language models. World Scientific Research Journal, 10(5), 97-102.
Mo Yuhong, Zhang Yuchen, Li Hanzhe, Wang Han, & Yan Xu. (2024). Prediction of heart failure patients based on multiple machine learning algorithms. Applied and Computational Engineering, 75. 1-7. doi:10.54254/2755-2721/75/20240498.
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2024 Yue Zhu, Xiaoyi Zhang, Yuechen Zhang
This work is licensed under a Creative Commons Attribution 4.0 International License.