Lighter U-net for segmenting white matter hyperintensities in MR images

Abstract

White matter hyperintensities (WMH) is one of main consequences of small vessel diseases. Automated WMH segmentation techniques play an important role in clinical research and practice. U-Net has been demonstrated to yield the best precise segmentation results so far. However, sometimes it losses more detailed information as network goes deeper. In addition, it usually depends on data augmentation or a large number of filters. Large filters increase the complexity of model, which may be an obstacle for real-time segmentation on cloud computing. To solve these two issues, a new architecture, Lighter U-Net is proposed to reinforce feature use, to reduce the number of parameters as well as to retain sufficient receptive fields without losing resolution. The extensive experiments suggest that the proposed network achieves comparable performance as the state-of-the-art methods by only using 17% parameters of standard U-Net.