arXiv reaDer
CarveMix:脳病変セグメンテーションのための単純なデータ拡張方法
CarveMix: A Simple Data Augmentation Method for Brain Lesion Segmentation
脳病変のセグメンテーションは、臨床診断のための貴重なツールを提供し、畳み込みニューラルネットワーク(CNN)は、このタスクで前例のない成功を収めています。データ拡張は、CNNのトレーニングを改善するために広く使用されている戦略であり、脳病変のセグメンテーションのための拡張方法の設計は未解決の問題です。この作業では、CNNベースの脳病変セグメンテーションのためのCarveMixと呼ばれる単純なデータ拡張アプローチを提案します。 MixupやCutMixなどの他の「ミックス」ベースの方法と同様に、CarveMixは2つの既存のラベル付き画像を確率的に組み合わせて、新しいラベル付きサンプルを生成します。ただし、画像の組み合わせに基づくこれらの拡張戦略とは異なり、CarveMixは病変を認識し、病変に注意を払って組み合わせを実行し、生成された画像に対して適切な注釈を作成します。具体的には、1つのラベル付き画像から、病変の位置と形状に応じて関心領域(ROI)を切り分け、ROIのサイズを確率分布からサンプリングします。次に、刻まれたROIは、2番目のラベル付き画像の対応するボクセルを置き換え、2番目の画像の注釈もそれに応じて置き換えられます。このようにして、ネットワークトレーニング用の新しいラベル付き画像を生成し、病変情報を保存します。提案された方法を評価するために、2つの脳病変データセットで実験が行われました。結果は、私たちの方法が他の単純なデータ拡張アプローチと比較してセグメンテーションの精度を向上させることを示しています。
Brain lesion segmentation provides a valuable tool for clinical diagnosis, and convolutional neural networks (CNNs) have achieved unprecedented success in the task. Data augmentation is a widely used strategy that improves the training of CNNs, and the design of the augmentation method for brain lesion segmentation is still an open problem. In this work, we propose a simple data augmentation approach, dubbed as CarveMix, for CNN-based brain lesion segmentation. Like other "mix"-based methods, such as Mixup and CutMix, CarveMix stochastically combines two existing labeled images to generate new labeled samples. Yet, unlike these augmentation strategies based on image combination, CarveMix is lesion-aware, where the combination is performed with an attention on the lesions and a proper annotation is created for the generated image. Specifically, from one labeled image we carve a region of interest (ROI) according to the lesion location and geometry, and the size of the ROI is sampled from a probability distribution. The carved ROI then replaces the corresponding voxels in a second labeled image, and the annotation of the second image is replaced accordingly as well. In this way, we generate new labeled images for network training and the lesion information is preserved. To evaluate the proposed method, experiments were performed on two brain lesion datasets. The results show that our method improves the segmentation accuracy compared with other simple data augmentation approaches.
updated: Mon Aug 16 2021 03:54:52 GMT+0000 (UTC)
published: Mon Aug 16 2021 03:54:52 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト