Maintenance Notice

Due to necessary scheduled maintenance, the JMIR Publications website will be unavailable from Wednesday, July 01, 2020 at 8:00 PM to 10:00 PM EST. We apologize in advance for any inconvenience this may cause you.

Who will be affected?

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

  • Tiancheng Yang; 
  • Ilia Sucholutsky; 
  • Kuang-Yu Jen; 
  • Matthias Schonlau

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

Please cite as:

Yang T, Sucholutsky I, Jen KY, Schonlau M

exKidneyBERT: a Language Model for Kidney Transplant Pathology Reports and the Crucial Role of Extended Vocabularies: Algorithm Development and Validation

JMIR Preprints. 17/08/2022:41966

DOI: 10.2196/preprints.41966

URL: https://preprints.jmir.org/preprint/41966

Download PDF


Request queued. Please wait while the file is being generated. It may take some time.

© 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.