Prognostic prediction model for salivary gland carcinoma based on machine learning.

Journal: International journal of oral and maxillofacial surgery
Published Date:

Abstract

Although rare overall, salivary gland carcinomas (SGCs) are among the most common oral and maxillofacial malignancies. The aim of this study was to develop a machine learning-based model to predict the survival of patients with SGC. Patients in whom SGC was confirmed by histological testing and who underwent primary extirpation at the authors' institution between 1963 and 2014 were identified. Demographic and clinicopathological data with complete follow-up information were collected for analysis. Feature selection methods were used to determine the correlation between prognosis-related factors and survival in the collected patient data. The collected clinicopathological data and multiple machine learning algorithms were used to develop a survival prediction model. Three machine learning algorithms were applied to construct the prediction models. The area under the receiver operating characteristic curve (AUC) and accuracy were used to measure model performance. The best classification performance was achieved with a LightGBM algorithm (AUC = 0.83, accuracy = 0.91). This model enabled prognostic prediction of patient survival. The model may be useful in developing personalized diagnostic and treatment strategies and formulating individualized follow-up plans, as well as assisting in the communication between doctors and patients, facilitating a better understanding of and compliance with treatment.

Authors

  • W Du
    Department of Oral and Maxillofacial Surgery, Peking University School and Hospital of Stomatology, Beijing, China.
  • M Jia
    Department of Oral and Maxillofacial Surgery, Peking University School and Hospital of Stomatology, Beijing, China; Zhongguancun Hospital, Beijing, China.
  • J Li
    Department of Pulmonary and Critical Care Medicine, Beijing Anzhen Hospital, Capital Medical University, Beijing 100029, China.
  • M Gao
    Department of Dermatopathology, Institute of Dermatology, Peking Union Medical College & Chinese Academy of Medical Sciences, Nanjing, 210042, China.
  • W Zhang
    Department of Dermatopathology, Institute of Dermatology, Peking Union Medical College & Chinese Academy of Medical Sciences, Nanjing, 210042, China.
  • Y Yu
  • H Wang
    Department of Mechanical Engineering, Columbia University, 500 West 120th Street, New York, NY 10027, USA.
  • X Peng
    Department of Oral and Maxillofacial Surgery, Peking University School and Hospital of Stomatology & National Center for Stomatology & National Clinical Research Center for Oral Diseases & National Engineering Research Center of Oral Biomaterials and Digital Medical Devices & Beijing Key Laboratory of Digital Stomatology, Beijing 100081, China.