该网络由收缩路径和扩展路径组成,收缩路径是一种典型的卷积网络,由重复应用卷积组成,每个卷积后跟一个整流线性单元 (ReLU) 和一个最大池化操作。在收缩过程中,空间信息减少,而特征信息增加。扩展路径通过一系列向上卷积和连接将特征和空间信息与收缩路径中的高分辨率特征相结合。

在生物医学图像分割中有很多应用,例如脑图像分割和肝脏图像分割以及蛋白质结合位点预测。也应用于物理科学,例如在材料显微照片的分析中。以下是此网络的一些变体和应用:

网络训练中数学计算

能量函数是通过最终特征图上的逐像素 soft-max 与交叉熵损失函数相结合来计算的。soft-max 定义为$p_k( x )=\exp \left(a_k( x )\right) /\left(\sum_{k^{\prime}=1}^K \exp \left(a_{k^{\prime}}( x )\right)\right)$,其中 $a_k( x )$ 表示像素位置 $x \in \Omega$且 $\Omega \subset Z ^2$ 处特征通道 $k$ 的激活。$K$ 是类别的数量,$p_k( x )$是近似的最大函数。然后,交叉熵在每个位置上惩罚 $p_{\ell( x )}( x )$ 与 1 的偏差,使用下式

$$ E=\sum_{ x \in \Omega} w( x ) \log \left(p_{\ell( x )}( x )\right) $$

其中 $\ell: \Omega \rightarrow\{1, \ldots, K\}$ 是每个像素的真实标签,$w: \Omega \rightarrow R$ 是我们引入的权重图,以赋予某些像素更多的重要性在训练中。我们预先计算每个地面真实分割的权重图,以补偿训练数据集中某一类像素的不同频率,并迫使网络学习我们在接触细胞之间引入的小分离边界。分离边界使用形态学运算计算。然后计算权重图为

$$ w( x )=w_c( x )+w_0 \cdot \exp \left(-\frac{\left(d_1( x )+d_2( x )\right)^2}{2 \sigma^2}\right) $$

其中$w_c:\Omega\rightarrow R$是平衡类别频率的权重图,$d_1:\Omega\rightarrow R$表示到最近单元格边界的距离,$d_2:\Omega\rightarrow R$表示到第二个最近单元格边界的距离。在我们的实验中,我们设置 $w_0=10$ 和 $\sigma \approx 5$像素。

当只有少量训练样本可用时,数据增强对于教会网络所需的不变性和鲁棒性至关重要。对于显微图像,我们主要需要平移和旋转不变性以及对变形和灰度值变化的鲁棒性。尤其是训练样本的随机弹性变形似乎是用很少的带注释图像训练分割网络的关键概念。

代码构建模型

实现可分为三个部分。首先,我们将定义收缩路径中使用的编码器块。该块由两个 3×3 卷积层、后跟 ReLU 激活层和 2×2 最大池化层组成。第二部分是解码器块,它从下层获取特征图,对其进行上转换、裁剪并将其与同级编码器数据连接,然后执行两个 3×3 卷积层,然后执行 ReLU 激活。第三部分是使用这些块定义模型。

编码模块

def encoder_block(inputs, num_filters):
	x = tf.keras.layers.Conv2D(num_filters,
							3,
							padding = 'valid')(inputs)
	x = tf.keras.layers.Activation('relu')(x)
	x = tf.keras.layers.Conv2D(num_filters,
							3,
							padding = 'valid')(x)
	x = tf.keras.layers.Activation('relu')(x)
	x = tf.keras.layers.MaxPool2D(pool_size = (2, 2),
								strides = 2)(x)
	return x

解码模块

def decoder_block(inputs, skip_features, num_filters):
	x = tf.keras.layers.Conv2DTranspose(num_filters,
										(2, 2),
										strides = 2,
										padding = 'valid')(inputs)
	skip_features = tf.image.resize(skip_features,
									size = (x.shape[1],
											x.shape[2]))
	x = tf.keras.layers.Concatenate()([x, skip_features])
	x = tf.keras.layers.Conv2D(num_filters,
							3,
							padding = 'valid')(x)
	x = tf.keras.layers.Activation('relu')(x)
	x = tf.keras.layers.Conv2D(num_filters, 3, padding = 'valid')(x)
	x = tf.keras.layers.Activation('relu')(x)

	return x