预训练数据集和精细数据集之间的手色差异?

Difference in hand color between pretrain dataset and fine dataset?

我有一个在数据集上预训练的姿势估计模型,其中手是自然颜色。我想在外科医生做手术的手的数据集上微调该模型。那些手戴着手术手套,所以手的图像与正常手有点不同。

pretraine image

finetune image

手部颜色的这种差异会影响模型性能吗? 如果我能让那些手术手的图像更像正常的手,我会得到更好的表现吗?

好吧,这取决于您的 pre-trained 模型已经学会从 pre-training(初始)数据集中捕获的内容。假设您的模型有很多特征图,并且 pre-training 数据集中的肤色变化不足(导致过度拟合问题)。在这种情况下,您的模型可能“选择了阻力最小的路径”并利用它来学习依赖颜色 space 作为特征提取手段的特征图(由于颜色差异可能无法很好地概括)。

你的pre-training数据集match/overlap与你的目标数据集越多,迁移学习的效果就越好。所以是的,很有可能使您的目标数据集(手术手)看起来更类似于您的 pre-training 数据集会对模型的性能产生积极影响。此外,我猜想在您的 pre-training 数据集中引入一些颜色变化(例如,颜色抖动增强)也可以帮助您的模型推广到您的目标数据集。