用纯文本训练消除视觉语言模型偏见,无需图像数据
Debiasing Vison-Language Models with Text-Only Training
- 用文本生成图像的思路,让文本编码器当图像编码器
- 在Waterbirds和CelebA上显著提升少数群体鲁棒性
- 适合无图像数据或标注成本高的场景
预训练视觉语言模型(如CLIP)通过统一嵌入空间对齐图文,在下游任务中表现优异。但因预训练数据分布不均,实际应用中存在偏见问题。现有去偏方法受限于少数群体图像样本不足及标注成本高。为此,我们提出纯文本去偏框架TOD,利用文本作为图像的训练范式,将文本编码器重用于图像编码,无需图像数据。同时,借助大语言模型生成平衡文本数据,用于提示调优。然而发现模型会过度依赖文本中的标签名称作为监督信号。为此,我们引入多目标预测(MTP)任务,促使模型关注复杂语境,区分目标与偏见信息。在Waterbirds和CelebA数据集上的实验表明,该方法显著提升群体鲁棒性,达到无图像方法的领先水平,甚至媲美有图像监督的方法。且可适应多属性或未知偏见场景,展现强泛化能力。
原文摘要 · Abstract (English)
Pre-trained vision-language models (VLMs), such as CLIP, have exhibited remarkable performance across various downstream tasks by aligning text and images in a unified embedding space. However, due to the imbalanced distribution of pre-trained datasets, CLIP suffers from the bias problem in real-world applications. Existing debiasing methods struggle to obtain sufficient image samples for minority groups and incur high costs for group labeling. To address the limitations, we propose a Text-Only Debiasing framework called TOD, leveraging a text-as-image training paradigm to mitigate visual biases. Specifically, this approach repurposes the text encoder to function as an image encoder, thereby eliminating the need for image data. Simultaneously, it utilizes a large language model (LLM) to generate a balanced text dataset, which is then used for prompt tuning. However, we observed that the model overfits to the text modality because label names, serving as supervision signals, appear explicitly in the texts. To address this issue, we further introduce a Multi-Target Prediction (MTP) task that motivates the model to focus on complex contexts and distinguish between target and biased information. Extensive experiments on the Waterbirds and CelebA datasets show that our method significantly improves group robustness, achieving state-of-the-art results among image-free methods and even competitive performance compared to image-supervised methods. Furthermore, the proposed method can be adapted to challenging scenarios with multiple or unknown bias attributes, demonstrating its strong generalization and robustness.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。