[PyTorch Implementation] 3D Segmentation model - VoxResNet, Attention U-Net, V-Net

PyTorch로 구현한 3D Segmentation 모델 3가지입니다.

 

github.com/bo-10000/pytorch_3d_segmentation

bo-10000/pytorch_3d_segmentation

 

 

VoxResNet

residual learning을 적용한 모델입니다. residual module(b)을 여러 겹 쌓은 구조를 가지고 있고, 중간중간 auxiliary classifier을 배치하여 그 결과를 합쳤습니다.

MICCAI MRIBrainS challenge에서 1위를 차지했습니다.

 

논문 링크 : https://arxiv.org/pdf/1608.05895.pdf

 

 

Attention U-Net

U-Net 구조에 Attention Gate 개념을 추가했습니다. Attention Gate는 필요한 feature만을 강조하면서, 불필요한 부분은 억제하는 효과가 있다고 합니다. Attention Gate의 구조는 Feature2에 나타나 있습니다.

 

논문 링크 : https://arxiv.org/pdf/1804.03999.pdf

 

 

V-Net

U-Net과 매우 유사한 구조를 가지고 있으며, 매 stage마다 residual connection을 적용하였습니다. 또한 PReLU를 activation function으로 사용하였습니다.

 

논문 링크 : https://arxiv.org/pdf/1606.04797.pdf