医療画像の分類またはセグメンテーションのための堅牢なディープラーニング(DL)システムのトレーニングは、さまざまな疾患の種類や重症度をカバーする画像が限られているため困難です。最も有益なサンプルを選択し、トレーニングデータに追加するためのアクティブラーニング(AL)フレームワークを提案します。条件付きの生成的敵対ネットワーク(cGAN)を使用して、実際の画像サンプルで生成を調整することにより、さまざまな疾患特性を持つ現実的な胸部X線画像を生成します。トレーニングセットに追加する有益なサンプルは、ベイジアンニューラルネットワークを使用して識別されます。実験では、提案されたALフレームワークが完全なデータセットの約35%を使用して最先端のパフォーマンスを達成できることを示しているため、従来の方法よりも大幅な時間と労力を節約できます。
Training robust deep learning (DL) systems for medical image classification or segmentation is challenging due to limited images covering different disease types and severity. We propose an active learning (AL) framework to select most informative samples and add to the training data. We use conditional generative adversarial networks (cGANs) to generate realistic chest xray images with different disease characteristics by conditioning its generation on a real image sample. Informative samples to add to the training set are identified using a Bayesian neural network. Experiments show our proposed AL framework is able to achieve state of the art performance by using about 35% of the full dataset, thus saving significant time and effort over conventional methods.