arXiv reaDer
硬さ加重サンプリングを使用した分布的にロバストな深層学習
Distributionally Robust Deep Learning using Hardness Weighted Sampling
機械学習システムの障害を制限することは、セーフティクリティカルなアプリケーションにとって最も重要です。機械学習システムの堅牢性を向上させるために、経験的リスク最小化(ERM)の一般化として分布ロバスト最適化(DRO)が提案されています。ただし、深層学習での使用は、ERMの確率的勾配降下(SGD)オプティマイザーの広範なバリアントと比較して、DROで使用可能なオプティマイザーが比較的非効率であるため、厳しく制限されています。硬さ加重サンプリングを使用したSGDを提案します。これは、深層学習のコンテキストで特に適した、機械学習におけるDROの原理的で効率的な最適化手法です。実際のハードな例のマイニング戦略と同様に、提案されたアルゴリズムは、実装が簡単で、ディープラーニングに使用されるSGDベースのオプティマイザーと同じくらい計算効率が高く、最小限のオーバーヘッド計算で済みます。典型的なアドホックハードマイニングアプローチとは対照的に、ReLUアクティベーションと有限数のレイヤーとパラメーターを使用して、パラメーターが過剰な深層学習ネットワークに対するDROアルゴリズムの収束を証明します。 MRIにおける胎児の脳の3DMRIセグメンテーションと脳腫瘍のセグメンテーションに関する私たちの実験は、私たちのアプローチの実現可能性と有用性を示しています。最先端の深層学習パイプラインをトレーニングするために硬度加重サンプリングを使用すると、深層学習を使用した自動胎児脳3D MRIセグメンテーションの解剖学的変動に対するロバスト性が向上し、脳腫瘍セグメンテーションの画像プロトコル変動に対するロバスト性が向上します。私たちのコードはhttps://github.com/LucasFidon/HardnessWeightedSamplerで入手できます。
Limiting failures of machine learning systems is of paramount importance for safety-critical applications. In order to improve the robustness of machine learning systems, Distributionally Robust Optimization (DRO) has been proposed as a generalization of Empirical Risk Minimization (ERM). However, its use in deep learning has been severely restricted due to the relative inefficiency of the optimizers available for DRO in comparison to the wide-spread variants of Stochastic Gradient Descent (SGD) optimizers for ERM. We propose SGD with hardness weighted sampling, a principled and efficient optimization method for DRO in machine learning that is particularly suited in the context of deep learning. Similar to a hard example mining strategy in practice, the proposed algorithm is straightforward to implement and computationally as efficient as SGD-based optimizers used for deep learning, requiring minimal overhead computation. In contrast to typical ad hoc hard mining approaches, we prove the convergence of our DRO algorithm for over-parameterized deep learning networks with ReLU activation and a finite number of layers and parameters. Our experiments on fetal brain 3D MRI segmentation and brain tumor segmentation in MRI demonstrate the feasibility and the usefulness of our approach. Using our hardness weighted sampling for training a state-of-the-art deep learning pipeline leads to improved robustness to anatomical variabilities in automatic fetal brain 3D MRI segmentation using deep learning and to improved robustness to the image protocol variations in brain tumor segmentation. Our code is available at https://github.com/LucasFidon/HardnessWeightedSampler.
updated: Wed Feb 09 2022 14:53:52 GMT+0000 (UTC)
published: Wed Jan 08 2020 18:02:56 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アソシエイト