Automatic extraction of cancer registry reportable information from free-text pathology reports using multitask convolutional neural networks

Journal Article · Sat Nov 09 04:00:00 UTC 2019 · Journal of the American Medical Informatics Association

DOI: https://doi.org/10.1093/jamia/ocz153 · OSTI ID:1574032

Alawad, Mohammed ^[1]; Gao, Shang ^[1]; Qiu, John X. ^[1]; Yoon, Hong Jun ^[1]; Blair Christian, J. ^[1]; Penberthy, Lynne ^[2]; Mumphrey, Brent ^[3]; Wu, Xiao-Cheng ^[3]; Coyle, Linda ^[4]; Tourassi, Georgia ^[1]

---

1. Computational Sciences and Engineering Division, Health Data Sciences Institute, Oak Ridge National Laboratory, Oak Ridge, Tennessee, USA
2. Surveillance Research Program, Division of Cancer Control and Population Sciences, National Cancer Institute, Bethesda, Maryland, USA
3. Louisiana Tumor Registry, Louisiana State University Health Sciences Center School of Public Health, New Orleans, Louisiana, USA
4. Information Management Services Inc, Calverton, Maryland, USA

Objective: In this work we implement 2 different multitask learning (MTL) techniques, hard parameter sharing and cross-stitch, to train a word-level convolutional neural network (CNN) specifically designed for automatic extraction of cancer data from unstructured text in pathology reports. We show the importance of learning related information extraction (IE) tasks leveraging shared representations across the tasks to achieve state-of-the-art performance in classification accuracy and computational efficiency. Materials and Methods: Multitask CNN (MTCNN) attempts to tackle document information extraction by learning to extract multiple key cancer characteristics simultaneously. We trained our MTCNN to perform 5 information extraction tasks: (1) primary cancer site (65 classes), (2) laterality (4 classes), (3) behavior (3 classes), (4) histological type (63 classes), and (5) histological grade (5 classes). We evaluated the performance on a corpus of 95 231 pathology documents (71 223 unique tumors) obtained from the Louisiana Tumor Registry. We compared the performance of the MTCNN models against single-task CNN models and 2 traditional machine learning approaches, namely support vector machine (SVM) and random forest classifier (RFC). Results: MTCNNs offered superior performance across all 5 tasks in terms of classification accuracy as compared with the other machine learning models. Based on retrospective evaluation, the hard parameter sharing and cross-stitch MTCNN models correctly classified 59.04% and 57.93% of the pathology reports respectively across all 5 tasks. The baseline models achieved 53.68% (CNN), 46.37% (RFC), and 36.75% (SVM). Based on prospective evaluation, the percentages of correctly classified cases across the 5 tasks were 60.11% (hard parameter sharing), 58.13% (cross-stitch), 51.30% (single-task CNN), 42.07% (RFC), and 35.16% (SVM). Moreover, hard parameter sharing MTCNNs outperformed the other models in computational efficiency by using about the same number of trainable parameters as a single-task CNN. Conclusions: The hard parameter sharing MTCNN offers superior classification accuracy for automated coding support of pathology documents across a wide range of cancers and multiple information extraction tasks while maintaining similar training and inference time as those of a single task–specific model.

• Sponsoring Organization: National Cancer Institute (NCI); National Institutes of Health (NIH); USDOE Office of Science (SC)
• Grant/Contract Number: AC05-00OR22725; AC02-06CH11357; AC52-07NA27344; AC52-06NA25396
• OSTI ID: 1574032
• Journal Information: Journal of the American Medical Informatics Association, Journal Name: Journal of the American Medical Informatics Association; ISSN 1067-5027
• Publisher: Oxford University PressCopyright Statement
• Country of Publication: United Kingdom
• Language: English