Multi-Branch CNN-LSTM Fusion Network-Driven System With BERT Semantic Evaluator for Radiology Reporting in Emergency Head CTs.

Journal: IEEE journal of translational engineering in health and medicine
PMID:

Abstract

The high volume of emergency room patients often necessitates head CT examinations to rule out ischemic, hemorrhagic, or other organic pathologies. A system that enhances the diagnostic efficacy of head CT imaging in emergency settings through structured reporting would significantly improve clinical decision making. Currently, no AI solutions address this need. Thus, our research aims to develop an automatic radiology reporting system by directly analyzing brain anomalies in head CT data. We propose a multi-branch CNN-LSTM fusion network-driven system for enhanced radiology reporting in emergency settings. We preprocessed head CT scans by resizing all slices, selecting those with significant variability, and applying PCA to retain 95% of the original data variance, ultimately saving the most representative five slices for each scan. We linked the reports to their respective slice IDs, divided them into individual captions, and preprocessed each. We performed an 80-20 split of the dataset for ten times, with 15% of the training set used for validation. Our model utilizes a pretrained VGG16, processing groups of five slices simultaneously, and features multiple end-to-end LSTM branches, each specialized in predicting one caption, subsequently combined to form the ordered reports after a BERT-based semantic evaluation. Our system demonstrates effectiveness and stability, with the postprocessing stage refining the syntax of the generated descriptions. However, there remains an opportunity to empower the evaluation framework to more accurately assess the clinical relevance of the automatically-written reports. Part of future work will include transitioning to 3D and developing an improved version based on vision-language models.

Authors

  • Selene Tomassini
    Department of Information Engineering and Computer ScienceUniversity of Trento Trento 38121 Italy.
  • Damiano Duranti
    Department of Information Engineering and Computer ScienceUniversity of Trento Trento 38121 Italy.
  • Abdallah Zeggada
    Department of Information Engineering and Computer ScienceUniversity of Trento Trento 38121 Italy.
  • Carlo Cosimo Quattrocchi
    Centre for Medical SciencesUniversity of Trento Trento 38121 Italy.
  • Farid Melgani
    Department of Information Engineering and Computer Science, University of Trento, Via Sommarive 9, 38123 Trento, Italy.
  • Paolo Giorgini
    Department of Information Engineering and Computer ScienceUniversity of Trento Trento 38121 Italy.