医療画像のセグメンテーションは、これまで畳み込みニューラルネットワーク(CNN)で有望な結果を達成してきました。ただし、従来のCNNでは、そのプーリング層が位置などの重要な情報を破棄する傾向があることは議論の余地があります。さらに、CNNは回転とアフィン変換に敏感です。カプセルネットワークは、プーリングレイヤーを動的ルーティングと畳み込みストライドに置き換えることでこのような制限を克服するために提案されたデータ効率の高いネットワーク設計であり、部分全体の関係を維持することを目的としています。カプセルネットワークは、画像認識と自然言語処理で優れたパフォーマンスを示していますが、医療画像セグメンテーション、特にボリューム画像セグメンテーションのアプリケーションは限られています。この作業では、医療用ボリューム画像セグメンテーション用の3Dボクセルベースのカプセルネットワークである3D-UCapsを提案します。 2つの経路を持つネットワークを設計することにより、カプセルの概念をCNNに組み込みます。最初の経路は3D Capsuleブロックによってエンコードされ、2番目の経路は3DCNNブロックによってデコードされます。したがって、3D-UCapsは、空間関係を維持するためのカプセルネットワークと視覚的表現を学習するためのCNNの両方からメリットを継承します。さまざまなデータセットで実験を行い、iSeg-2017、LUNA16、海馬、心臓などの3D-UCapの堅牢性を実証しました。この方法は、以前のカプセルネットワークや3D-Unetsよりも優れています。
Medical image segmentation has been so far achieving promising results with Convolutional Neural Networks (CNNs). However, it is arguable that in traditional CNNs, its pooling layer tends to discard important information such as positions. Moreover, CNNs are sensitive to rotation and affine transformation. Capsule network is a data-efficient network design proposed to overcome such limitations by replacing pooling layers with dynamic routing and convolutional strides, which aims to preserve the part-whole relationships. Capsule network has shown a great performance in image recognition and natural language processing, but applications for medical image segmentation, particularly volumetric image segmentation, has been limited. In this work, we propose 3D-UCaps, a 3D voxel-based Capsule network for medical volumetric image segmentation. We build the concept of capsules into a CNN by designing a network with two pathways: the first pathway is encoded by 3D Capsule blocks, whereas the second pathway is decoded by 3D CNNs blocks. 3D-UCaps, therefore inherits the merits from both Capsule network to preserve the spatial relationship and CNNs to learn visual representation. We conducted experiments on various datasets to demonstrate the robustness of 3D-UCaps including iSeg-2017, LUNA16, Hippocampus, and Cardiac, where our method outperforms previous Capsule networks and 3D-Unets.