7 天前

任意风格迁移与领域泛化的精确特征分布匹配

Yabin Zhang, Minghan Li, Ruihuang Li, Kui Jia, Lei Zhang
任意风格迁移与领域泛化的精确特征分布匹配
摘要

任意风格迁移(Arbitrary Style Transfer, AST)与域泛化(Domain Generalization, DG)是视觉学习中重要且具有挑战性的任务,均可建模为特征分布匹配问题。传统特征分布匹配方法通常基于特征服从高斯分布的假设,通过匹配特征的均值与标准差来实现分布对齐。然而,现实世界数据的特征分布远比高斯分布复杂,仅依赖一阶与二阶统计量难以精确匹配;而采用高阶统计量进行分布匹配在计算上又往往不可行。在本工作中,据我们所知,首次提出通过精确匹配图像特征的经验累积分布函数(empirical Cumulative Distribution Functions, eCDFs)来实现精确特征分布匹配(Exact Feature Distribution Matching, EFDM)。该方法可通过在图像特征空间中应用精确直方图匹配(Exact Histogram Matching, EHM)来实现。特别地,本文引入一种高效的EHM算法——Sort-Matching,以轻量级、即插即用的方式实现EFDM,计算开销极低。所提出的EFDM方法在多种AST与DG任务上均得到验证,取得了新的最先进性能。相关代码已开源,地址为:https://github.com/YBZh/EFDM。

任意风格迁移与领域泛化的精确特征分布匹配 | 最新论文 | HyperAI超神经