医療画像セグメンテーションタスクでのニューラルネットワークの成功は、通常、モデルトレーニング用の大きなラベル付きデータセットに依存しています。ただし、大規模な医用画像セットを取得して手動でラベル付けすることは、リソースを大量に消費し、費用がかかり、データ共有とプライバシーの問題のために非現実的である場合があります。この課題に対処するために、医療画像セグメンテーションタスクのトレーニングデータの多様性と有効性の両方を改善することを目的とした、一般的な敵対的データ拡張フレームワークであるAdvChainを提案します。 AdvChainは、動的データ拡張を使用してデータを拡張し、ランダムにチェーン化されたフォトメトリックおよび幾何学的変換を生成して、トレーニングデータを拡張するための現実的でありながら挑戦的なイメージングバリエーションに似ています。トレーニング中にデータ拡張モデルとセグメンテーションネットワークを共同で最適化することにより、ダウンストリームタスクのネットワークの一般化可能性を強化するための挑戦的な例が生成されます。提案された敵対的データ拡張は、生成ネットワークに依存せず、一般的なセグメンテーションネットワークのプラグインモジュールとして使用できます。これは計算効率が高く、ローショット教師あり学習と半教師あり学習の両方に適用できます。心臓セグメンテーションと限られたラベル付きデータを使用した前立腺セグメンテーションの2つのMR画像セグメンテーションタスクでこの方法を分析および評価します。結果は、提案されたアプローチがモデルの一般化能力を向上させながらラベル付けされたデータの必要性を軽減できることを示しており、医用画像アプリケーションにおけるその実用的な価値を示しています。
The success of neural networks on medical image segmentation tasks typically relies on large labeled datasets for model training. However, acquiring and manually labeling a large medical image set is resource-intensive, expensive, and sometimes impractical due to data sharing and privacy issues. To address this challenge, we propose AdvChain, a generic adversarial data augmentation framework, aiming at improving both the diversity and effectiveness of training data for medical image segmentation tasks. AdvChain augments data with dynamic data augmentation, generating randomly chained photo-metric and geometric transformations to resemble realistic yet challenging imaging variations to expand training data. By jointly optimizing the data augmentation model and a segmentation network during training, challenging examples are generated to enhance network generalizability for the downstream task. The proposed adversarial data augmentation does not rely on generative networks and can be used as a plug-in module in general segmentation networks. It is computationally efficient and applicable for both low-shot supervised and semi-supervised learning. We analyze and evaluate the method on two MR image segmentation tasks: cardiac segmentation and prostate segmentation with limited labeled data. Results show that the proposed approach can alleviate the need for labeled data while improving model generalization ability, indicating its practical value in medical imaging applications.