ディープラーニングは、さまざまなアプリケーションで人間レベルのパフォーマンスを示しています。ただし、現在の深層学習モデルは、新しいクラスを学習するときに古い知識を壊滅的に忘れることを特徴としています。これは、特に限られた数の疾患のトレーニングデータしか最初に利用できないインテリジェント診断システムで課題を提起します。この場合、新しい病気のデータでインテリジェントシステムを更新すると、必然的に、以前に学習した病気のパフォーマンスが低下します。人間の脳で新しい知識を学習するプロセスに触発されて、固定の事前トレーニングされた特徴抽出器に基づいて構築された継続的な学習のためのベイズ生成モデルを提案します。このモデルでは、各古いクラスの知識は、たとえばガウス混合モデルを使用して、統計分布のコレクションによってコンパクトに表すことができ、時間の経過とともに継続的な学習を忘れないようにすることができます。既存のクラス増分学習方法とは異なり、提案されたアプローチは継続的な学習プロセスに敏感ではなく、データ増分学習シナリオにさらにうまく適用できます。複数の医療および自然画像分類タスクに関する実験は、提案されたアプローチが、新しいクラスの継続的な学習中に古いクラスの一部の画像を保持する最先端のアプローチよりも優れていることを示しました。
Deep learning has shown its human-level performance in various applications. However, current deep learning models are characterised by catastrophic forgetting of old knowledge when learning new classes. This poses a challenge particularly in intelligent diagnosis systems where initially only training data of a limited number of diseases are available. In this case, updating the intelligent system with data of new diseases would inevitably downgrade its performance on previously learned diseases. Inspired by the process of learning new knowledge in human brains, we propose a Bayesian generative model for continual learning built on a fixed pre-trained feature extractor. In this model, knowledge of each old class can be compactly represented by a collection of statistical distributions, e.g. with Gaussian mixture models, and naturally kept from forgetting in continual learning over time. Unlike existing class-incremental learning methods, the proposed approach is not sensitive to the continual learning process and can be additionally well applied to the data-incremental learning scenario. Experiments on multiple medical and natural image classification tasks showed that the proposed approach outperforms state-of-the-art approaches which even keep some images of old classes during continual learning of new classes.