Hierarchical attention networks for information extraction from cancer pathology reports

被引:82
作者
Gao, Shang [1 ]
Young, Michael T. [1 ]
Qiu, John X. [1 ]
Yoon, Hong-Jun [1 ]
Christian, James B. [1 ]
Fearn, Paul A. [2 ]
Tourassi, Georgia D. [1 ]
Ramanthan, Arvind [1 ]
机构
[1] Oak Ridge Natl Lab, Computat Sci & Engn Div, Oak Ridge, TN USA
[2] NCI, Surveillance Informat Branch, Div Canc Control & Populat Sci, Bethesda, MD 20892 USA
基金
美国国家卫生研究院;
关键词
clinical pathology reports; information retrieval; recurrent neural nets; attention networks; classification;
D O I
10.1093/jamia/ocx131
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Objective: We explored how a deep learning (DL) approach based on hierarchical attention networks (HANs) can improve model performance for multiple information extraction tasks from unstructured cancer pathology reports compared to conventional methods that do not sufinotsignciently capture syntactic and semantic contexts from free-text documents. Materials and Methods: Data for our analyses were obtained from 942 deidentiinotsigned pathology reports collected by the National Cancer Institute Surveillance, Epidemiology, and End Results program. The HAN was implemented for 2 information extraction tasks: (1) primary site, matched to 12 International Classification of Diseases for Oncology topography codes (7 breast, 5 lung primary sites), and (2) histological grade classiinotsigncation, matched to G1-G4. Model performance metrics were compared to conventional machine learning (ML) approaches including naive Bayes, logistic regression, support vector machine, random forest, and extreme gradient boosting, and other DL models, including a recurrent neural network (RNN), a recurrent neural network with attention (RNN w/A), and a convolutional neural network. Results: Our results demonstrate that for both information tasks, HAN performed signiinotsigncantly better compared to the conventional ML and DL techniques. In particular, across the 2 tasks, the mean micro and macro F-scores for the HAN with pretraining were (0.852,0.708), compared to naive Bayes (0.518, 0.213), logistic regression (0.682, 0.453), support vector machine (0.634, 0.434), random forest (0.698, 0.508), extreme gradient boosting (0.696, 0.522), RNN (0.505, 0.301), RNN w/A (0.637, 0.471), and convolutional neural network (0.714, 0.460). Conclusions: HAN-based DL models show promise in information abstraction tasks within unstructured clinical pathology reports.
引用
收藏
页码:321 / 330
页数:10
相关论文
共 29 条
[1]  
[Anonymous], 2014, Generating sequences with recurrent neural networks
[2]  
[Anonymous], OPTIMIZING HYPERPARA
[3]  
[Anonymous], 2014, C EMPIRICAL METHODS
[4]  
[Anonymous], COD GUID BREAST C500
[5]  
[Anonymous], IEEE J BIOMED HLTH I
[6]  
[Anonymous], 2016, INNS C BIG DATA
[7]  
[Anonymous], 2016, IEEE T NEURAL NETW L
[8]  
[Anonymous], 2017, OV SEER PROGR
[9]  
[Anonymous], 2016, ARXIV160309381
[10]   Using Natural Language Processing to Improve Efficiency of Manual Chart Abstraction in Research: The Case of Breast Cancer Recurrence [J].
Carrell, David S. ;
Halgrim, Scott ;
Diem-Thy Tran ;
Buist, Diana S. M. ;
Chubak, Jessica ;
Chapman, Wendy W. ;
Savova, Guergana .
AMERICAN JOURNAL OF EPIDEMIOLOGY, 2014, 179 (06) :749-758