arXiv reaDer
実用的なエネルギーベースの回帰のための学習提案
Learning Proposals for Practical Energy-Based Regression
エネルギーベースモデル(EBM)は、確率的回帰の有望な代替手段としてなど、近年、機械学習内で復活を遂げています。ただし、エネルギーベースの回帰では、トレーニング用にプロポーザル分布を手動で設計する必要があり、テスト時に初期推定値を提供する必要があります。概念的に単純な方法を導入して、効果的な提案分布を自動的に学習することで、これらの問題の両方に対処します。この方法は、別のネットワークヘッドによってパラメーター化されます。この目的のために、私たちは驚くべき結果を導き出し、提案からEBMへのKLの相違、およびEBMの負の対数尤度を共同で最小化する統一されたトレーニング目標につながります。テスト時に、トレーニングされた提案で重要度サンプリングを使用して、学習したEBMを効率的に評価し、スタンドアロンの予測を生成できます。さらに、派生したトレーニング目標を利用して、共同でトレーニングされたエネルギーベースの教師と混合密度ネットワーク(MDN)を学習し、コンピュータービジョン内の4つの実世界の回帰タスクで従来のMDNトレーニングを一貫して上回っています。コードはhttps://github.com/fregu856/ebms_proposalsで入手できます。
Energy-based models (EBMs) have experienced a resurgence within machine learning in recent years, including as a promising alternative for probabilistic regression. However, energy-based regression requires a proposal distribution to be manually designed for training, and an initial estimate has to be provided at test-time. We address both of these issues by introducing a conceptually simple method to automatically learn an effective proposal distribution, which is parameterized by a separate network head. To this end, we derive a surprising result, leading to a unified training objective that jointly minimizes the KL divergence from the proposal to the EBM, and the negative log-likelihood of the EBM. At test-time, we can then employ importance sampling with the trained proposal to efficiently evaluate the learned EBM and produce stand-alone predictions. Furthermore, we utilize our derived training objective to learn mixture density networks (MDNs) with a jointly trained energy-based teacher, consistently outperforming conventional MDN training on four real-world regression tasks within computer vision. Code is available at https://github.com/fregu856/ebms_proposals.
updated: Fri Oct 22 2021 17:58:05 GMT+0000 (UTC)
published: Fri Oct 22 2021 17:58:05 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アソシエイト