arXiv reaDer
クラス識別プルーニングによるフェデレーション非学習
Federated Unlearning via Class-Discriminative Pruning
連合学習(FL)でトレーニングされたCNN分類モデルからカテゴリを選択的に忘却する問題を調査します。トレーニングに使用されるデータにFLでグローバルにアクセスできないことを考えると、私たちの洞察は、各チャネルの内部の影響を深く調査します。さまざまなチャネルによってアクティブ化された特徴マップの視覚化を通じて、さまざまなチャネルが画像分類のさまざまなカテゴリにさまざまな貢献をしていることがわかります。これに触発されて、特定のカテゴリに関する情報をモデルからスクラブする方法を提案します。この方法では、最初から再トレーニングしたり、トレーニングに使用したデータにグローバルにアクセスしたりする必要はありません。代わりに、チャネルのクラス識別を量子化するために、用語周波数逆ドキュメント周波数(TF-IDF)の概念を導入します。 TF-IDFスコアが高いチャネルは、ターゲットカテゴリに対する識別が多いため、学習を取り消すためにプルーニングする必要があります。チャネルプルーニングの後に、プルーニングされたモデルのパフォーマンスを回復するための微調整プロセスが続きます。 CIFAR10データセットで評価されたこの方法は、最初から再トレーニングする場合と比較して、精度を低下させることなく、ResNetモデルで8.9倍、VGGモデルで7.9倍の学習解除速度を加速します。 CIFAR100データセットの場合、スピードアップはそれぞれ9.9倍と8.4倍です。この作業は、法的および倫理的基準の遵守に向けたFLの補完的なブロックとして想定されています。
We explore the problem of selectively forgetting categories from trained CNN classification models in the federated learning (FL). Given that the data used for training cannot be accessed globally in FL, our insights probe deep into the internal influence of each channel. Through the visualization of feature maps activated by different channels, we observe that different channels have a varying contribution to different categories in image classification. Inspired by this, we propose a method for scrubbing the model clean of information about particular categories. The method does not require retraining from scratch, nor global access to the data used for training. Instead, we introduce the concept of Term Frequency Inverse Document Frequency (TF-IDF) to quantize the class discrimination of channels. Channels with high TF-IDF scores have more discrimination on the target categories and thus need to be pruned to unlearn. The channel pruning is followed by a fine-tuning process to recover the performance of the pruned model. Evaluated on CIFAR10 dataset, our method accelerates the speed of unlearning by 8.9x for the ResNet model, and 7.9x for the VGG model under no degradation in accuracy, compared to retraining from scratch. For CIFAR100 dataset, the speedups are 9.9x and 8.4x, respectively. We envision this work as a complementary block for FL towards compliance with legal and ethical criteria.
updated: Fri Oct 22 2021 14:01:42 GMT+0000 (UTC)
published: Fri Oct 22 2021 14:01:42 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アソシエイト