arXiv reaDer
重要なものと一致する:継続的な学習のための生成的な暗黙の機能の再生
Match What Matters: Generative Implicit Feature Replay for Continual Learning
ニューラルネットワークは、さまざまなタスクについて段階的にトレーニングすると、壊滅的な忘却を起こしがちです。忘却を防ぐために、既存の方法のほとんどは、以前に見たサンプルの小さなサブセットを保持し、新しいタスクとの共同トレーニングに使用できます。これは確かに効果的ですが、たとえばデータ保護規制のために、そのようなサンプルを保存できるとは限りません。このような場合、代わりに生成モデルを使用して、前のタスクからの記憶を表す人工的なサンプルまたは特徴を作成できます。同様の方向性に従って、クラス増分学習用のGenIFeR(Generative Implicit Feature Replay)を提案します。主なアイデアは、現実的な特徴を含む画像を生成するために、生成的敵対的ネットワーク(GAN)をトレーニングすることです。ジェネレータはフル解像度で画像を作成しますが、弁別器は継続的にトレーニングされた分類器によって抽出された対応する特徴のみを確認します。分類器は生の画像を実際に分類に関連する特徴に圧縮するため、GANはこのターゲット分布をより正確に一致させることができます。一方、ジェネレーターがフル解像度の画像を作成できるようにすることには、いくつかの利点があります。以前のアプローチとは対照的に、分類器の特徴抽出器をフリーズする必要はありません。さらに、生成された画像に拡張を採用することで、分類のパフォーマンスを向上させるだけでなく、GANトレーニング中の弁別器の過剰適合を軽減することもできます。 GenIFeRは、従来の生成画像と機能再生の両方よりも優れていることを経験的に示しています。特に、CIFAR-100およびCUB-200データセットのさまざまな設定で、最新の生成再生を大幅に上回っています。
Neural networks are prone to catastrophic forgetting when trained incrementally on different tasks. In order to prevent forgetting, most existing methods retain a small subset of previously seen samples, which in turn can be used for joint training with new tasks. While this is indeed effective, it may not always be possible to store such samples, e.g., due to data protection regulations. In these cases, one can instead employ generative models to create artificial samples or features representing memories from previous tasks. Following a similar direction, we propose GenIFeR (Generative Implicit Feature Replay) for class-incremental learning. The main idea is to train a generative adversarial network (GAN) to generate images that contain realistic features. While the generator creates images at full resolution, the discriminator only sees the corresponding features extracted by the continually trained classifier. Since the classifier compresses raw images into features that are actually relevant for classification, the GAN can match this target distribution more accurately. On the other hand, allowing the generator to create full resolution images has several benefits: In contrast to previous approaches, the feature extractor of the classifier does not have to be frozen. In addition, we can employ augmentations on generated images, which not only boosts classification performance, but also mitigates discriminator overfitting during GAN training. We empirically show that GenIFeR is superior to both conventional generative image and feature replay. In particular, we significantly outperform the state-of-the-art in generative replay for various settings on the CIFAR-100 and CUB-200 datasets.
updated: Wed Jun 09 2021 19:29:41 GMT+0000 (UTC)
published: Wed Jun 09 2021 19:29:41 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アソシエイト