Faster Nonlocal UNet for Cell Segmentation in Microscopy Images

Xuhao Lin,Shengsheng Wang
DOI: https://doi.org/10.1007/978-3-030-82153-1_38
2021-01-01
Abstract:Deep neural networks have shown great potential in medical image segmentation fields. Most of the current methods are based on UNet, which generates long-range dependence step by step by stacking a large number of local operations. However, global information cannot be effectively aggregated using only local operations. The nonlocal module is an effective method for obtaining global information, but the nonlocal block is always criticized for its exorbitant computation and GPU consumption. For the purpose of solving this problem, a faster nonlocal UNet (FN-UNet) is proposed for obtaining the long-range dependency of biomedical images through a more effective and efficient method. Inspired by the recently introduced nonlocal U-Nets and asymmetric nonlocal neural networks, we integrate the asymmetric fusion nonlocal block (AFNB) and asymmetric pyramid nonlocal block (APNB) into the proposed network. Additionally, thorough experiments are performed on the cell segmentation dataset of the ISBI Challenge. The results show that compared with the typical nonlocal UNet network, the proposed network can obtain better results on the basis of occupying less GPU memory and computation.
What problem does this paper attempt to address?