Deakin University
Browse

File(s) under permanent embargo

Prediction model for recurrence of hepatocellular carcinoma after resection by using neighbor2vec based algorithms

journal contribution
posted on 2021-03-01, 00:00 authored by Yuankui Cao, Junqing Fan, Hong Cao, Yunliang Chen, Jie Li, Jianxin LiJianxin Li, Simin Zhang
Liver cancer has become the third cause that leads to the cancer death. For hepatocellular carcinoma (HCC), as the highly malignant type of liver cancer, its recurrence rate after operation is still very high because there is no reliable clinical data to provide better advice for patients after operation. To solve the challenging issue, in this work, we design a novel prediction model for recurrence of HCC using neighbor2vec based algorithm. It consists of three stages: (a) In the preparation stage, the Pearson correlation coefficient was used to explore the independent predictors of HCC recurrence, (b) due to the low correlation between individual dimension and prediction target, K‐nearest neighbors (KNN) were found as a K‐vectors list for each patient (neighbor2vec), (c) all vectors lists were applied as the input of machine learning methods such as logistic regression, KNN, decision tree, naive Bayes (NB), and deep neural network to establish the neighbor2vec based prediction model. From the experimental results on the real data from Shandong Provincial Hospital in China, the proposed neighbor2vec based prediction model outperforms all the other models. Especially, the NB model with neighbor2vec achieves up to 83.02, 82.86, 77.6%, in terms of accuracy, recall rates, and precision.

History

Journal

WIREs : Data Mining and Knowledge Discovery

Volume

11

Issue

2

Season

March/April 2021

Article number

e1390

Publisher

John Wiley & Sons

Location

Oxford, Eng.

ISSN

1942-4787

eISSN

1942-4795

Language

eng

Publication classification

C1 Refereed article in a scholarly journal

Copyright notice

2020, Wiley Periodicals