arXiv reaDer
有益な入力特徴を特定することによるCOVID-19および胸部病理学モデルの予測の説明
Explaining COVID-19 and Thoracic Pathology Model Predictions by Identifying Informative Input Features
ニューラルネットワークは、胸部X線の分類および回帰タスクで顕著なパフォーマンスを示しています。臨床ルーチンへの信頼を確立するために、ネットワークの予測メカニズムは解釈可能である必要があります。解釈への主要なアプローチの1つは、機能の帰属です。特徴帰属法は、出力予測のための入力特徴の重要性を識別します。 Information Bottleneck Attribution(IBA)メソッドに基づいて、予測ごとに、ネットワークの出力との相互情報量が高い胸部X線領域を特定します。元のIBAは、十分な予測情報を持つ入力領域を識別します。すべての有益な地域を特定するために、逆IBAを提案します。したがって、病状のすべての予測手がかりは、胸部X線診断に望ましい特性であるX線で強調表示されます。さらに、回帰モデルを説明するための回帰IBAを提案します。回帰IBAを使用して、累積重大度スコアラベルでトレーニングされたモデルが、さまざまなX線領域の重大度を暗黙的に学習することを確認します。最後に、より高解像度でより詳細な帰属/顕著性マップを生成するためのマルチレイヤーIBAを提案します。人間中心の(グラウンドトゥルースベースの)解釈可能性メトリックと、NIH胸部X線8およびBrixIAデータセットの人間に依存しない特徴重要度メトリックの両方を使用してメソッドを評価します。コードは公開されています。
Neural networks have demonstrated remarkable performance in classification and regression tasks on chest X-rays. In order to establish trust in the clinical routine, the networks' prediction mechanism needs to be interpretable. One principal approach to interpretation is feature attribution. Feature attribution methods identify the importance of input features for the output prediction. Building on Information Bottleneck Attribution (IBA) method, for each prediction we identify the chest X-ray regions that have high mutual information with the network's output. Original IBA identifies input regions that have sufficient predictive information. We propose Inverse IBA to identify all informative regions. Thus all predictive cues for pathologies are highlighted on the X-rays, a desirable property for chest X-ray diagnosis. Moreover, we propose Regression IBA for explaining regression models. Using Regression IBA we observe that a model trained on cumulative severity score labels implicitly learns the severity of different X-ray regions. Finally, we propose Multi-layer IBA to generate higher resolution and more detailed attribution/saliency maps. We evaluate our methods using both human-centric (ground-truth-based) interpretability metrics, and human-independent feature importance metrics on NIH Chest X-ray8 and BrixIA datasets. The Code is publicly available.
updated: Wed Aug 04 2021 15:21:55 GMT+0000 (UTC)
published: Thu Apr 01 2021 11:42:39 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト