arXiv reaDer
敵対的反事実増強:アルツハイマー病分類への応用
Adversarial Counterfactual Augmentation: Application in Alzheimer's Disease Classification
利用可能な医療データが限られているため、医療画像分析のディープ ラーニング アプローチは、目に見えないデータに一般化する傾向があります。ランダムな変換を使用してトレーニング中にデータを拡張することが役立つことが示されており、ニューラル ネットワークをトレーニングするためのユビキタスな手法になりました。ここでは、事前にトレーニングされた生成モデルを考慮して、下流のタスクを改善するための最も効果的な合成画像を見つけることを目的とした、新しい敵対的反事実増強スキームを提案します。具体的には、ジェネレーターの入力条件因子と下流の分類器を勾配逆伝播法で交互に繰り返し更新する敵対的なゲームを構築します。これは、分類子の「弱点」を発見し、生成モデルを介して意図的にその弱点を克服するように強制することと見なすことができます。提案されたアプローチの有効性を実証するために、下流のタスクとしてアルツハイマー病 (AD) の分類を使用してメソッドを検証します。事前トレーニング済みの生成モデルは、年齢を条件因子として使用して脳画像を合成します。提案されたアプローチが分類パフォーマンスを改善し、偽の相関関係と壊滅的な忘却を軽減する可能性があることを示すために、広範な実験とアブレーション研究が行われました。コードは承認後にリリースされます。
Due to the limited availability of medical data, deep learning approaches for medical image analysis tend to generalise poorly to unseen data. Augmenting data during training with random transformations has been shown to help and became a ubiquitous technique for training neural networks. Here, we propose a novel adversarial counterfactual augmentation scheme that aims at finding the most effective synthesised images to improve downstream tasks, given 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. 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 a downstream task. The pre-trained generative model synthesises brain images using age as conditional factor. Extensive experiments and ablation studies have been performed to show that the proposed approach improves classification performance and has potential to alleviate spurious correlations and catastrophic forgetting. Code will be released upon acceptance.
updated: Sat Oct 01 2022 23:41:00 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アソシエイト