arXiv reaDer
深いアンサンブルを使用した貪欲なベイジアン事後近似
Greedy Bayesian Posterior Approximation with Deep Ensembles
独立してトレーニングされたニューラル ネットワークのアンサンブルは、深層学習における予測の不確実性を推定するための最先端のアプローチであり、デルタ関数の混合による事後分布の近似として解釈できます。アンサンブルのトレーニングは、損失状況の非凸性と個々のメンバーのランダムな初期化に依存しているため、結果として得られる事後近似が制御されなくなります。この論文では、この制限に対処するための新しい原理に基づいた方法を提案し、関数空間における真の事後推定量とカーネル密度推定量の間の f-乖離を最小限に抑えます。この目的を組み合わせの観点から分析し、任意の f の混合成分に関して劣モジュラであることを示します。次に、貪欲なアンサンブル構成の問題を考慮し、全体の目的の限界利得から、アンサンブル法の新しい多様性項を導出します。私たちのアプローチのパフォーマンスは、複数のデータセットでトレーニングされたさまざまなアーキテクチャのコンピューター ビジョン アウトオブディストリビューション ベンチマークで実証されています。このメソッドのソース コードは、https://github.com/MIPT-Oulu/greedy_ensembles_training で公開されています。
Ensembles of independently trained neural networks are a state-of-the-art approach to estimate predictive uncertainty in Deep Learning, and can be interpreted as an approximation of the posterior distribution via a mixture of delta functions. The training of ensembles relies on non-convexity of the loss landscape and random initialization of their individual members, making the resulting posterior approximation uncontrolled. This paper proposes a novel and principled method to tackle this limitation, minimizing an f-divergence between the true posterior and a kernel density estimator in a function space. We analyze this objective from a combinatorial point of view, and show that it is submodular with respect to mixture components for any f. Subsequently, we consider the problem of greedy ensemble construction, and from the marginal gain of the total objective, we derive a novel diversity term for ensemble methods. The performance of our approach is demonstrated on computer vision out-of-distribution benchmarks in a range of architectures trained on multiple datasets. The source code of our method is publicly available at https://github.com/MIPT-Oulu/greedy_ensembles_training.
updated: Sat May 29 2021 11:35:27 GMT+0000 (UTC)
published: Sat May 29 2021 11:35:27 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アソシエイト