大きな学習能力を備えた深層学習モデルは、多くの場合、医用画像データセットに適合しません。これは、医療データの取得とラベリングに多大な時間と費用がかかるため、トレーニングセットが比較的小さいことが多いためです。したがって、データ拡張は、トレーニングデータの可用性を拡張し、一般化を高めるためによく使用されます。ただし、拡張戦略は、正当化されることなくアドホックベースで選択されることがよくあります。本論文では、モデル分類性能の向上を目的とした拡張ポリシー検索方法を提示します。拡張ポリシー検索には、医療画像分析でよく使用される追加の変換を含め、そのパフォーマンスを評価します。さらに、拡張ポリシー検索を拡張して、非線形の混合例データ拡張戦略を含めます。これらの学習したポリシーを使用して、医用画像モデルトレーニングの原則的なデータ拡張が超音波標準平面検出の大幅な改善につながる可能性があり、超音波胎児標準平面分類のナイーブデータ拡張戦略よりも全体で平均F1スコアが7.0%向上することを示します。超音波画像の学習された表現は、最適化されたデータ拡張を使用して、より適切にクラスタリングおよび定義されていることがわかります。
Deep learning models with large learning capacities often overfit to medical imaging datasets. This is because training sets are often relatively small due to the significant time and financial costs incurred in medical data acquisition and labelling. Data augmentation is therefore often used to expand the availability of training data and to increase generalization. However, augmentation strategies are often chosen on an ad-hoc basis without justification. In this paper, we present an augmentation policy search method with the goal of improving model classification performance. We include in the augmentation policy search additional transformations that are often used in medical image analysis and evaluate their performance. In addition, we extend the augmentation policy search to include non-linear mixed-example data augmentation strategies. Using these learned policies, we show that principled data augmentation for medical image model training can lead to significant improvements in ultrasound standard plane detection, with an an average F1-score improvement of 7.0% overall over naive data augmentation strategies in ultrasound fetal standard plane classification. We find that the learned representations of ultrasound images are better clustered and defined with optimized data augmentation.