Flexible-CLmser - Regularized Feedback Connections for Biomedical Image Segmentation.

Boheng Cao,Shikui Tu,Lei Xu
DOI: https://doi.org/10.1109/BIBM52615.2021.9669446
2021-01-01
Abstract:The skip connections in U-Net pass features from the levels of encoder to the ones of decoder in a symmetrical way, which makes U-Net and its variants become state-of-the-art approaches for biomedical image segmentation. However, the U-Net skip connections are unidirectional without considering feedback from the decoder, while this paper exploits the feedback information to refine the segmentation. We develop a deep bidirectional network based on the least mean square error reconstruction (Lmser) self-organizing network, an early model that folds an autoencoder along the central hidden layer such that the neurons on the paired layers between encoder and decoder merge into one, equivalently forming bidirectional skip connections between encoder and decoder. We find that the feedback links indeed increase the segmentation accuracy, but may also bring certain noise into the segmentation. To tackle this problem, we present a gating and masking mechanism on the feedback connections to filter the irrelevant information. Experimental results on MoNuSeg, TNBC, and EM membrane datasets demonstrate that our method are robust and outperforms state-of-the-art methods.
What problem does this paper attempt to address?