XGBoost-SHAP-based interpretable diagnostic framework for knee osteoarthritis: a population-based retrospective cohort study.

Journal: Arthritis research & therapy
Published Date:

Abstract

OBJECTIVE: To use routine demographic and clinical data to develop an interpretable individual-level machine learning (ML) model to diagnose knee osteoarthritis (KOA) and to identify highly ranked features.

Authors

  • Zijuan Fan
    Department of Orthopaedic Surgery, The First Affiliated Hospital, Zhejiang University School of Medicine, Qingchun Road No. 79, Hangzhou, China.
  • Wenzhu Song
    Department of Big Data in Health Science School of Public Health, Zhejiang University School of Medicine, Hangzhou, Zhejiang, China.
  • Yan Ke
    Arthritis Clinic & Research Center, Peking University People's Hospital, Beijing, China.
  • Ligan Jia
    School of Computer Science and Technology, Xinjiang University, Urumchi, China.
  • Songyan Li
    Department of Orthopaedic Surgery, The First Affiliated Hospital, Zhejiang University School of Medicine, Qingchun Road No. 79, Hangzhou, China.
  • Jiao Jiao Li
    School of Biomedical Engineering, Faculty of Engineering and IT, University of Technology Sydney, Sydney, Australia.
  • Yuqing Zhang
    Division of Rheumatology, Allergy, and Immunology, Harvard Medical School, 2348Massachusetts General Hospital, Boston, MA, USA.
  • Jianhao Lin
    Arthritis Clinic and Research Center, Peking University People's Hospital, No. 11 Xicheng District, Beijing 100044, China. Electronic address: jianhao_lin@hotmail.com.
  • Bin Wang
    State Key Laboratory of Soil Erosion and Dryland Farming on the Loess Plateau, Northwest A&F University, Yangling 712100, China; New South Wales Department of Primary Industries, Wagga Wagga Agricultural Institute, Wagga Wagga 2650, Australia. Electronic address: bin.a.wang@dpi.nsw.gov.au.