Previously submitted to: JMIR AI (no longer under consideration since Feb 24, 2023)
Date Submitted: Aug 17, 2022
Warning: This is an author submission that is not peer-reviewed or edited. Preprints - unless they show as "accepted" - should not be relied on to guide clinical practice or health-related behavior and should not be reported in news media as established information.
exKidneyBERT: a Language Model for Kidney Transplant Pathology Reports and the Crucial Role of Extended Vocabularies: Algorithm Development and Validation
ABSTRACT
Background:
Pathology reports contain key information about the patient’s diagnosis as well as important gross and microscopic findings. These information-rich clinical reports offer an invaluable resource for clinical studies, but data extraction and analysis from such unstructured texts is often manual and tedious. Thus, an automated data extraction method from pathology reports would be of significant value and utility.
Objective:
to develop a language model for renal transplant-pathology reports to extract the answers for two predefined questions.
Methods:
The study aimed to automatically answer two pre-defined questions given text from renal pathology reports: 1) “What kind of rejection does the patient show?”; and 2) “What is the grade of interstitial fibrosis and tubular atrophy (IFTA)?”. First, we followed the conventionally recommended procedure and pre-trained Clinical BERT further with the corpus which contains 3.4K renal transplant pathology reports and 1.5M words using Masked Language Modeling to obtain ‘Kidney BERT’. Second, we hypothesized that the conventional pre-training procedure fails to capture the intricate vocabulary of narrow technical domains. We created extended Kidney BERT (‘exKidneyBERT’) by extending the tokenizer of Clinical BERT with six technical keywords from our corpus (which we determined were missing from the original tokenizer vocabulary) and then repeating the pre-training procedure. Third, to further improve performance, all three models were fine-tuned with question-answering (QA) heads tailored to the two questions of interest.
Results:
For the first question regarding rejection, the overlap ratio at word level for exKidneyBERT – 83.3% for antibody-mediated rejection (ABMR) and 79.2% for T-cell mediated rejection (TCMR) – beats that of both Clinical BERT and Kidney BERT (both are 46.1% for ABMR, and 65.2% for TCMR). For the second question regarding IFTA, the exact match rate of exKidneyBERT (95.8%) beats that of Kidney BERT (95.0%) and Clinical BERT (94.7%),
Conclusions:
We developed exKidneyBERT, a high-performing model for automatically extracting information from renal pathology reports. More broadly, we found that when working in domains with highly specialized vocabulary, it is essential to extend the vocabulary library of the BERT tokenizer to improve model performance, otherwise, pre-training (especially on small corpuses) is ineffective. In our case, pre-training BERT language models for kidney pathology reports improved model performance even though the training data were relatively small.
Citation
Request queued. Please wait while the file is being generated. It may take some time.
Copyright
© The authors. All rights reserved. This is a privileged document currently under peer-review/community review (or an accepted/rejected manuscript). Authors have provided JMIR Publications with an exclusive license to publish this preprint on it's website for review and ahead-of-print citation purposes only. While the final peer-reviewed paper may be licensed under a cc-by license on publication, at this stage authors and publisher expressively prohibit redistribution of this draft paper other than for review purposes.