arXiv reaDer
不確実性推定のためのマスクサンブル
Masksembles for Uncertainty Estimation
ディープニューラルネットワークはその能力を十分に実証していますが、予測の信頼性を推定することは依然として困難です。ディープアンサンブルは、不確実性の推定値を生成するための最良の方法の1つであると広く考えられていますが、トレーニングと評価には非常に費用がかかります。 MC-Dropoutは、もう1つの人気のある代替手段であり、安価ですが、信頼性も低くなります。私たちの中心的な直感は、MC-DropoutとDeepEnsemblesが極端な例であるアンサンブルのようなモデルの連続スペクトルがあるということです。 1つ目は、事実上無限の数の高度に相関するモデルを使用し、2つ目は有限数の独立したモデルに依存します。両方の利点を組み合わせるために、Masksemblesを紹介します。 MC-dropoutのようにネットワークの一部をランダムにドロップする代わりに、Masksembleは、個々のモデル間の相関を変更できるようにパラメーター化された、固定数のバイナリマスクに依存します。つまり、マスク間のオーバーラップとその密度を制御することにより、目前のタスクに最適な構成を選択できます。これにより、わずかなコストでEnsemblesと同等のパフォーマンスを備えたシンプルで実装が容易なメソッドが実現します。広く使用されている2つのデータセット、CIFAR10とImageNetでMasksemblesを実験的に検証します。
Deep neural networks have amply demonstrated their prowess but estimating the reliability of their predictions remains challenging. Deep Ensembles are widely considered as being one of the best methods for generating uncertainty estimates but are very expensive to train and evaluate. MC-Dropout is another popular alternative, which is less expensive, but also less reliable. Our central intuition is that there is a continuous spectrum of ensemble-like models of which MC-Dropout and Deep Ensembles are extreme examples. The first uses an effectively infinite number of highly correlated models while the second relies on a finite number of independent models. To combine the benefits of both, we introduce Masksembles. Instead of randomly dropping parts of the network as in MC-dropout, Masksemble relies on a fixed number of binary masks, which are parameterized in a way that allows to change correlations between individual models. Namely, by controlling the overlap between the masks and their density one can choose the optimal configuration for the task at hand. This leads to a simple and easy to implement method with performance on par with Ensembles at a fraction of the cost. We experimentally validate Masksembles on two widely used datasets, CIFAR10 and ImageNet.
updated: Tue Dec 15 2020 14:39:57 GMT+0000 (UTC)
published: Tue Dec 15 2020 14:39:57 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アソシエイト