arXiv reaDer
Towards annotation-efficient segmentation via image-to-image translation
多くの場合、医用画像では、正確な腫瘍セグメンテーションのためにディープニューラルネットワークをトレーニングするのに十分な境界アノテーションを作成することは非常に困難です。画像が腫瘍を示しているかどうか、またはこれらの注釈がない画像にトレーニングを拡張することがないかどうかについて、弱いラベルを使用することを提案します。具体的には、教師なしの目的として、癌の存在と不在という2つのドメイン間の対になっていない画像から画像への変換を採用する半教師ありフレームワークを提案します。翻訳はセグメンテーションに役立つと推測されます。どちらの場合も、ターゲットを背景から分離する必要があります。画像を2つのコードにエンコードします。1つは両方のドメインに共通で、もう1つはプレゼンスドメインに固有です。一般的なコードからデコードすると、健全な画像が生成されます。一意のコードを追加してデコードすると、この画像に残りの変化が生じ、癌が追加されます。翻訳は、存在から不在へ、またはその逆に進行します。最初のケースでは、腫瘍が画像に再度追加され、残余デコーダーをうまく利用してセグメンテーションも実行します。 2番目のケースでは、一意のコードがサンプリングされ、可能性のある腫瘍の分布が生成されます。メソッドを検証するために、公開BRATS(脳、MRI)およびLitS(肝臓、CT)データセットから挑戦的な合成タスクと腫瘍セグメンテーションデータセットを作成しました。自動エンコード(0.73、0.66)と平均教師アプローチ(0.75、0.69)を使用したベースラインの半教師ありトレーニングに対して、明確な改善(脳で0.83ダイス、肝臓で0.74)を示し、注釈付きサンプルのより小さな分布から一般化する能力を示しています。 。
Often in medical imaging, it is prohibitively challenging to produce enough boundary annotations to train deep neural networks for accurate tumor segmentation. We propose the use of weak labels about whether an image presents tumor or whether it is absent to extend training over images that lack these annotations. Specifically, we propose a semi-supervised framework that employs unpaired image-to-image translation between two domains, presence vs. absence of cancer, as the unsupervised objective. We conjecture that translation helps segmentation -- both require the target to be separated from the background. We encode images into two codes: one that is common to both domains and one that is unique to the presence domain. Decoding from the common code yields healthy images; decoding with the addition of the unique code produces a residual change to this image that adds cancer. Translation proceeds from presence to absence and vice versa. In the first case, the tumor is re-added to the image and we successfully exploit the residual decoder to also perform segmentation. In the second case, unique codes are sampled, producing a distribution of possible tumors. To validate the method, we created challenging synthetic tasks and tumor segmentation datasets from public BRATS (brain, MRI) and LitS (liver, CT) datasets. We show a clear improvement (0.83 Dice on brain, 0.74 on liver) over baseline semi-supervised training with autoencoding (0.73, 0.66) and a mean teacher approach (0.75, 0.69), demonstrating the ability to generalize from smaller distributions of annotated samples.
updated: Sat Jun 12 2021 01:10:35 GMT+0000 (UTC)
published: Tue Apr 02 2019 19:35:27 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト