カプセルネットワークは、医療画像のセグメンテーションタスクに正常に適用された最近の新しいディープネットワークアーキテクチャです。この作業は、自己監視学習による体積医療画像セグメンテーションのためのカプセルネットワークを拡張します。以前のカプセルネットワークと比較して重みの初期化の問題を改善するために、カプセルネットワークの事前トレーニングに自己監視学習を活用します。この場合、プレテキストタスクは自己再構築によって最適化されます。当社のカプセルネットワークSS-3DCapsNetは、3DCapsuleエンコーダーと3DCNNsデコーダーを備えたUNetベースのアーキテクチャーを備えています。 iSeg-2017、海馬、心臓を含む複数のデータセットでの実験は、自己監視型の事前トレーニングを備えた3Dカプセルネットワークが、以前のカプセルネットワークや3D-UNetsよりも大幅に優れていることを示しています。
Capsule network is a recent new deep network architecture that has been applied successfully for medical image segmentation tasks. This work extends capsule networks for volumetric medical image segmentation with self-supervised learning. To improve on the problem of weight initialization compared to previous capsule networks, we leverage self-supervised learning for capsule networks pre-training, where our pretext-task is optimized by self-reconstruction. Our capsule network, SS-3DCapsNet, has a UNet-based architecture with a 3D Capsule encoder and 3D CNNs decoder. Our experiments on multiple datasets including iSeg-2017, Hippocampus, and Cardiac demonstrate that our 3D capsule network with self-supervised pre-training considerably outperforms previous capsule networks and 3D-UNets.