Path-BigBird: An AI-Driven Transformer Approach to Classification of Cancer Pathology Reports.

Abstract

Surgical pathology reports are critical for cancer diagnosis and management. To accurately extract information about tumor characteristics from pathology reports in near real time, we explore the impact of using domain-specific transformer models that understand cancer pathology reports. We built a pathology transformer model, Path-BigBird, by using 2.7 million pathology reports from six SEER cancer registries. We then compare different variations of Path-BigBird with two less computationally intensive methods: Hierarchical Self-Attention Network (HiSAN) classification model and an off-the-shelf clinical transformer model (Clinical BigBird). We use five pathology information extraction tasks for evaluation: site, subsite, laterality, histology, and behavior. Model performance is evaluated by using macro and micro <i>F</i>₁ scores. We found that Path-BigBird and Clinical BigBird outperformed the HiSAN in all tasks. Clinical BigBird performed better on the <i>site</i> and <i>laterality</i> tasks. Versions of the Path-BigBird model performed best on the two most difficult tasks: <i>subsite</i> (micro <i>F</i>₁ score of 72.53, macro <i>F</i>₁ score of 35.76) and <i>histology</i> (micro <i>F</i>₁ score of 80.96, macro <i>F</i>₁ score of 37.94). The largest performance gains over the HiSAN model were for <i>histology</i>, for which a Path-BigBird model increased the micro <i>F</i>₁ score by 1.44 points and the macro <i>F</i>₁ score by 3.55 points. Overall, the results suggest that a Path-BigBird model with a vocabulary derived from well-curated and deidentified data is the best-performing model. The Path-BigBird pathology transformer model improves automated information extraction from pathology reports. Although Path-BigBird outperforms Clinical BigBird and HiSAN, these less computationally expensive models still have utility when resources are constrained.