arXiv reaDer
強化された深層学習肝腫瘍セグメンテーションのための部分畳み込み生成的敵対的ネットワークを使用した自由形式病変合成
Free-form Lesion Synthesis Using a Partial Convolution Generative Adversarial Network for Enhanced Deep Learning Liver Tumor Segmentation
自動深層学習セグメンテーションモデルは、セグメンテーションの効率と精度の両方を向上させることが示されています。ただし、堅牢なセグメンテーションモデルのトレーニングには、かなり大きなラベル付きトレーニングサンプルが必要であり、実用的でない場合があります。この研究は、ネットワークトレーニングを強化するために使用できる合成病変を生成するための深層学習フレームワークを開発することを目的としています。病変合成ネットワークは、修正された生成的敵対的ネットワーク(GAN)です。具体的には、Unetのようなジェネレーターを構築するための部分畳み込み戦略を革新しました。弁別器は、勾配ペナルティとスペクトル正規化を備えたワッサースタインGANを使用して設計されています。主成分分析に基づくマスク生成法は、さまざまな病変形状をモデル化するために開発されました。生成されたマスクは、病変合成ネットワークを介して肝臓病変に変換されます。病変合成フレームワークの病変テクスチャを評価し、合成病変を使用して病変セグメンテーションネットワークをトレーニングし、このフレームワークの有効性をさらに検証しました。すべてのネットワークは、LITSの公開データセットでトレーニングおよびテストされています。提案されたアプローチによって生成された合成病変は、2つの使用されたテクスチャパラメータ、GLCMエネルギーとGLCM相関の実際の病変と比較して非常に類似したヒストグラム分布を持っています。 GLCMエネルギーとGLCM相関のカルバックライブラー発散は、それぞれ0.01と0.10でした。腫瘍セグメンテーションネットワークに合成病変を含めると、U-Netのセグメンテーションダイスのパフォーマンスが67.3%から71.4%に大幅に向上しました(p <0.05)。一方、ボリュームの精度と感度は、それぞれ74.6%から76.0%(p = 0.23)と66.1%から70.9%(p <0.01)に向上します。合成データにより、セグメンテーションのパフォーマンスが大幅に向上します。
Automatic deep learning segmentation models has been shown to improve both the segmentation efficiency and the accuracy. However, training a robust segmentation model requires considerably large labeled training samples, which may be impractical. This study aimed to develop a deep learning framework for generating synthetic lesions that can be used to enhance network training. The lesion synthesis network is a modified generative adversarial network (GAN). Specifically, we innovated a partial convolution strategy to construct an Unet-like generator. The discriminator is designed using Wasserstein GAN with gradient penalty and spectral normalization. A mask generation method based on principal component analysis was developed to model various lesion shapes. The generated masks are then converted into liver lesions through a lesion synthesis network. The lesion synthesis framework was evaluated for lesion textures, and the synthetic lesions were used to train a lesion segmentation network to further validate the effectiveness of this framework. All the networks are trained and tested on the public dataset from LITS. The synthetic lesions generated by the proposed approach have very similar histogram distributions compared to the real lesions for the two employed texture parameters, GLCM-energy and GLCM-correlation. The Kullback-Leibler divergence of GLCM-energy and GLCM-correlation were 0.01 and 0.10, respectively. Including the synthetic lesions in the tumor segmentation network improved the segmentation dice performance of U-Net significantly from 67.3% to 71.4% (p<0.05). Meanwhile, the volume precision and sensitivity improve from 74.6% to 76.0% (p=0.23) and 66.1% to 70.9% (p<0.01), respectively. The synthetic data significantly improves the segmentation performance.
updated: Sat Jun 18 2022 00:40:41 GMT+0000 (UTC)
published: Sat Jun 18 2022 00:40:41 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アソシエイト