论文标题
IB-U-NET:使用3D电感偏置内核改进医学图像分割任务
IB-U-Nets: Improving medical image segmentation tasks with 3D Inductive Biased kernels
论文作者
论文摘要
尽管卷积神经网络在3D医疗图像分割方面取得了成功,但当前使用的体系结构仍然不足以适应不同的扫描仪的协议以及它们产生的图像属性的多样性。此外,访问带有感兴趣的注释区域的大规模数据集很少,因此很难获得良好的结果。为了克服这些挑战,我们介绍了IB-U-Nets,这是一种具有诱导性偏见的新型结构,灵感来自脊椎动物的视觉处理。将3D U-NET作为基座,我们将两个3D残差组件添加到第二个编码器块中。它们提供了感应性偏置,帮助U-NETs从3D图像中分割出具有稳健性和准确性的3D图像的解剖结构。我们使用相同的培训和测试管道(包括数据处理,增强和交叉验证)将IB-U-NET与最新的3D U-NET与多种方式和器官(例如前列腺和脾脏)进行了比较。我们的结果表明,IB-U-NET的鲁棒性和准确性,尤其是在小型数据集上,就像医疗形象分析中通常情况一样。 IB-U-NETS源代码和模型公开可用。
Despite the success of convolutional neural networks for 3D medical-image segmentation, the architectures currently used are still not robust enough to the protocols of different scanners, and the variety of image properties they produce. Moreover, access to large-scale datasets with annotated regions of interest is scarce, and obtaining good results is thus difficult. To overcome these challenges, we introduce IB-U-Nets, a novel architecture with inductive bias, inspired by the visual processing in vertebrates. With the 3D U-Net as the base, we add two 3D residual components to the second encoder blocks. They provide an inductive bias, helping U-Nets to segment anatomical structures from 3D images with increased robustness and accuracy. We compared IB-U-Nets with state-of-the-art 3D U-Nets on multiple modalities and organs, such as the prostate and spleen, using the same training and testing pipeline, including data processing, augmentation and cross-validation. Our results demonstrate the superior robustness and accuracy of IB-U-Nets, especially on small datasets, as is typically the case in medical-image analysis. IB-U-Nets source code and models are publicly available.