arXiv:2507.18794cs.LG2025-07被引 1

让模型忽略表面特征干扰,提升跨场景泛化能力

CLEAR: Unlearning Spurious Style-Content Associations with Contrastive LEarning with Anti-contrastive Regularization

  • 用对比学习+反对比正则分离内容与风格特征
  • 在新组合下分类准确率提升,风格变化不影响结果
  • 适合医疗等需公平泛化的高风险场景

学习不受表层特征影响的表示对确保测试时特征变化不损害下游预测性能至关重要。例如,在医疗应用中,我们希望学习与病理相关但不受种族、性别等生理差异影响的特征,从而保证预测在所有人群中公平且可泛化。本文提出对比学习结合反对比正则(CLEAR)框架,有效在训练阶段分离任务相关的本质特征与无关的表面特征,使测试时表层特征发生改变仍能保持良好性能。假设数据表示可语义分为任务相关的内容特征和无关的风格特征,而后者虽与任务无关,但在训练数据中与内容存在非泛化关联,可能降低性能。我们证明所提出的配对切换(Pair-Switching, PS)反对比惩罚能最小化风格属性与内容标签之间的互信息。将CLEAR应用于变分自编码器(VAE)的潜在空间,并在多个图像数据集上进行定量与定性评估。结果表明,CLEAR-VAE可实现:(a) 任意样本间的内容与风格交换与插值;(b) 在先前未见的内容-风格组合下提升下游分类性能。代码将公开。

原文摘要 · Abstract (English)

Learning representations unaffected by superficial characteristics is important to ensure that shifts in these characteristics at test time do not compromise downstream prediction performance. For instance, in healthcare applications, we might like to learn features that contain information about pathology yet are unaffected by race, sex, and other sources of physiologic variability, thereby ensuring predictions are equitable and generalizable across all demographics. Here we propose Contrastive LEarning with Anti-contrastive Regularization (CLEAR), an intuitive and easy-to-implement framework that effectively separates essential (i.e., task-relevant) characteristics from superficial (i.e., task-irrelevant) characteristics during training, leading to better performance when superficial characteristics shift at test time. We begin by supposing that data representations can be semantically separated into task-relevant content features, which contain information relevant to downstream tasks, and task-irrelevant style features, which encompass superficial attributes that are irrelevant to these tasks, yet may degrade performance due to associations with content present in training data that do not generalize. We then prove that our anti-contrastive penalty, which we call Pair-Switching (PS), minimizes the Mutual Information between the style attributes and content labels. Finally, we instantiate CLEAR in the latent space of a Variational Auto-Encoder (VAE), then perform experiments to quantitatively and qualitatively evaluate the resulting CLEAR-VAE over several image datasets. Our results show that CLEAR-VAE allows us to: (a) swap and interpolate content and style between any pair of samples, and (b) improve downstream classification performance in the presence of previously unseen combinations of content and style. Our code will be made publicly available.

表示学习公平性风格分离VAE

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。