Gaussian process regression and classification using International Classification of Disease codes as covariates

Abstract

In electronic health records (EHRs) data analysis, nonparametric regression and classification using International Classification of Disease (ICD) codes as covariates remain understudied. Automated methods have been developed over the years for predicting biomedical responses using EHRs, but relatively less attention has been paid to developing patient similarity measures that use ICD codes and chronic conditions, where a chronic condition is defined as a set of ICD codes. We address this problem by first developing a string kernel function for measuring the similarity between a pair of primary chronic conditions, represented as subsets of ICD codes. Second, we extend this similarity measure to a family of covariance functions on subsets of chronic conditions. This family is used in developing Gaussian process (GP) priors for Bayesian nonparametric regression and classification using diagnoses and other demographic information as covariates. Markov chain Monte Carlo (MCMC) algorithms are used for posterior inference and predictions. The proposed methods are tuning free, so they are ideal for automated prediction of biomedical responses depending on chronic conditions. We evaluate the practical performance of our method on EHR data collected from 1660 patients at the University of Iowa Hospitals and Clinics (UIHC) with six different primary cancer sites. Our method provides better sensitivity and specificity than its competitors in classifying different primary cancer sites and estimates the marginal associations between chronic conditions and primary cancer sites.

Document Details

Document Type
Pub Defense Publication
Publication Date
Jan 01, 2023
Source ID
10.1002/sta4.618

Entities

People

  • Sanvesh Srivastava
  • Stephanie Gilbertson‐white
  • W. Nick Street
  • Yunyi Li
  • Zongyi Xu

Organizations

  • National Science Foundation
  • Office of Naval Research
  • University of Iowa
  • University of Texas at Austin

Tags

Readers

  • Medical or Health Care Field.
  • Regression Analysis.
  • Statistical inference.

Technology Areas

  • AI & ML
  • AI & ML - Bayesian Inference
  • Biotechnology
  • Microelectronics
  • Space