计算机与现代化 ›› 2021, Vol. 0 ›› Issue (11): 7-11.

• 人工智能 • 上一篇    下一篇

基于SMOTE和RNN的肾移植排斥反应预测

  

  1. (1.北京信息科技大学计算机学院,北京100101;2.北京大学第三医院,北京100191)
  • 出版日期:2021-12-13 发布日期:2021-12-13
  • 作者简介:杨欣怡(1996—),女,河北邢台人,硕士研究生,研究方向:机器学习,算法研究与设计,E-mail: 2018021002@mail.bistu.edu.cn; 通信作者:侯凌燕(1964—),女,教授,硕士,研究方向:模式识别,机器视觉,E-mail: lyhou@bistu.edu.cn; 杨大利(1963—),男,副教授,博士,研究方向:语音识别,计算机视觉,信号增强,E-mail: yangdali@bistu.edu.cn; 崔丽艳(1974—),女,教授,博士,研究方向:临床生物化学,临床免疫学检验,E-mail: cliyan@163.com。
  • 基金资助:
    国家自然科学基金资助项目(6177010360)

Prediction of Renal Transplantation Rejection Based on SMOTE and RNN

  1. (1. School of Computer Science, Beijing Information Science & Technology University, Beijing 100101, China;
    2. Peking University Third Hospital, Beijing 100191, China)
  • Online:2021-12-13 Published:2021-12-13

摘要: 肾移植手术在当今的应用越来越广泛,对于排斥反应的预测变得更加重要。针对排斥反应数据特点中存在的数据的维度高、数据时序性、样本不均衡等问题,将循环神经网络应用于肾移植排斥反应的预测,本文提出一种结合SMOTE(Synthetic Minority Over-sampling Technique)以及RNN(Recurrent Neural Network)的算法。该方法先处理数据,降低正负样本的不平衡度,且解决样本量不足的问题,再根据RNN的学习过程进行关键参数调整、优化。经过实验发现,该方法可以有效提升正负分类的准确率,与传统的马尔可夫时间序列预测算法相比,准确率提高了16.7%,传统RNN训练经过优化后,相对错误率下降了5.03%,可以使用该方法进行肾移植排斥反应的有效预测。

关键词: 肾移植排斥反应, 序列分类, 循环神经网络, SMOTE

Abstract: Nowadays, kidney transplantation is more and more widely used, and the prediction of rejection becomes more and more important. In order to solve the problems of high dimension, time sequence and sample imbalance in rejection data, recurrent neural network is applied to predict renal transplantation rejection, in this paper, an algorithm combining SMOTE (Synthetic Minimum Over-sampling Technology) with RNN (Recurrent Neural Network) is proposed. The method first processes the data, reduces the imbalance between samples, and solves the problem of insufficient sample size. Then the key parameters are adjusted and optimized according to the learning process of RNN. The experimental results show that this method can effectively improve the accuracy of positive and negative classification. Compared with the traditional Markov time series prediction algorithm, the accuracy is improved by 16.7%. After the traditional RNN training is optimized, the relative error rate is reduced by 5.03%. This method can be used to effectively predict renal transplantation rejection.

Key words: renal transplantation rejection, sequence classification, RNN, SMOTE