{
  "abstract": "Objectives To develop a robust and interpretable machine learning framework for arthritis risk prediction and to identify important risk factors associated with the disease.Methods This study used four datasets, which are Behavioral Risk Factor Surveillance System (BRFSS) 2019, BRFSS 2021, National Health Interview Survey (NHIS) 2020 and NHIS 2022. We evaluated 11 machine learning and deep learning architectures including a custom stacked ensemble combined with five resampling techniques to address class imbalance and 9 imputation methods to handle missing data. Area under the receiver operating characteristic curve (AUROC), sensitivity, specificity and balanced accuracy are used to evaluate model performance, with fivefold cross-validation to ensure robustness. SHapley Additive exPlanations (SHAP) provided feature interpretability in prediction.Results The combination of the stacked ensemble, Generative Adversarial Network (GAN) imputation and random undersampling (RUS) achieved the best performance with an AUROC of 0.8007 and sensitivity of 0.8060 in the test set of BRFSS 2019. This combination also showed a balanced performance across both subgroups of male and female by maintaining high performance with AUROC of 0.808 and 0.800, respectively, in the test set of BRFSS 2021. The stacked ensemble model with GAN imputation and either Random Undersampling (RUS) or Random Oversampling (ROS) achieved the best performance across all four datasets. SHAP analysis identified higher age, walking difficulty and high body mass index (BMI) as the top physical factors. Among the psychosocial factors, parental separation showed a high impact on predicting arthritis.Discussion The risk of arthritis is influenced by health, lifestyle and childhood experiences. SHAP shows that difficulty walking, age, BMI and childhood stress play important roles in predicting arthritis risk.Conclusion These results confirm important predictors of arthritis risk and provide a clear, interpretable framework.",
  "authors": [
    {
      "affiliations": [
        "Department of Computer Science and Engineering, Notre Dame University Bangladesh, Dhaka, Bangladesh",
        "Department of Computer Science and Engineering, University of Asia Pacific, Dhaka, Bangladesh"
      ],
      "name": "Md. Atik Shams"
    },
    {
      "affiliations": [
        "Department of Computer Science and Engineering, University of Asia Pacific, Dhaka, Bangladesh"
      ],
      "name": "Sumaiya Fatema"
    },
    {
      "affiliations": [
        "Department of Computer Science and Engineering, University of Asia Pacific, Dhaka, Bangladesh"
      ],
      "name": "D M Hasibul Islam"
    },
    {
      "affiliations": [
        "Department of Fish and Wildlife Conservation, Virginia Tech, Blacksburg, Virginia, USA"
      ],
      "name": "Anindita Datta"
    },
    {
      "affiliations": [
        "Department of Information Management and Business Analytics, Montclair State University, Montclair, New Jersey, USA"
      ],
      "name": "David Eisenberg"
    },
    {
      "affiliations": [
        "Department of Computer Science and Engineering, Northern University Bangladesh, Dhaka, Bangladesh"
      ],
      "name": "Danastan Tasaouf Mridula"
    },
    {
      "affiliations": [
        "Department of Computer Science and Engineering, University of Asia Pacific, Dhaka, Bangladesh"
      ],
      "name": "Junnatul Mawa"
    },
    {
      "affiliations": [
        "Department of Computer Science and Engineering, University of Asia Pacific, Dhaka, Bangladesh"
      ],
      "name": "Asma Sultana"
    },
    {
      "affiliations": [
        "Department of Computer Science, BRAC University, Dhaka, Bangladesh"
      ],
      "name": "Nafiya Ahmed"
    },
    {
      "affiliations": [
        "Department of Electrical and Electronic Engineering, Islamic University, Kushtia District, Bangladesh"
      ],
      "name": "Monowarul Islam"
    },
    {
      "affiliations": [
        "Stanford Medicine, Stanford University, Stanford, California, USA",
        "Department of Computer Science, Virginia Tech, Blacksburg, Virginia, USA"
      ],
      "name": "Tanmoy Sarkar Pias"
    }
  ],
  "title": "Explainable machine learning revealing the impact of mental and physical health on arthritis",
  "uid": "d7d94fa7-854d-5952-be7c-4dd8bb142cf8"
}