通过彩色图像着色实现无监督特征学习.

着色(colorization)是一种很强大的自监督学习任务:训练模型以对输入灰度图像进行着色;准确地说,将该图像映射到量化颜色值输出的分布上。

本文中图像着色是在CIE Lab*颜色空间中进行的,相比于适用于物理设备的颜色空间RGBCMYKLab*颜色更适合人类视觉;Lab*颜色空间包括:

通常构造在量化颜色值上预测概率分布的交叉熵损失比构造原始颜色值的L2损失表现更好,因此对ab颜色空间进行量化。图a给出了栅格尺寸为$10$的ab量化颜色空间,则共有$313$种ab对的颜色取值;图b给出了ab值的经验概率分布;图c给出了根据L值决定的ab值的经验概率分布。

为了在常见颜色(通常对应较低ab值的常见背景如云、墙和灰尘)和罕见颜色(可能与图像中的关键对象有关)之间进行平衡,使用加权项重新平衡损失函数,以增加不常见颜色取值的损失。加权项被构造为$(1-λ)*$高斯核平滑经验概率分布$+λ*$均匀分布,其中两个分布都在量化ab颜色空间上定义。

实现图像着色的网络构造如下: