深い畳み込みニューラルネットワークは、さまざまな医療用画像処理モダリティで病変と解剖学をセグメント化するのに効果的であることが証明されています。ただし、小さなサンプルサイズとドメインシフトの問題が存在する場合、これらのモデルは、直感的ではないセグメンテーションエラーのあるマスクを生成することがよくあります。この論文では、ErrorNetと呼ばれるセグメンテーションフレームワークを提案します。これは、事前に学習された形状に基づいて体系的なセグメンテーションエラーをセグメンテーション結果に注入し、その後、注入されたエラーを予測しようとするプロセスを繰り返すことにより、これらのセグメンテーションエラーを修正することを学習します。推論中、ErrorNetは、予測されたエラーマップを初期セグメンテーション結果に追加することにより、セグメンテーションの誤りを修正します。 ErrorNetは、ドメイン固有のパラメーター調整もターゲットドメインからのデータも必要としないため、ドメイン適応またはCRFベースのポスト処理に基づく代替手段よりも優れています。網膜血管セグメンテーションのタスクのために、5つのパブリックデータセットを使用してErrorNetを評価しました。選択したデータセットはサイズと患者数が異なるため、小さなサンプルサイズとドメインシフトの問題を処理する際のErrorNetの有効性を評価できます。この実験は、前述のデータセットの制限がある場合に、ErrorNetがベースセグメンテーションモデル、CRFベースのポストプロセッシングスキーム、およびドメイン適応方法よりも優れていることを示しています。
Deep convolutional neural networks have proved effective in segmenting lesions and anatomies in various medical imaging modalities. However, in the presence of small sample size and domain shift problems, these models often produce masks with non-intuitive segmentation mistakes. In this paper, we propose a segmentation framework called ErrorNet, which learns to correct these segmentation mistakes through the repeated process of injecting systematic segmentation errors to the segmentation result based on a learned shape prior, followed by attempting to predict the injected error. During inference, ErrorNet corrects the segmentation mistakes by adding the predicted error map to the initial segmentation result. ErrorNet has advantages over alternatives based on domain adaptation or CRF-based post processing, because it requires neither domain-specific parameter tuning nor any data from the target domains. We have evaluated ErrorNet using five public datasets for the task of retinal vessel segmentation. The selected datasets differ in size and patient population, allowing us to evaluate the effectiveness of ErrorNet in handling small sample size and domain shift problems. Our experiments demonstrate that ErrorNet outperforms a base segmentation model, a CRF-based post processing scheme, and a domain adaptation method, with a greater performance gain in the presence of the aforementioned dataset limitations.