クラス増分学習(CIL)問題を検討します。この問題では、学習エージェントは、増分的に到着するトレーニングデータバッチから新しいクラスを継続的に学習し、これまでに学習したすべてのクラスを適切に予測することを目的としています。問題の主な課題は壊滅的な忘却であり、模範記憶ベースのCILメソッドの場合、忘却は一般に、新しいクラスと古いクラス(エグザンプラメモリ内)。スコアの再スケーリングやバランスの取れた微調整など、追加の後処理によってそのようなスコアの偏りを修正するためのいくつかの方法が提案されていますが、そのような偏りの根本原因に関する体系的な分析は行われていません。そのために、すべての新旧のクラスの出力スコアを組み合わせてソフトマックス確率を計算することがバイアスの主な原因である可能性があることを分析し、新しいCILメソッドであるインクリメンタル学習用の分離ソフトマックス(SS-IL)を提案します。私たちのSS-ILは、分離されたソフトマックス(SS)出力層と比率保存(RP)ミニバッチをタスクワイズ知識蒸留(TKD)と組み合わせて構成されており、広範な実験結果を通じて、SS-ILが非常に強力な状態を達成していることを示しています-いくつかの大規模ベンチマークでの最先端の精度。また、SS-ILは、他のベースラインで行われるような追加の後処理ステップなしで、はるかにバランスの取れた予測を行うことも示しています。
We consider class incremental learning (CIL) problem, in which a learning agent continuously learns new classes from incrementally arriving training data batches and aims to predict well on all the classes learned so far. The main challenge of the problem is the catastrophic forgetting, and for the exemplar-memory based CIL methods, it is generally known that the forgetting is commonly caused by the prediction score bias that is injected due to the data imbalance between the new classes and the old classes (in the exemplar-memory). While several methods have been proposed to correct such score bias by some additional post-processing, e.g., score re-scaling or balanced fine-tuning, no systematic analysis on the root cause of such bias has been done. To that end, we analyze that computing the softmax probabilities by combining the output scores for all old and new classes could be the main source of the bias and propose a new CIL method, Separated Softmax for Incremental Learning (SS-IL). Our SS-IL consists of separated softmax (SS) output layer and ratio-preserving (RP) mini-batches combined with task-wise knowledge distillation (TKD), and through extensive experimental results, we show our SS-IL achieves very strong state-of-the-art accuracy on several large-scale benchmarks. We also show SS-IL makes much more balanced prediction, without any additional post-processing steps as is done in other baselines.