3D-CNN-SPP: A patient risk prediction system from electronic health records via 3D CNN and spatial pyramid pooling

Ronghui Ju, Pan Zhou, Shiping Wen, Wei Wei, Yuan Xue, Xiaolei Huang, Xin Yang

Research output: Contribution to journalArticlepeer-review

12 Scopus citations

Abstract

The problem of extracting useful clinical representations from longitudinal electronic health record (EHR) data, also known as the computational phenotyping problem, is an important yet challenging task in the health-care academia and industry. Recent progress in the design and applications of deep learning methods has shown promising results towards solving this problem. In this paper, we propose 3D-CNN-SPP (3D Convolutional Neural Networks and Spatial Pyramid Pooling), a novel patient risk prediction system, to investigate the application of deep neural networks in modeling longitudinal EHR data. Particularly, we propose a 3D CNN structure, which is featured by SPP. Compared with 2D CNN methods, our proposed method can capture the complex relationships in EHRs more effectively and efficiently. Furthermore, previous works handle the issue of variable length in patient records by padding zeros to all vectors so that they have a fixed length. In our work, the proposed spatial pyramid pooling divides the records into several length sections for respective pooling processing, hence handling the variable length problem easily and naturally. We take heart failure and diabetes as examples to test the performance of the system, and the experiment results demonstrate great effectiveness in patient risk prediction, compared with several strong baselines.

Original languageEnglish (US)
Article number8949441
Pages (from-to)247-261
Number of pages15
JournalIEEE Transactions on Emerging Topics in Computational Intelligence
Volume5
Issue number2
DOIs
StatePublished - Apr 2021

All Science Journal Classification (ASJC) codes

  • Computer Science Applications
  • Control and Optimization
  • Computational Mathematics
  • Artificial Intelligence

Fingerprint

Dive into the research topics of '3D-CNN-SPP: A patient risk prediction system from electronic health records via 3D CNN and spatial pyramid pooling'. Together they form a unique fingerprint.

Cite this