IDEAS home Printed from https://ideas.repec.org/a/plo/pone00/0306090.html
   My bibliography  Save this article

A deep neural network prediction method for diabetes based on Kendall’s correlation coefficient and attention mechanism

Author

Listed:
  • Xiaobo Qi
  • Yachen Lu
  • Ying Shi
  • Hui Qi
  • Lifang Ren

Abstract

Diabetes is a chronic disease, which is characterized by abnormally high blood sugar levels. It may affect various organs and tissues, and even lead to life-threatening complications. Accurate prediction of diabetes can significantly reduce its incidence. However, the current prediction methods struggle to accurately capture the essential characteristics of nonlinear data, and the black-box nature of these methods hampers its clinical application. To address these challenges, we propose KCCAM_DNN, a diabetes prediction method that integrates Kendall’s correlation coefficient and an attention mechanism within a deep neural network. In the KCCAM_DNN, Kendall’s correlation coefficient is initially employed for feature selection, which effectively filters out key features influencing diabetes prediction. For missing values in the data, polynomial regression is utilized for imputation, ensuring data completeness. Subsequently, we construct a deep neural network (KCCAM_DNN) based on the self-attention mechanism, which assigns greater weight to crucial features affecting diabetes and enhances the model’s predictive performance. Finally, we employ the SHAP model to analyze the impact of each feature on diabetes prediction, augmenting the model’s interpretability. Experimental results show that KCCAM_DNN exhibits superior performance on both PIMA Indian and LMCH diabetes datasets, achieving test accuracies of 99.090% and 99.333%, respectively, approximately 2% higher than the best existing method. These results suggest that KCCAM_DNN is proficient in diabetes prediction, providing a foundation for informed decision-making in the diagnosis and prevention of diabetes.

Suggested Citation

  • Xiaobo Qi & Yachen Lu & Ying Shi & Hui Qi & Lifang Ren, 2024. "A deep neural network prediction method for diabetes based on Kendall’s correlation coefficient and attention mechanism," PLOS ONE, Public Library of Science, vol. 19(7), pages 1-19, July.
  • Handle: RePEc:plo:pone00:0306090
    DOI: 10.1371/journal.pone.0306090
    as

    Download full text from publisher

    File URL: https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0306090
    Download Restriction: no

    File URL: https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0306090&type=printable
    Download Restriction: no

    File URL: https://libkey.io/10.1371/journal.pone.0306090?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    References listed on IDEAS

    as
    1. Patrick Royston, 2005. "Multiple imputation of missing values: update," Stata Journal, StataCorp LLC, vol. 5(2), pages 188-201, June.
    2. Patrick Royston, 2005. "Multiple imputation of missing values: Update of ice," Stata Journal, StataCorp LLC, vol. 5(4), pages 527-536, December.
    3. Patrick Royston, 2005. "MICE for multiple imputation of missing values," United Kingdom Stata Users' Group Meetings 2005 02, Stata Users Group.
    Full references (including those not matched with items on IDEAS)

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Lee, Chioun & Ryff, Carol D., 2016. "Early parenthood as a link between childhood disadvantage and adult heart problems: A gender-based approach," Social Science & Medicine, Elsevier, vol. 171(C), pages 58-66.
    2. Denney, Justin T. & Brewer, Mackenzie & Kimbro, Rachel Tolbert, 2020. "Food insecurity in households with young children: A test of contextual congruence," Social Science & Medicine, Elsevier, vol. 263(C).
    3. Gerko Vink & Laurence E. Frank & Jeroen Pannekoek & Stef Buuren, 2014. "Predictive mean matching imputation of semicontinuous variables," Statistica Neerlandica, Netherlands Society for Statistics and Operations Research, vol. 68(1), pages 61-90, February.
    4. Watkins, Adam M. & Melde, Chris, 2018. "Gangs, gender, and involvement in crime, victimization, and exposure to violence," Journal of Criminal Justice, Elsevier, vol. 57(C), pages 11-25.
    5. HwaJung Choi & Robert F. Schoeni & Kenneth M. Langa & Michele M. Heisler, 2015. "Spouse and Child Availability for Newly Disabled Older Adults: Socioeconomic Differences and Potential Role of Residential Proximity," The Journals of Gerontology: Series B, The Gerontological Society of America, vol. 70(3), pages 462-469.
    6. Simon Grund & Oliver Lüdtke & Alexander Robitzsch, 2018. "Multiple Imputation of Missing Data at Level 2: A Comparison of Fully Conditional and Joint Modeling in Multilevel Designs," Journal of Educational and Behavioral Statistics, , vol. 43(3), pages 316-353, June.
    7. Jason R. D. Rarick & Carly Tubbs Dolan & Wen‐Jui Han & Jun Wen, 2018. "Relations Between Socioeconomic Status, Subjective Social Status, and Health in Shanghai, China," Social Science Quarterly, Southwestern Social Science Association, vol. 99(1), pages 390-405, March.
    8. David W Lawson & Arijeta Makoli & Anna Goodman, 2013. "Sibling Configuration Predicts Individual and Descendant Socioeconomic Success in a Modern Post-Industrial Society," PLOS ONE, Public Library of Science, vol. 8(9), pages 1-9, September.
    9. HwaJung Choi & Robert F. Schoeni & Kenneth M. Langa & Michele M. Heisler, 2015. "Older Adults’ Residential Proximity to Their Children: Changes After Cardiovascular Events," The Journals of Gerontology: Series B, The Gerontological Society of America, vol. 70(6), pages 995-1004.
    10. Brewer, Mackenzie & Kimbro, Rachel Tolbert, 2014. "Neighborhood context and immigrant children's physical activity," Social Science & Medicine, Elsevier, vol. 116(C), pages 1-9.
    11. Lee, RaeHyuck & Brooks-Gunn, Jeanne & Han, Wen-Jui & Waldfogel, Jane & Zhai, Fuhua, 2014. "Is participation in Head Start associated with less maternal spanking for boys and girls?," Children and Youth Services Review, Elsevier, vol. 46(C), pages 55-63.
    12. Lombardi, Caitlin McPherran, 2021. "Family income and mothers’ parenting quality: Within-family associations from infancy to late childhood," Children and Youth Services Review, Elsevier, vol. 120(C).
    13. Minda Tan & Shuiyun Liu, 2023. "A Way of Human Capital Accumulation: Heterogeneous Impact of Shadow Education on Students’ Academic Performance in China," SAGE Open, , vol. 13(4), pages 21582440231, November.
    14. Angela Cipollone & Carlo D'Ippoliti, 2009. "Women's Employment: Beyond Individual Characteristics vs. Contextual Factors Explanations," Working Papers CELEG 0901, Dipartimento di Economia e Finanza, LUISS Guido Carli.
    15. King, Christian, 2018. "Food insecurity and child behavior problems in fragile families," Economics & Human Biology, Elsevier, vol. 28(C), pages 14-22.
    16. Emma Zang & Anthony R. Bardo, 2019. "Objective and Subjective Socioeconomic Status, Their Discrepancy, and Health: Evidence from East Asia," Social Indicators Research: An International and Interdisciplinary Journal for Quality-of-Life Measurement, Springer, vol. 143(2), pages 765-794, June.
    17. Álvaro Choi & María Gil & Mauro Mediavilla & Javier Valbuena, 2016. "Double toil and trouble: grade retention and academic performance," Working Papers 2016/7, Institut d'Economia de Barcelona (IEB).
    18. Oya Kalaycioglu & Andrew Copas & Michael King & Rumana Z. Omar, 2016. "A comparison of multiple-imputation methods for handling missing data in repeated measurements observational studies," Journal of the Royal Statistical Society Series A, Royal Statistical Society, vol. 179(3), pages 683-706, June.
    19. Shaun R. Seaman & Ian R. White & Andrew J. Copas & Leah Li, 2012. "Combining Multiple Imputation and Inverse-Probability Weighting," Biometrics, The International Biometric Society, vol. 68(1), pages 129-137, March.
    20. Asit Bhattacharyya & Lorne Cummings, 2015. "Measuring Corporate Environmental Performance – Stakeholder Engagement Evaluation," Business Strategy and the Environment, Wiley Blackwell, vol. 24(5), pages 309-325, July.

    More about this item

    Statistics

    Access and download statistics

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:plo:pone00:0306090. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: plosone (email available below). General contact details of provider: https://journals.plos.org/plosone/ .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.