论文标题

学习决策合奏使用图神经网络合并胸部X光片筛查

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

论文作者

Chakravarty, Arunava, Sarkar, Tandra, Ghosh, Nirmalya, Sethuraman, Ramanathan, Sheet, Debdoot

论文摘要

胸部X光片主要用于筛查心脏,胸腔和肺部状况。正在开发基于机器学习的自动解决方案,以减轻放射科医生的常规筛查负担,从而使他们专注于关键案例。尽管最近的努力证明了深卷卷神经网络(CNN)的合奏,但他们不考虑疾病合并症,从而降低了其筛查性能。为了解决此问题,我们提出了一个基于图形神经网络(GNN)的解决方案,以获得集合预测,其中模拟了不同疾病之间的依赖性。对所提出的方法的全面评估通过在各种整体结构中提高标准结合技术来提高性能,从而证明了其潜力。使用Densenet121的GNN合奏实现了最佳性能,在13个疾病合并症中平均AUC为0.821。

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.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源