You are on page 1of 12

Advanced Machine Learning

Predicting Metastasis in Gastric Cancer Patients:


Machine Learning-Based Approaches
Background


Data Collection

Data Preprocessing


Model Development


Statistical Analysis


1. Baseline features of GC patients classified as metastatic or non-metastatic. 2. Model performance in ML techniques using k-fold cross validation with both data.

Metastasis P value Models Dataset Sensitivity% Specificity% Precision% F1 AUC


No Yes
LR Original data Train .86 .86 .76 .81 .93
(n = 471) (n = 262)
Test .86 .78 .77 .81 .88*
Age, mean (SD) 59.53 (12.56) 60.02 (13.18) .616
Sex, n (%) Balanced data Train .93 .83 .88 .87 .93
Female 130 (61.6) 81 (38.4) .342
Test .90 .78 .84 .84 .86
Male 341 (65.3) 181 (34.7)
Marital Status, n (%) NN Original data Train .93 .93 .88 .88 .98*
Married 461 (64.0) 259 (36.0) .398
Test .76 .78 .75 .77 .86
Single 10 (76.9) 3 (23.1)
BMI, n (%) Balanced data Train .98-3 .87 .95 .95 .99*
Underweight 99 (62.7) 59 (37.3) .321
Test .91 .81-2 .86 .86* .87*
Normal 285 (63.6) 163 (36.4)
Overweight 60 (67.4) 29 (32.6) RF Original data Train .89 .95 .92 .90 .98*
Obese 27 (71.1) 11 (28.9)
Test .80 .83 .80 .80 .87
Smoking, n (%)
Never used 306 (66.2) 156 (33.8) .145 Balanced data Train .98-3 .95 .96 .96 .99*
Previous or current user 165 (60.9) 106 (39.1)
Family history, n (%) Test .91 .78 .85 .85 .87*
No 340 (61.8) 210 (38.2) .017 NB Original data Train .83 .85 .74 .79 .90
Yes 131 (71.6) 52 (28.4)
Grade of tumour, n (%) Test .89 .78 .77 .83 .88*
Poorly 157 (78.5) 43 (21.5) < .001 Balanced data Train .87 .84 .86 .86 .91
Moderately 114 (83.2) 23 (16.8)
Well 85 (76.6) 26 (23.4) Test .86 .81-2 .83 .83 .86
Undifferentiated 115 (40.4) 170 (59.6) DT Original data Train .82 .97 .94 .88 .96
Number of involved lymph node, n (%)
N1 154 (73.3) 56 (26.7) .007 Test .58 .78 .69 .63 .75
N2 251 (60.2) 166 (39.8) Balanced data Train .94 .96 .95 .95 .98
N3 66 (62.3) 40 (37.7)
Tumour size, n (%) Test .80 .81 .80 .80 .83
T1 25 (89.3) 3 (10.7) < .001
SVM Original data Train .94 .86 .77 .84 .94
T2 68 (93.2) 5 (6.8)
T3 276 (94.8) 15 (5.2) Test .92 .76 .76 .85 .85
T4 102 (29.9) 239 (70.1) Balanced data Train .98-3 .87 .93 .92 .93
Type of treatment, n (%)
Other treatments 366 (81.9) 81 (18.1) < .001 Test .93 .80 .87 .86* .85
Surgery 105 (36.7) 181 (63.3)

Models Dataset Precision-recall


curve AUC

LR Balanced data Train .91

Test .81

NN Balanced data Train .98

Test .84

RF Balanced data Train .99

Test .86

NB Balanced data Train .89

Test .81

DT Balanced data Train .98

Test .79

SVM Balanced data Train .87

Test .76
Results


Conclusion / Question?


Answer


Citations
 Atefeh Talebi, Carlos A. Celis-Morales, Nasrin Borumandnia, Somayeh Abbasi,
Mohamad Amin, Pourhoseingholi, Abolfazl Akbari & Javad Yousefi.

 Link:- https://www.nature.com/articles/s41598-023-31272-w

 Link:- https://www.freecodecamp.org/news/svm-machine-learning-tutorial-what-is-
the-support-vector-machine-algorithm-explained-with-code-
examples/#:~:text=SVMs%20are%20different%20from%20other,the%20maximum
%20margin%20hyper%20plane.


Thank you

You might also like