arXiv reaDer
軽量の分布外検出のためのVicinalRiskMinimizationの調査
Exploring Vicinal Risk Minimization for Lightweight Out-of-Distribution Detection
ディープニューラルネットワークは、画像認識から自然言語処理に至るまでの複雑なタスクの解決に広く採用されています。ただし、これらのネットワークは、トレーニング分布に属さないデータ、つまり分布外(OoD)サンプルが提示されると、自信を持って誤予測を起こします。このホワイトペーパーでは、異なるクラス境界間をスムーズに補間するVicinal Risk Minimization(VRM)のプロパティが、より優れたOoD検出器のトレーニングに役立つかどうかを調査します。 VRMを既存のOoD検出技術に適用し、パフォーマンスの向上を示します。既存のOoD検出器にはかなりのメモリと計算オーバーヘッドがあるため、VRMを活用して、耳に聞こえないOoD検出器を開発します。私たちの検出方法は、OoDサンプルを分類するための補助クラスを導入しています。 Vicinal Risk Minimizationを実装するには、2つの方法で混合を利用します。まず、同じクラス内でミックスアップを実行し、次に、補助クラスをトレーニングするときにガウスノイズとのミックスアップを実行します。私たちの方法は、既存のOoD検出技術と比較して、計算とメモリのオーバーヘッドを大幅に削減して、ほぼ競争力のあるパフォーマンスを実現します。これにより、エッジデバイスでのOoD検出の展開が容易になり、OoD検出器のトレーニングで使用するためのVicinal RiskMinimizationの理解が深まります。
Deep neural networks have found widespread adoption in solving complex tasks ranging from image recognition to natural language processing. However, these networks make confident mispredictions when presented with data that does not belong to the training distribution, i.e. out-of-distribution (OoD) samples. In this paper we explore whether the property of Vicinal Risk Minimization (VRM) to smoothly interpolate between different class boundaries helps to train better OoD detectors. We apply VRM to existing OoD detection techniques and show their improved performance. We observe that existing OoD detectors have significant memory and compute overhead, hence we leverage VRM to develop an OoD detector with minimal overheard. Our detection method introduces an auxiliary class for classifying OoD samples. We utilize mixup in two ways to implement Vicinal Risk Minimization. First, we perform mixup within the same class and second, we perform mixup with Gaussian noise when training the auxiliary class. Our method achieves near competitive performance with significantly less compute and memory overhead when compared to existing OoD detection techniques. This facilitates the deployment of OoD detection on edge devices and expands our understanding of Vicinal Risk Minimization for use in training OoD detectors.
updated: Tue Dec 15 2020 16:13:33 GMT+0000 (UTC)
published: Tue Dec 15 2020 16:13:33 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アソシエイト