arXiv reaDer
Towards Semantic Interpretation of Thoracic Disease and COVID-19 Diagnosis Models
畳み込みニューラルネットワークは、胸部X線での胸部病変の自動診断に有望です。それらのブラックボックスの性質は、入力特徴帰属法(別名顕著性法)を介して予測を説明するための多くの最近の研究を引き起こしました。ただし、入力特徴の帰属方法は、予測のための入力領域の重要性を特定するだけであり、モデルの動作の意味解釈が欠けています。この作業では、最初にネットワークの内部ユニット(特徴マップ)に関連付けられたセマンティクスを識別します。以下の質問の調査に進みます。 COVID-19重症度スコアのみでトレーニングされた回帰モデルは、胸部病変に関連する視覚パターンを暗黙的に学習しますか?弱くラベル付けされたデータ(たとえば、健康、不健康)でトレーニングされたネットワークは、暗黙的に病状を学習しますか?さらに、学習した特徴の解釈可能性に対する事前トレーニングとデータの不均衡の影響を調査します。分析に加えて、各予測を意味的に説明するための意味帰属を提案します。公開されている胸部病変(CheXpert、NIH ChestX-ray8)およびCOVID-19データセット(BrixIA、およびCOVID-19胸部X線セグメンテーションデータセット)を使用して、調査結果を提示します。コードは公開されています。
Convolutional neural networks are showing promise in the automatic diagnosis of thoracic pathologies on chest x-rays. Their black-box nature has sparked many recent works to explain the prediction via input feature attribution methods (aka saliency methods). However, input feature attribution methods merely identify the importance of input regions for the prediction and lack semantic interpretation of model behavior. In this work, we first identify the semantics associated with internal units (feature maps) of the network. We proceed to investigate the following questions; Does a regression model that is only trained with COVID-19 severity scores implicitly learn visual patterns associated with thoracic pathologies? Does a network that is trained on weakly labeled data (e.g. healthy, unhealthy) implicitly learn pathologies? Moreover, we investigate the effect of pretraining and data imbalance on the interpretability of learned features. In addition to the analysis, we propose semantic attribution to semantically explain each prediction. We present our findings using publicly available chest pathologies (CheXpert, NIH ChestX-ray8) and COVID-19 datasets (BrixIA, and COVID-19 chest X-ray segmentation dataset). The Code is publicly available.
updated: Wed Aug 04 2021 15:30:28 GMT+0000 (UTC)
published: Sun Apr 04 2021 17:35:13 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト