arXiv reaDer
継続的学習のためのジェネレーティブ フィーチャ駆動型画像リプレイ
Generative Feature-driven Image Replay for Continual Learning
ニューラル ネットワークは、さまざまなタスクで段階的にトレーニングを行うと、壊滅的な忘却を起こしがちです。一般的な増分学習法は、以前に見たサンプルのサブセットを保持し、後続のタスクのトレーニング中にそれらを再生することで、このような忘却を軽減します。ただし、データ保護規則などにより、これが常に可能であるとは限りません。このような制限されたシナリオでは、生成モデルを使用して、人工画像または非表示の特徴を分類子に再生できます。この作業では、Genifer (GENeratIve FEature-driven image Replay) を提案します。この手法では、分類器を通過したときに実際のサンプルと同じ隠れた特徴を誘発する必要がある画像を再生するように生成モデルをトレーニングします。したがって、私たちの手法には、画像と特徴の再生の両方の利点が組み込まれています。つまり、(1) 従来の画像の再生とは異なり、生成モデルは、分類に関連する特徴の分布を明示的に学習します。 (2) 機能のリプレイとは対照的に、分類子全体が引き続きトレーニング可能です。 (3) 生成モデルのトレーニング中のオーバーフィッティングを軽減しながら、蒸留パフォーマンスを向上させる画像空間増強を活用できます。 Genifer は、CIFAR-100 および CUB-200 データセットのさまざまな設定で、以前の最先端技術よりも大幅に優れていることを示しています。
Neural networks are prone to catastrophic forgetting when trained incrementally on different tasks. Popular incremental learning methods mitigate such forgetting by retaining a subset of previously seen samples and replaying them during the training on subsequent tasks. However, this is not always possible, e.g., due to data protection regulations. In such restricted scenarios, one can employ generative models to replay either artificial images or hidden features to a classifier. In this work, we propose Genifer (GENeratIve FEature-driven image Replay), where a generative model is trained to replay images that must induce the same hidden features as real samples when they are passed through the classifier. Our technique therefore incorporates the benefits of both image and feature replay, i.e.: (1) unlike conventional image replay, our generative model explicitly learns the distribution of features that are relevant for classification; (2) in contrast to feature replay, our entire classifier remains trainable; and (3) we can leverage image-space augmentations, which increase distillation performance while also mitigating overfitting during the training of the generative model. We show that Genifer substantially outperforms the previous state of the art for various settings on the CIFAR-100 and CUB-200 datasets.
updated: Tue Dec 06 2022 17:58:49 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アソシエイト