IDEAS home Printed from https://ideas.repec.org/a/gam/jmathe/v11y2023i14p3145-d1195658.html
   My bibliography  Save this article

Predicting Depression during the COVID-19 Pandemic Using Interpretable TabNet: A Case Study in South Korea

Author

Listed:
  • Hung Viet Nguyen

    (Department of Digital Anti-Aging Healthcare (BK21), Inje University, Gimhae 50834, Republic of Korea)

  • Haewon Byeon

    (Department of Digital Anti-Aging Healthcare (BK21), Inje University, Gimhae 50834, Republic of Korea)

Abstract

COVID-19 has further aggravated problems by compelling people to stay indoors and limit social interactions, leading to a worsening of the depression situation. This study aimed to construct a TabNet model combined with SHapley Additive exPlanations (SHAP) to predict depression in South Korean society during the COVID-19 pandemic. We used a tabular dataset extracted from the Seoul Welfare Survey with a total of 3027 samples. The TabNet model was trained on this dataset, and its performance was compared to that of several other machine learning models, including Random Forest, eXtreme Gradient Boosting, Light Gradient Boosting, and CatBoost. According to the results, the TabNet model achieved an Area under the receiver operating characteristic curve value (AUC) of 0.9957 on the training set and an AUC of 0.9937 on the test set. Additionally, the study investigated the TabNet model’s local interpretability using SHapley Additive exPlanations (SHAP) to provide post hoc global and local explanations for the proposed model. By combining the TabNet model with SHAP, our proposed model might offer a valuable tool for professionals in social fields, and psychologists without expert knowledge in the field of data analysis can easily comprehend the decision-making process of this AI model.

Suggested Citation

  • Hung Viet Nguyen & Haewon Byeon, 2023. "Predicting Depression during the COVID-19 Pandemic Using Interpretable TabNet: A Case Study in South Korea," Mathematics, MDPI, vol. 11(14), pages 1-21, July.
  • Handle: RePEc:gam:jmathe:v:11:y:2023:i:14:p:3145-:d:1195658
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2227-7390/11/14/3145/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2227-7390/11/14/3145/
    Download Restriction: no
    ---><---

    References listed on IDEAS

    as
    1. Joanna F Dipnall & Julie A Pasco & Michael Berk & Lana J Williams & Seetal Dodd & Felice N Jacka & Denny Meyer, 2016. "Fusing Data Mining, Machine Learning and Traditional Statistics to Detect Biomarkers Associated with Depression," PLOS ONE, Public Library of Science, vol. 11(2), pages 1-23, February.
    2. Hung Viet Nguyen & Haewon Byeon, 2023. "Prediction of Out-of-Hospital Cardiac Arrest Survival Outcomes Using a Hybrid Agnostic Explanation TabNet Model," Mathematics, MDPI, vol. 11(9), pages 1-17, April.
    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. Joanna F Dipnall & Julie A Pasco & Michael Berk & Lana J Williams & Seetal Dodd & Felice N Jacka & Denny Meyer, 2016. "Into the Bowels of Depression: Unravelling Medical Symptoms Associated with Depression by Applying Machine-Learning Techniques to a Community Based Population Sample," PLOS ONE, Public Library of Science, vol. 11(12), pages 1-19, December.
    2. Nkiruka C. Atuegwu & Cheryl Oncken & Reinhard C. Laubenbacher & Mario F. Perez & Eric M. Mortensen, 2020. "Factors Associated with E-Cigarette Use in U.S. Young Adult Never Smokers of Conventional Cigarettes: A Machine Learning Approach," IJERPH, MDPI, vol. 17(19), pages 1-16, October.

    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:gam:jmathe:v:11:y:2023:i:14:p:3145-:d:1195658. 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: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    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.