论文标题

用于脑肿瘤检测和分割的多任务上下文遗迹网络

A Multi-task Contextual Atrous Residual Network for Brain Tumor Detection & Segmentation

论文作者

Le, Ngan, Yamazaki, Kashu, Truong, Dat, Quach, Kha Gia, Savvides, Marios

论文摘要

近年来,深层神经网络在包括脑肿瘤分割在内的医学成像中的各种识别和分割任务中取得了最先进的表现。我们调查,分割脑肿瘤正面临着不平衡的数据问题,其中属于背景类别的像素数(非肿瘤像素)大得多,远大于前景类别(肿瘤像素)的像素数量。为了解决这个问题,我们提出了一个多任务网络,该网络形成为级联结构。我们的模型由两个目标组成,即(i)有效区分脑肿瘤区域,(ii)估计脑肿瘤面膜。第一个目标是由我们提出的上下文脑肿瘤检测网络执行的,该网络起着注意门的作用,并专注于脑肿瘤周围的区域,而忽略了与肿瘤相关的遥远邻居背景。第二个目标是建立在3D非常残留的网络和编码decode网络下的,以有效分割大小物体(脑肿瘤)。我们的3D非常残留网络设计具有跳过连接,以使深层的梯度直接传播到浅层层,因此,保留了不同深度的特征并用于相互精炼。为了从卷MRI数据中合并较大的上下文信息,我们的网络利用具有各种内核大小的3D非常卷积,从而扩大了过滤器的接受场。我们提出的网络已在包括BRATS2015,BRATS2017和BRATS2018数据集的各种数据集上进行了评估,并具有验证集和测试集。基于区域的指标和基于表面的指标,我们的性能都基准为基准。我们还与最新方法进行了比较。

In recent years, deep neural networks have achieved state-of-the-art performance in a variety of recognition and segmentation tasks in medical imaging including brain tumor segmentation. We investigate that segmenting a brain tumor is facing to the imbalanced data problem where the number of pixels belonging to the background class (non tumor pixel) is much larger than the number of pixels belonging to the foreground class (tumor pixel). To address this problem, we propose a multi-task network which is formed as a cascaded structure. Our model consists of two targets, i.e., (i) effectively differentiate the brain tumor regions and (ii) estimate the brain tumor mask. The first objective is performed by our proposed contextual brain tumor detection network, which plays a role of an attention gate and focuses on the region around brain tumor only while ignoring the far neighbor background which is less correlated to the tumor. The second objective is built upon a 3D atrous residual network and under an encode-decode network in order to effectively segment both large and small objects (brain tumor). Our 3D atrous residual network is designed with a skip connection to enables the gradient from the deep layers to be directly propagated to shallow layers, thus, features of different depths are preserved and used for refining each other. In order to incorporate larger contextual information from volume MRI data, our network utilizes the 3D atrous convolution with various kernel sizes, which enlarges the receptive field of filters. Our proposed network has been evaluated on various datasets including BRATS2015, BRATS2017 and BRATS2018 datasets with both validation set and testing set. Our performance has been benchmarked by both region-based metrics and surface-based metrics. We also have conducted comparisons against state-of-the-art approaches.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源