RT Journal Article
SR Electronic
T1 Early-Stage NSCLC Patients’ Prognostic Prediction with Multi-information Using Transformer and Graph Neural Network Model
JF medRxiv
FD Cold Spring Harbor Laboratory Press
SP 2022.06.14.22276385
DO 10.1101/2022.06.14.22276385
A1 Lian, Jie
A1 Deng, Jiajun
A1 Hui, Sai Kam
A1 Koohi-Moghadam, Mohamad
A1 She, Yunlang
A1 Chen, Chang
A1 Vardhanabhuti, Varut
YR 2022
UL http://medrxiv.org/content/early/2022/06/16/2022.06.14.22276385.abstract
AB Background We proposed a population graph with Transformer-generated and clinical features for the purpose of predicting overall survival and recurrence-free survival for patients with early-stage NSCLC and to compare this model with traditional models.Methods The study included 1705 patients with lung cancer (stage I and II), and a public dataset for external validation (n=127). We proposed a graph with edges representing non-imaging patient characteristics and nodes representing imaging tumour region characteristics generated by a pretrained Vision Transformer. The model was compared with a TNM model and a ResNet-Graph model. To evaluate the models’ performance, the area under the receiver operator characteristic curve (ROC-AUC) was calculated for both overall survival (OS) and recurrence-free survival (RFS) prediction. The Kaplan–Meier method was used to generate prognostic and survival estimates for low- and high-risk groups, along with net reclassification improvement (NRI), integrated discrimination improvement (IDI), and decision curve analysis (DCA). An additional subanalysis was conducted to examine the relationship between clinical data and imaging features associated with risk prediction.Results Our model achieved AUC values of 0·785 (95 % CI:0·716 - 0·855) and 0·695 (95 % CI:0·603 - 0·787) on the testing and external datasets for OS prediction, and 0·726 (95 % CI:0·653 - 0·800) and 0·700 (95 % CI:0·615 - 0·785) for RFS prediction. Additional survival analyses indicated that our model outperformed the present TNM and ResNet-Graph models in terms of net benefit for survival prediction.Conclusion Our Transformer-Graph model was effective at predicting survival in patients with early-stage lung cancer, which was constructed using both imaging and non-imaging clinical features. Some high-risk patients were distinguishable by using a similarity score function defined by non-imaging characteristics such as age, gender, histology type, and tumour location, while Transformer-generated features demonstrated additional benefits for patients whose non-imaging characteristics were non-discriminatory for survival outcomes.Funding There was no funding source for this study.Competing Interest StatementThe authors have declared no competing interest.Funding StatementNo external funding was receivedAuthor DeclarationsI confirm all relevant ethical guidelines have been followed, and any necessary IRB and/or ethics committee approvals have been obtained.Not ApplicableThe details of the IRB/oversight body that provided approval or exemption for the research described are given below:The study protocol was approved by the Shanghai Pulmonary Hospital's Institutional Review Board and informed consent was waived owing to retrospective nature.I confirm that all necessary patient/participant consent has been obtained and the appropriate institutional forms have been archived, and that any patient/participant/sample identifiers included were not known to anyone (e.g., hospital staff, patients or participants themselves) outside the research group so cannot be used to identify individuals.Not ApplicableI understand that all clinical trials and any other prospective interventional studies must be registered with an ICMJE-approved registry, such as ClinicalTrials.gov. I confirm that any such study reported in the manuscript has been registered and the trial registration ID is provided (note: if posting a prospective study registered retrospectively, please provide a statement in the trial ID field explaining why the study was not registered in advance).Not ApplicableI have followed all appropriate research reporting guidelines and uploaded the relevant EQUATOR Network research reporting checklist(s) and other pertinent material as supplementary files, if applicable.Not ApplicableData is available from authors upon reasonable request.