U-net of joint spatial domains with multi-scale atrous convolution for rectal image segmentation
摘要
Medical image segmentation is very important for the diagnosis of related diseases. To reduce the labeling work of related medical images, numerous models based on U-Net have been proposed to achieve automatic segmentation of target regions. However, most of these models are only trained in one coordinate system, ignoring the joint effects of different spatial coordinate systems. In addition, most of the encoding modules in these models do not pay attention to multi-scale spatial information. Our proposed solution for the aforementioned challenges involves using U-Net model with joint spatial domains and multi-scale encoding module, which enables us to segment rectal image better. The model includes a self-designed multi-layer dilated convolution encoding module named AIR (Atrous Inception Residual Block), to achieve a multi-scale content fusion. Besides this, it utilizes the center point and polar coordinates to realize attention mechanism and rotation invariance. Furthermore, retraining the output of polar coordinate network with Cartesian coordinate system realizes the translation invariance of segmentation. Compared with the commonly used medical segmentation models, the dice coefficient of our model is improved by about 2% on our in-house rectal dataset.