モデルの解釈可能性、信頼性、および検証には、医療画像の深層学習法の証拠として使用される空間要素の知識が重要です。回帰タスクのモデルには、このような手法がありません。疾患の重症度を特徴付ける回帰ターゲット値を含むデータセットに特に敵対者のトレーニングを策定するために、生成敵対ネットワーク(VR-GAN)による回帰の視覚化と呼ばれる方法を提案します。ジェネレーターは、元の画像に追加される疾患効果マップを作成することにより、リグレッサーの出力をシフトすることを学習しようとする条件付きの生成的敵対ネットワークを使用します。一方、リグレッサーは、修正された画像の元の回帰値を予測するように訓練されます。この技術で訓練されたモデルは、病気のさまざまな段階で画像がどのように見えるかを視覚化することを学びます。慢性閉塞性肺疾患(COPD)の診断に使用される、肺機能検査に関連する胸部X線のデータセットでメソッドを分析します。検証のために、異なる時点で同じ患者の2つの登録されたX線の差を計算し、生成された疾患効果マップに関連付けます。提案された方法は、分類に基づく手法よりも優れており、現実的に見える画像を提供し、放射線科医が通常この病気について観察するものに従って画像を修正します。実装コードはhttps://github.com/ricbl/vrganで入手できます。
Knowledge of what spatial elements of medical images deep learning methods use as evidence is important for model interpretability, trustiness, and validation. There is a lack of such techniques for models in regression tasks. We propose a method, called visualization for regression with a generative adversarial network (VR-GAN), for formulating adversarial training specifically for datasets containing regression target values characterizing disease severity. We use a conditional generative adversarial network where the generator attempts to learn to shift the output of a regressor through creating disease effect maps that are added to the original images. Meanwhile, the regressor is trained to predict the original regression value for the modified images. A model trained with this technique learns to provide visualization for how the image would appear at different stages of the disease. We analyze our method in a dataset of chest x-rays associated with pulmonary function tests, used for diagnosing chronic obstructive pulmonary disease (COPD). For validation, we compute the difference of two registered x-rays of the same patient at different time points and correlate it to the generated disease effect map. The proposed method outperforms a technique based on classification and provides realistic-looking images, making modifications to images following what radiologists usually observe for this disease. Implementation code is available at https://github.com/ricbl/vrgan.