データ拡張は、ディープニューラルネットワークの一般化パフォーマンスを改善するための効果的で普遍的な手法です。 1)医用画像データセットの規模は通常小さいため、過剰適合のリスクが高まる可能性があるため、医用画像セグメンテーションタスクに不可欠なトレーニングサンプルの多様性を高めることができます。 2)臓器や腫瘍などのさまざまなオブジェクトの形状とモダリティは固有であるため、カスタマイズされたデータ拡張ポリシーが必要です。ただし、ほとんどのデータ拡張の実装は手作りであり、医用画像処理では最適ではありません。データ拡張の可能性を十分に活用するために、最適な拡張戦略を自動的に検索する効率的なアルゴリズムを提案します。ネットワークの重みと拡張パラメータを組み合わせた最適化を、確率的緩和によって微分可能な形式に定式化します。この定式化により、代替の勾配ベースの方法、つまり適応ステップサイズを使用した確率的自然勾配法を適用できます。私たちの知る限りでは、微分可能な自動データ拡張が医療画像セグメンテーションタスクで採用されるのは初めてです。私たちの数値実験は、提案されたアプローチが最先端モデルの既存の組み込みデータ拡張を大幅に上回っていることを示しています。
Data augmentation is an effective and universal technique for improving generalization performance of deep neural networks. It could enrich diversity of training samples that is essential in medical image segmentation tasks because 1) the scale of medical image dataset is typically smaller, which may increase the risk of overfitting; 2) the shape and modality of different objects such as organs or tumors are unique, thus requiring customized data augmentation policy. However, most data augmentation implementations are hand-crafted and suboptimal in medical image processing. To fully exploit the potential of data augmentation, we propose an efficient algorithm to automatically search for the optimal augmentation strategies. We formulate the coupled optimization w.r.t. network weights and augmentation parameters into a differentiable form by means of stochastic relaxation. This formulation allows us to apply alternative gradient-based methods to solve it, i.e. stochastic natural gradient method with adaptive step-size. To the best of our knowledge, it is the first time that differentiable automatic data augmentation is employed in medical image segmentation tasks. Our numerical experiments demonstrate that the proposed approach significantly outperforms existing build-in data augmentation of state-of-the-art models.