论文标题
测试时间傅立叶样式校准用于域概括
Test-time Fourier Style Calibration for Domain Generalization
论文作者
论文摘要
从源域收集到未知目标域的概括机器学习模型的主题是具有挑战性的。尽管许多域的概括(DG)方法已经取得了令人鼓舞的结果,但它们主要依赖于火车时间的源域,而无需在测试时操作目标域。因此,这些方法仍然有可能过度拟合到源域并在目标域上表现不佳。在观察到域与样式密切相关的观察过程中,我们认为减少源和目标样式之间的差距可以提高模型的通用性。为了解决训练期间无法访问目标域的困境,我们引入了测试时间傅立叶样式校准(TF-CAL),以在测试过程中校准目标域样式。为了访问样式,我们利用傅立叶变换将功能分解为振幅(样式)功能和相位(语义)功能。此外,我们提出了一种有效的技术来增强振幅特征(AAF)以补充TF-CAL。在几个流行的DG基准和医学图像的分割数据集上进行了广泛的实验表明,我们的方法表现优于最先进的方法。
The topic of generalizing machine learning models learned on a collection of source domains to unknown target domains is challenging. While many domain generalization (DG) methods have achieved promising results, they primarily rely on the source domains at train-time without manipulating the target domains at test-time. Thus, it is still possible that those methods can overfit to source domains and perform poorly on target domains. Driven by the observation that domains are strongly related to styles, we argue that reducing the gap between source and target styles can boost models' generalizability. To solve the dilemma of having no access to the target domain during training, we introduce Test-time Fourier Style Calibration (TF-Cal) for calibrating the target domain style on the fly during testing. To access styles, we utilize Fourier transformation to decompose features into amplitude (style) features and phase (semantic) features. Furthermore, we present an effective technique to Augment Amplitude Features (AAF) to complement TF-Cal. Extensive experiments on several popular DG benchmarks and a segmentation dataset for medical images demonstrate that our method outperforms state-of-the-art methods.