一般化可能性は、特に病院の変更や画像処理ルーチンがモデルの完全な失敗につながる可能性がある医用画像処理の分野で、深層学習における主要な課題の1つと見なされています。これに取り組むために、一貫性トレーニングを導入します。これは、分散外の一般化を促進するために、拡張データと非拡張データ全体でモデルの予測一貫性を最大化することに基づくトレーニング手順とデータ拡張の代替手段です。この目的のために、セグメンテーション不整合損失(SIL)と呼ばれる新しい領域ベースのセグメンテーション損失関数を開発します。これは、拡張された予測と拡張されていない予測およびラベルのペアの違いを考慮します。一貫性トレーニングは、一般的な医療タスクであるポリープセグメンテーションのいくつかの分布外データセットで従来のデータ拡張よりも優れていることを示しています。
Generalizability is seen as one of the major challenges in deep learning, in particular in the domain of medical imaging, where a change of hospital or in imaging routines can lead to a complete failure of a model. To tackle this, we introduce Consistency Training, a training procedure and alternative to data augmentation based on maximizing models' prediction consistency across augmented and unaugmented data in order to facilitate better out-of-distribution generalization. To this end, we develop a novel region-based segmentation loss function called Segmentation Inconsistency Loss (SIL), which considers the differences between pairs of augmented and unaugmented predictions and labels. We demonstrate that Consistency Training outperforms conventional data augmentation on several out-of-distribution datasets on polyp segmentation, a popular medical task.