Stroke recurrence prediction using machine learning and segmented neural network risk factor aggregation

被引:0
|
作者
Ding, Xueting [1 ]
Meng, Yang [2 ]
Xiang, Liner [2 ]
Boden-Albala, Bernadette [1 ,3 ,4 ]
机构
[1] Univ Calif Irvine, Henry & Susan Samueli Coll Hlth Sci, Joe C Wen Sch Populat & Publ Hlth, Dept Hlth Soc & Behav,UCI Hlth Sci Complex, 856 Hlth Sci Quad, Irvine, CA 92697 USA
[2] Univ Calif Irvine, Donald Bren Sch Informat & Comp Sci, Dept Stat, Bren Hall 2019, Irvine, CA 92697 USA
[3] Univ Calif Irvine, Henry & Susan Samueli Coll Hlth Sci, Sch Med, Dept Neurol, Irvine, CA 92697 USA
[4] Univ Calif Irvine, Henry & Susan Samueli Coll Hlth Sci, Joe C Wen Sch Populat & Publ Hlth, Dept Epidemiol & Biostat,UCI Hlth Sci Complex, 856 Hlth Sci Quad, Irvine, CA 92697 USA
关键词
Stroke recurrence; Data aggregation; Machine learning; Interpretable neural network; Supervised encoder; Logistic regression; Random forest;
D O I
10.1186/s12982-024-00199-6
中图分类号
R1 [预防医学、卫生学];
学科分类号
1004 ; 120402 ;
摘要
Stroke has remained a major cause of mortality and disability in the United States for years, and its recurrence significantly increased the risks. For predicting stroke recurrence, traditional data aggregation methods have limitations in effectively handling the numerous subcategories of stroke risk factors. This pilot study proposed a Segmented Neural Network-Driven Aggregation (SNA) method, and it aimed to improve the prediction model's accuracy. Utilizing the TriNetX diagnosis dataset, we processed various risk factors and demographic information through traditional and our proposed data aggregation techniques. We applied logistic regression and random forest classifiers to predict stroke recurrence. Our findings revealed that using the SNA method significantly outperformed other aggregation methods for both classifiers. Using the SNA method with a random forest classifier achieved higher accuracy (84.2%) and a better balance between sensitivity and specificity (AUC of ROC = 0.928, AUC of PR = 0.940) compared to other combinations. These results showed the potential of machine-learning supervised encoding methods in stroke recurrence predictions, providing implications for clinical practice and future epidemiological research.
引用
收藏
页数:10
相关论文
共 50 条
  • [31] Prediction of distant recurrence in breast cancer using a deep neural network
    Azman, Balqis Mohd
    Hussain, Saiful Izzuan
    Azmi, Nor Aniza
    Abd Ghani, Muhammad Zahin Athir
    Norlen, Nor Irfan Danial
    REVISTA INTERNACIONAL DE METODOS NUMERICOS PARA CALCULO Y DISENO EN INGENIERIA, 2022, 38 (01):
  • [32] Using neural network and decision tree for machine reliability prediction
    Kuo, Yiyo
    Lin, Kuo-Ping
    INTERNATIONAL JOURNAL OF ADVANCED MANUFACTURING TECHNOLOGY, 2010, 50 (9-12): : 1243 - 1251
  • [33] Thermal deformation prediction in machine tools by using neural network
    Chang, Chuan-Wei
    Kang, Yuan
    Chen, Yi-Wei
    Chu, Ming-Hui
    Wang, Yea-Ping
    NEURAL INFORMATION PROCESSING, PT 2, PROCEEDINGS, 2006, 4233 : 850 - 859
  • [34] Using neural network and decision tree for machine reliability prediction
    Yiyo Kuo
    Kuo-Ping Lin
    The International Journal of Advanced Manufacturing Technology, 2010, 50 : 1243 - 1251
  • [35] Prediction of inhibitor risk in haemophilia A using machine learning
    Sottilotta, Gianluca
    Luise, Francesca
    Nicolo, Giovanna Maria
    Piromalli, Angela
    Fazio, Manlio
    Grasso, Stephanie
    Giunta, Giuliana
    Gullo, Lara
    Santuccio, Gabriella
    Sapuppo, Gabriele
    Sorbello, Chiara Maria Catena
    Giuffrida, Gaetano
    HAEMOPHILIA, 2024, 30 : 78 - 78
  • [36] Prediction of inhibitor risk in haemophilia A using machine learning
    Lopes, Tiago Jose da Silva
    Pinotti, Mirko
    Bernardi, Francesco
    Balestra, Dario
    HAEMOPHILIA, 2024, 30 : 78 - 79
  • [37] Prediction of Intracranial Aneurysm Risk using Machine Learning
    Jaehyuk Heo
    Sang Jun Park
    Si-Hyuck Kang
    Chang Wan Oh
    Jae Seung Bang
    Tackeun Kim
    Scientific Reports, 10
  • [38] Obesity disease risk prediction using machine learning
    Dutta, Raja Ram
    Mukherjee, Indrajit
    Chakraborty, Chinmay
    INTERNATIONAL JOURNAL OF DATA SCIENCE AND ANALYTICS, 2024,
  • [39] Prediction of student attrition risk using machine learning
    Barramuno, Mauricio
    Meza-Narvaez, Claudia
    Galvez-Garcia, German
    JOURNAL OF APPLIED RESEARCH IN HIGHER EDUCATION, 2022, 14 (03) : 974 - 986
  • [40] Injury Risk Prediction in Soccer Using Machine Learning
    Shen, Brendan
    Shalaginov, Mikhail Y.
    Zeng, Tingying Helen
    22ND IEEE INTERNATIONAL CONFERENCE ON MACHINE LEARNING AND APPLICATIONS, ICMLA 2023, 2023, : 2103 - 2106