arXiv reaDer
敵対的反事実的増強:アルツハイマー病分類への応用
Adversarial Counterfactual Augmentation: Application in Alzheimer's Disease Classification
データ拡張は、過剰適合を減らし、モデルの堅牢性を向上させるために、深層学習で広く使用されています。ただし、回転、トリミング、反転などの従来のデータ拡張手法では、脳画像の年齢の変更などの意味変換は考慮されていません。以前の作品は、反事実を生成することによって意味的増強を達成しようとしましたが、どの反事実が下流のトレーニングを改善するのに最も効果的であるかを考慮せずに、生成モデルを使用して深い生成モデルとランダムに作成された反事実をトレーニングする方法に焦点を当てました。これらのアプローチとは異なり、この作業では、事前にトレーニングされた生成モデルを使用してダウンストリームタスクを改善するための最も効果的な反事実を見つけることを目的とした、新しい敵対的反事実拡張スキームを提案します。具体的には、ジェネレーターとダウンストリーム分類器の入力条件付き因子を勾配バックプロパゲーションで交互に繰り返し更新する敵対的なゲームを構築します。重要なアイデアは、分類器に厳しい反事実をもたらす可能性のある条件付き要因を見つけることです。これは、分類器の「弱点」を見つけ、生成モデルを介してその弱点を克服するように意図的に強制するものと見なすことができます。提案されたアプローチの有効性を実証するために、事前に訓練された脳老化合成モデルに基づいて、アルツハイマー病(AD)を下流のタスクとして分類する方法を検証します。提案されたアプローチがテスト精度を改善し、疑似相関を軽減できることを示します。コードは承認時にリリースされます。
Data augmentation has been widely used in deep learning to reduce over-fitting and improve the robustness of models. However, traditional data augmentation techniques, e.g., rotation, cropping, flipping, etc., do not consider semantic transformations, e.g., changing the age of a brain image. Previous works tried to achieve semantic augmentation by generating counterfactuals, but they focused on how to train deep generative models and randomly created counterfactuals with the generative models without considering which counterfactuals are most effective for improving downstream training. Different from these approaches, in this work, we propose a novel adversarial counterfactual augmentation scheme that aims to find the most effective counterfactuals to improve downstream tasks with a pre-trained generative model. Specifically, we construct an adversarial game where we update the input conditional factor of the generator and the downstream classifier with gradient backpropagation alternatively and iteratively. The key idea is to find conditional factors that can result in hard counterfactuals for the classifier. This can be viewed as finding the `weakness' of the classifier and purposely forcing it to overcome its weakness via the generative model. To demonstrate the effectiveness of the proposed approach, we validate the method with the classification of Alzheimer's Disease (AD) as the downstream task based on a pre-trained brain ageing synthesis model. We show the proposed approach improves test accuracy and can alleviate spurious correlations. Code will be released upon acceptance.
updated: Tue Mar 15 2022 12:11:05 GMT+0000 (UTC)
published: Tue Mar 15 2022 12:11:05 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アソシエイト