ディープラーニングの出現により、複雑な識別モデルによる画像分類の改善が可能になりました。ただし、複雑さを増したこのような深いモデルでは、トレーニングを一般化するためにラベル付きサンプルの膨大なセットが必要です。このような分類モデルは、トレーニングデータが限られているため、医療画像に適用すると簡単にオーバーフィットする可能性があります。これは、医療画像分析の分野で一般的な問題です。この論文は、いくつかの利用可能なトレーニングデータの下で一般化を改善するための強化された分類器を提案し、調査します。強化学習の概念に部分的に従って、提案された分類器は、トレーニングデータに関する従来のクロスエントロピー損失のみを使用する代わりに、トレーニングデータのサブセットからの汎化フィードバックを使用してそのパラメーターを更新します。提案する分類器の改善を、既存のオーバーフィッティング防止技術を備えた標準の深い分類器に対して3つの異なる分類問題に適用することにより評価します。分類性能の全体的な改善に加えて、提案された分類器は一般化学習の顕著な特性を示し、これは医療分類タスクに大きな可能性を秘めています。
With the advents of deep learning, improved image classification with complex discriminative models has been made possible. However, such deep models with increased complexity require a huge set of labeled samples to generalize the training. Such classification models can easily overfit when applied for medical images because of limited training data, which is a common problem in the field of medical image analysis. This paper proposes and investigates a reinforced classifier for improving the generalization under a few available training data. Partially following the idea of reinforcement learning, the proposed classifier uses a generalization-feedback from a subset of the training data to update its parameter instead of only using the conventional cross-entropy loss about the training data. We evaluate the improvement of the proposed classifier by applying it on three different classification problems against the standard deep classifiers equipped with existing overfitting-prevention techniques. Besides an overall improvement in classification performance, the proposed classifier showed remarkable characteristics of generalized learning, which can have great potential in medical classification tasks.