A Cross-institutional Evaluation on Breast Cancer Phenotyping NLP Algorithms on Electronic Health Records

Sicheng Zhou,Nan Wang,Liwei Wang,Ju Sun,Anne Blaes,Hongfang Liu,Rui Zhang
DOI: https://doi.org/10.1016/j.csbj.2023.08.018
IF: 6.155
2023-08-24
Computational and Structural Biotechnology Journal
Abstract:Highlights • Evaluated the impact of corpus heterogeneity on the generalizability of series NLP models, including CRF, BiLSTM-CRF and BERT based models. • Novel analysis of robustness of the machine learning models using permutation dataset. • Compared two strategies for transferring machine learning models between clinical institutes, i.e., i) direct transfer vs ii) continuous fine-tuning. Objective Transformer-based language models are prevailing in the clinical domain due to their excellent performance on clinical NLP tasks. The generalizability of those models is usually ignored during the model development process. This study evaluated the generalizability of CancerBERT, a Transformer-based clinical NLP model, along with classic machine learning models, i.e., conditional random field (CRF), bi-directional long short-term memory CRF (BiLSTM-CRF), across different clinical institutes through a breast cancer phenotype extraction task. Materials and Methods Two clinical corpora of breast cancer patients were collected from the electronic health records from the University of Minnesota and Mayo Clinic, and annotated following the same guideline. We developed three types of NLP models (i.e., CRF, BiLSTM-CRF and CancerBERT) to extract cancer phenotypes from clinical texts. We evaluated the generalizability of models on different test sets with different learning strategies (model transfer vs locally trained). The entity coverage score was assessed with their association with the model performances. Results We manually annotated 200 and 161 clinical documents at UMN and MC. The corpora of the two institutes were found to have higher similarity between the target entities than the overall corpora. The CancerBERT models obtained the best performances among the independent test sets from two clinical institutes and the permutation test set. The CancerBERT model developed in one institute and further fine-tuned in another institute achieved reasonable performance compared to the model developed on local data (micro-F1: 0.925 vs 0.932). Conclusions The results indicate the CancerBERT model has superior learning ability and generalizability among the three types of clinical NLP models for our named entity recognition task. It has the advantage to recognize complex entities, e.g., entities with different labels. Graphical abstract Download : Download high-res image (110KB) Download : Download full-size image
biochemistry & molecular biology
What problem does this paper attempt to address?