Deep Learning in Cardiothoracic Ratio Calculation and Cardiomegaly Detection
Jakub Kufel, Iga Paszkiewicz, Szymon Kocot, Anna Lis, Piotr Dudek, Łukasz Czogalik, Michał Janik, Katarzyna Bargieł-Łączek, Wiktoria Bartnikowska, Maciej Koźlik, Maciej Cebula, Katarzyna Gruszczyńska, Zbigniew Nawrat
Abstract
Objectives: The purpose of this study is to evaluate the performance of our deep learning algorithm in calculating cardiothoracic ratio (CTR) and thus in the assessment of cardiomegaly or pericardial effusion occurrences on chest radiography (CXR). Methods: From a database of 8000 CXRs, 13 folders with a comparable number of images were created. Then, 1020 images were chosen randomly, in proportion to the number of images in each folder. Afterward, CTR was calculated using RadiAnt Digital Imaging and Communications in Medicine (DICOM) Viewer software (2023.1). Next, heart and lung anatomical areas were marked in 3D Slicer. From these data, we trained an AI model which segmented heart and lung anatomy and determined the CTR value. Results: Our model achieved an Intersection over Union metric of 88.28% for the augmented training subset and 83.06% for the validation subset. F1-score for subsets were accordingly 90.22% and 90.67%. In the comparative analysis of artificial intelligence (AI) vs. humans, significantly lower transverse thoracic diameter (TTD) (p < 0.001), transverse cardiac diameter (TCD) (p < 0.001), and CTR (p < 0.001) values obtained using the neural network were observed. Conclusions: Results confirm that there is a significant correlation between the measurements made by human observers and the neural network. After validation in clinical conditions, our method may be used as a screening test or advisory tool when a specialist is not available, especially on Intensive Care Units (ICUs) or Emergency Departments (ERs) where time plays a key role.