arXiv reaDer
転送不可能な事前トレーニング済みモデルによる正規化を使用した限られたデータの深い生成モデリング
Deep Generative Modeling on Limited Data with Regularization by Nontransferable Pre-trained Models
深層生成モデル (DGM) はデータを大量に消費します。基本的に、限られたデータで複雑なモデルを学習すると、分散が大きくなり、過適合になりやすいためです。バイアスと分散のジレンマに着想を得て、正規化された深い生成モデル (Reg-DGM) を提案します。これは、転送不可能な事前トレーニング済みモデルを活用して、限られたデータで生成モデリングの分散を減らします。正式には、Reg-DGM は、データ分布と DGM の間の特定の発散と、DGM に関する事前トレーニング済みモデルによって定義されたエネルギー関数の期待値の加重和を最適化します。理論的には、ノンパラメトリック設定でのReg-DGMのグローバル最小値の存在と一意性を特徴付け、平均二乗誤差と予想されるリスクに関するReg-DGMの統計的利点を、単純でありながら代表的なガウスフィッティングの例で厳密に証明します。経験的に、Reg-DGM で DGM と事前トレーニング済みモデルを指定することは非常に柔軟です。特に、ImageNet で事前トレーニングされた ResNet-18 分類器とデータ依存のエネルギー関数を使用して、Reg-DGM は StyleGAN2 と ADA を含む強力な DGM の生成パフォーマンスを、限られたデータを使用していくつかのベンチマークで一貫して改善し、州と競争力のある結果を達成します。 -最先端の方法。
Deep generative models (DGMs) are data-eager. Essentially, it is because learning a complex model on limited data suffers from a large variance and easily overfits. Inspired by the bias-variance dilemma, we propose regularized deep generative model (Reg-DGM), which leverages a nontransferable pre-trained model to reduce the variance of generative modeling with limited data. Formally, Reg-DGM optimizes a weighted sum of a certain divergence between the data distribution and the DGM and the expectation of an energy function defined by the pre-trained model w.r.t. the DGM. Theoretically, we characterize the existence and uniqueness of the global minimum of Reg-DGM in the nonparametric setting and rigorously prove the statistical benefits of Reg-DGM w.r.t. the mean squared error and the expected risk in a simple yet representative Gaussian-fitting example. Empirically, it is quite flexible to specify the DGM and the pre-trained model in Reg-DGM. In particular, with a ResNet-18 classifier pre-trained on ImageNet and a data-dependent energy function, Reg-DGM consistently improves the generation performance of strong DGMs including StyleGAN2 and ADA on several benchmarks with limited data and achieves competitive results to the state-of-the-art methods.
updated: Tue Aug 30 2022 10:28:50 GMT+0000 (UTC)
published: Tue Aug 30 2022 10:28:50 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アソシエイト