最近、畳み込みニューラルネットワーク(CNN)は、分類タスクで優れたパフォーマンスを達成しました。 CNNは「ブラックボックス」と見なされることが広く知られています。これは、予測メカニズムを理解し、誤った予測をデバッグするのが困難です。上記の欠点を解決するために、いくつかのモデルのデバッグと説明の作業が開発されています。ただし、これらの方法は、モデル予測の考えられる原因の説明と診断に重点を置いており、これに基づいて、研究者はモデルの次の最適化を手動で処理します。この論文では、モデルドクターと呼ばれる最初の完全自動モデル診断および治療ツールを提案します。 1)各カテゴリはスパースで特定の畳み込みカーネルとのみ相関し、2)敵対的なサンプルは分離され、通常のサンプルは特徴空間で連続しているという2つの発見に基づいて、CNN分類器を効果的に診断および最適化するための単純な集計勾配制約が考案されています。 。集約勾配戦略は、主流のCNN分類器のための用途の広いモジュールです。広範な実験により、提案されたModel Doctorが既存のすべてのCNN分類器に適用され、16の主流CNN分類器の精度が1%〜5%向上することが示されています。
Recently, Convolutional Neural Network (CNN) has achieved excellent performance in the classification task. It is widely known that CNN is deemed as a 'black-box', which is hard for understanding the prediction mechanism and debugging the wrong prediction. Some model debugging and explanation works are developed for solving the above drawbacks. However, those methods focus on explanation and diagnosing possible causes for model prediction, based on which the researchers handle the following optimization of models manually. In this paper, we propose the first completely automatic model diagnosing and treating tool, termed as Model Doctor. Based on two discoveries that 1) each category is only correlated with sparse and specific convolution kernels, and 2) adversarial samples are isolated while normal samples are successive in the feature space, a simple aggregate gradient constraint is devised for effectively diagnosing and optimizing CNN classifiers. The aggregate gradient strategy is a versatile module for mainstream CNN classifiers. Extensive experiments demonstrate that the proposed Model Doctor applies to all existing CNN classifiers, and improves the accuracy of 16 mainstream CNN classifiers by 1%-5%.