Alert button

Learning Decision Ensemble using a Graph Neural Network for Comorbidity Aware Chest Radiograph Screening

Apr 24, 2020
Arunava Chakravarty, Tandra Sarkar, Nirmalya Ghosh, Ramanathan Sethuraman, Debdoot Sheet

Figure 1 for Learning Decision Ensemble using a Graph Neural Network for Comorbidity Aware Chest Radiograph Screening
Figure 2 for Learning Decision Ensemble using a Graph Neural Network for Comorbidity Aware Chest Radiograph Screening
Figure 3 for Learning Decision Ensemble using a Graph Neural Network for Comorbidity Aware Chest Radiograph Screening
Figure 4 for Learning Decision Ensemble using a Graph Neural Network for Comorbidity Aware Chest Radiograph Screening

Share this with someone who'll enjoy it:

Chest radiographs are primarily employed for the screening of cardio, thoracic and pulmonary conditions. Machine learning based automated solutions are being developed to reduce the burden of routine screening on Radiologists, allowing them to focus on critical cases. While recent efforts demonstrate the use of ensemble of deep convolutional neural networks(CNN), they do not take disease comorbidity into consideration, thus lowering their screening performance. To address this issue, we propose a Graph Neural Network (GNN) based solution to obtain ensemble predictions which models the dependencies between different diseases. A comprehensive evaluation of the proposed method demonstrated its potential by improving the performance over standard ensembling technique across a wide range of ensemble constructions. The best performance was achieved using the GNN ensemble of DenseNet121 with an average AUC of 0.821 across thirteen disease comorbidities.

* accepted in EMBC 2020, 4pg+2pg Supplementary Material  
View paper onarxiv icon

Share this with someone who'll enjoy it: