Title
Learning Multiple Diagnosis Codes for ICU Patients with Local Disease Correlation Mining.
Abstract
In the era of big data, a mechanism that can automatically annotate disease codes to patients’ records in the medical information system is in demand. The purpose of this work is to propose a framework that automatically annotates the disease labels of multi-source patient data in Intensive Care Units (ICUs). We extract features from two main sources, medical charts and notes. The Bag-of-Words model is used to encode the features. Unlike most of the existing multi-label learning algorithms that globally consider correlations between diseases, our model learns disease correlation locally in the patient data. To achieve this, we derive a local disease correlation representation to enrich the discriminant power of each patient data. This representation is embedded into a unified multi-label learning framework. We develop an alternating algorithm to iteratively optimize the objective function. Extensive experiments have been conducted on a real-world ICU database. We have compared our algorithm with representative multi-label learning algorithms. Evaluation results have shown that our proposed method has state-of-the-art performance in the annotation of multiple diagnostic codes for ICU patients. This study suggests that problems in the automated diagnosis code annotation can be reliably addressed by using a multi-label learning model that exploits disease correlation. The findings of this study will greatly benefit health care and management in ICU considering that the automated diagnosis code annotation can significantly improve the quality and management of health care for both patients and caregivers.
Year
DOI
Venue
2017
10.1145/3003729
TKDD
Keywords
Field
DocType
Diagnosis code annotation,ICU data mining,pattern discovery,MIMIC II database,multi-label learning,local correlation exploiting
Health care,Information system,ENCODE,Data mining,Diagnosis code,Annotation,Computer science,Correlation,Artificial intelligence,Intensive care,Big data,Machine learning
Journal
Volume
Issue
ISSN
11
3
1556-4681
Citations 
PageRank 
References 
22
0.62
44
Authors
6
Name
Order
Citations
PageRank
Sen Wang147737.24
Xue Li22196186.96
Xiaojun Chang3158576.85
Lina Yao498193.63
Quan Z. Sheng53520301.77
Guodong Long665547.27