[控场AI]
· 4 分钟阅读· 2,197 字

FDSML:用参数感知三元组损失革新合成预训练

FDSML:用参数感知三元组损失革新合成预训练

FDSML用生成参数替代离散标签构建关系监督,显著提升合成预训练的检索迁移性能。

文章介绍了合成数据预训练方法FDSML(Formula-Driven Supervised Metric Learning),它针对传统FDSL仅用离散类别标签进行预训练的局限,将监督信号转向分形图像的生成参数(IFS参数),通过参数感知三元组损失(PA-Triplet Loss)依据参数空间距离分配正负样本对,从而学习连续的相似性度量空间。配套的FineFractalDB数据集提供更高的类内变异性和稳定结构。在ResNet-50上,FDSML在Stanford Online Products和Stanford Cars上的Recall@1分别从65.6%提升至74.0%、从56.2%提升至78.4%,同时下游分类性能保持竞争力并提升鲁棒性。文章认为关系型监督为合成预训练提供了新范式,未来可扩展到扩散模型等其他生成模型。

合成数据预训练正迎来一次关键升级。传统上,Formula-Driven Supervised Learning(FDSL)通过分形公式生成图像并离散化标签来完成分类式预训练,但这种方式忽略了生成过程中蕴含的连续关系信息。最新提出的FDSML(Formula-Driven Supervised Metric Learning)将监督信号从离散类别标签转向图像生成参数,通过参数感知三元组损失(PA-Triplet Loss)构建关系型监督,在图像检索任务上实现了显著提升。

FDSL的局限:离散标签丢失了连续性

FDSL的核心思想是利用数学公式(如分形)生成合成图像,无需真实数据即可预训练视觉模型。然而,现有方法将预训练建模为对离散化标签的分类任务,这导致生成过程中丰富的连续性信息被丢弃——图像生成参数(如迭代函数系统的参数)本可以揭示样本之间的相对相似性,但分类标签只告诉模型“属于哪一类”,无法表达“谁和谁更相似”。

据论文描述,这种离散化处理限制了模型学习细粒度关系的能力,尤其影响那些依赖相对排序的任务,例如图像检索。FDSML正是瞄准这一缺口,试图将监督信号从“类别”层面提升到“关系”层面。

迭代函数系统(Iterated Function System, IFS)是分形几何中的经典构造:它由一组压缩仿射变换组成,通过反复将这些变换作用于初始图形,最终收敛到一个自相似的分形吸引子(如谢尔宾斯基三角形或巴恩斯利蕨)。在FDSL中,每个分形图像正是由一组随机采样的IFS参数生成的,因此这些参数天然构成了一个连续的高维空间——参数微小的变化会导致图像纹理、结构发生平滑的渐变。FDSML正是利用这一性质:两个图像对应的IFS参数向量在欧氏空间中的距离越近,它们视觉上的相似度通常也越高。这使得IFS参数成为比离散类别标签更细粒度的监督信号。

三元组损失(Triplet Loss)是度量学习的核心目标函数之一,广泛用于人脸识别和图像检索。它每次接收一个三元组(锚点、正样本、负样本),通过优化使锚点与正样本的嵌入距离小于锚点与负样本的距离,并通常加入一个间隔(margin)防止退化解。FDSML的PA-Triplet Loss创新在于:正负样本的划分不再依赖人工标注的类别,而是依据样本在IFS参数空间中的距离自动确定,从而将生成参数直接转化为关系监督信号。

FDSML的核心:参数感知三元组损失

FDSML由两个关键组件构成:FineFractalDB数据集和PA-Triplet Loss。FineFractalDB是对现有分形数据集的精炼,具有更高的类内变异性和更稳定的结构,为关系学习提供了更丰富的样本对。而PA-Triplet Loss则根据底层迭代函数系统(IFS)参数空间中的距离来分配正负样本对——参数空间中距离近的样本被视为正对,距离远的视为负对,从而训练嵌入向量尊重由参数导出的相对相似性排序。

这种设计让模型不再仅仅学习区分类别,而是学习一个连续的相似性度量空间。论文指出,该目标函数直接优化嵌入以保持参数导出的相对顺序,这为迁移到检索任务奠定了更好的基础。

实验效果:检索性能大幅提升

在ResNet-50 backbone上,FDSML预训练模型在多个检索基准上优于FDSL。具体而言,在Stanford Online Products数据集上,Recall@1从65.6%提升至74.0%;在Stanford Cars上,从56.2%提升至78.4%。这些提升表明,参数导出的关系监督比基于标签的分类预训练提供了更强的合成预训练信号,尤其有利于识别之外的迁移任务。

除了检索任务,FDSML在下游分类任务上保持竞争力,并在多种损坏设置下展现出更好的鲁棒性。这意味着关系型监督不仅没有牺牲分类性能,还增强了模型对图像退化的抵抗能力。

Recall@1是图像检索中最常用的评价指标之一,指在检索结果排名第一的候选中命中正确匹配样本的查询比例。例如在Stanford Online Products(SOP)数据集上,Recall@1从65.6%提升至74.0%,意味着每100次查询中,正确匹配出现在首位的结果多了约8.4次。SOP是一个经典的度量学习基准,包含约12万张在线商品图像、2万多个类别,类别间视觉差异细微,非常考验模型对细粒度相似性的判别能力;Stanford Cars则包含196类车型、约1.6万张图像,同样以类内差异大、类间差异小著称。这两个基准上的显著提升,说明FDSML学到的嵌入空间在细粒度排序任务上具有更强的泛化能力。ResNet-50作为骨干网络,是衡量预训练方法迁移效果的常用标准架构,保证了比较的公平性。

意义与展望:关系监督成为合成预训练新范式

FDSML的提出提示了一个更广泛的趋势:合成预训练的价值不仅在于“无数据”,更在于如何设计监督信号。通过挖掘生成过程中的连续参数,模型可以学到更通用的表示,这对于数据稀缺或隐私敏感的场景尤为重要。未来,这种参数感知的关系学习思路或许可以扩展到其他生成模型(如扩散模型)的预训练中,进一步释放合成数据的潜力。

FDSML论文标题截图

#FDSML#合成数据预训练#三元组损失#图像检索#Formula-Driven Supervised Learning#参数感知#FineFractalDB#PA-Triplet Loss
分享:

相关推荐