A 2D U-Net to segment nuclei from images of cells from the Broad Bioimage Benchmark Collection.
In this notebook, we use a 2D U-Net to train a segmentation model that can segment nuclei from images of cells.
U-Net is a fully-convolutional network created with an encoder-decoder topology. Ronnenberger et al. created the topology in 2015 for use in biomedical imaging.
Residual or skip connections mitigate the vanishing gradient problem and smooth the loss landscape to make training very deep networks possible.
Hao Li, Zheng Xu, Gavin Taylor, Christoph Studer, Tom Goldstein. Visualizing the Loss Landscape of Neural Nets
Encoders/Decoder topologies compress the input space into a smaller, latent space and then expand the latent space back to some output space. It's similar to how we take books and assign (seemingly) random alphanumeric codes to them so that we can organize them properly in a large library.