arXiv reaDer
コンボ損失:多臓器セグメンテーションにおける入力と出力の不均衡の処理
Combo Loss: Handling Input and Output Imbalance in Multi-Organ Segmentation
異なる医用画像モダリティからの複数の臓器の同時セグメンテーションは、コンピュータ支援診断、コンピュータ支援手術、および治療計画に利用できるため、重要なタスクです。ディープラーニングの最近の進歩のおかげで、医療画像セグメンテーションのためのいくつかのディープニューラルネットワークがこの目的のためにうまく導入されました。この論文では、ボクセルにラベルを付ける深い多臓器セグメンテーションネットワークの学習に焦点を当てています。特に、学習モデルの入力と出力の両方を悩ませている悪名高い不均衡の問題を処理するために、損失関数の重要な選択を調べます。入力不均衡とは、入力トレーニングサンプル(つまり、さまざまなサイズの器官だけでなく、豊富な背景ボクセルに埋め込まれた小さな前景オブジェクト)のクラス不均衡を指します。出力の不均衡とは、推論モデルの誤検知と誤検知の間の不均衡を指します。トレーニングと推論の両方のタイプの不均衡に取り組むために、新しいカリキュラム学習ベースの損失関数を導入します。具体的には、ダイス類似度係数を利用して、モデルパラメーターが悪い極小値に保持されるのを防ぎ、同時に、クロスエントロピー項を使用して誤検出/ネガティブにペナルティを課すことにより、より良いモデルパラメーターを徐々に学習します。提案された損失関数を3つのデータセットで評価しました:5つの標的臓器を用いた全身陽電子放出断層撮影(PET)スキャン、磁気共鳴画像(MRI)前立腺スキャン、および単一の標的臓器、すなわち左心室を用いた超音波心エコー画像。提案された統合損失関数を使用した単純なネットワークアーキテクチャは、最先端の方法よりも優れたパフォーマンスを発揮し、提案された損失を使用すると、競合する方法の結果を改善できることを示します。
Simultaneous segmentation of multiple organs from different medical imaging modalities is a crucial task as it can be utilized for computer-aided diagnosis, computer-assisted surgery, and therapy planning. Thanks to the recent advances in deep learning, several deep neural networks for medical image segmentation have been introduced successfully for this purpose. In this paper, we focus on learning a deep multi-organ segmentation network that labels voxels. In particular, we examine the critical choice of a loss function in order to handle the notorious imbalance problem that plagues both the input and output of a learning model. The input imbalance refers to the class-imbalance in the input training samples (i.e., small foreground objects embedded in an abundance of background voxels, as well as organs of varying sizes). The output imbalance refers to the imbalance between the false positives and false negatives of the inference model. In order to tackle both types of imbalance during training and inference, we introduce a new curriculum learning based loss function. Specifically, we leverage Dice similarity coefficient to deter model parameters from being held at bad local minima and at the same time gradually learn better model parameters by penalizing for false positives/negatives using a cross entropy term. We evaluated the proposed loss function on three datasets: whole body positron emission tomography (PET) scans with 5 target organs, magnetic resonance imaging (MRI) prostate scans, and ultrasound echocardigraphy images with a single target organ i.e., left ventricular. We show that a simple network architecture with the proposed integrative loss function can outperform state-of-the-art methods and results of the competing methods can be improved when our proposed loss is used.
updated: Wed Sep 15 2021 07:46:34 GMT+0000 (UTC)
published: Tue May 08 2018 01:39:59 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アソシエイト