发现大模型在长尾数据上存在参数与数据双重偏差,提出因果调整新方法提升性能。
Rethinking the Bias of Foundation Model under Long-tailed Distribution
- 将长尾偏差拆解为参数与数据不平衡,识别参数偏差更难修复
- 提出因果背门调整法,消除输入与标签间的虚假关联
- 在多个数据集上平均提升1.67%,适合关注模型公平性的研究者
长尾学习因实际意义受关注,大模型微调成为主流。然而现有方法多聚焦利用大模型知识,忽视其预训练数据不均衡带来的固有偏差。本文研究预训练不平衡如何影响下游长尾任务,发现此类偏差表现为参数不平衡与数据不平衡。微调中,参数不平衡起主导作用,而数据不平衡可通过现有重平衡策略缓解;但当前重平衡技术(如调整logits)无法有效解决参数不平衡问题。为此,我们基于因果学习,将不完整语义因子视为混杂因子,引入新型背门调整方法,学习输入与标签的真实因果效应而非数据相关性。实验表明,该方法在各数据集上平均性能提升约1.67%。代码已公开:https://github.com/JiahaoChen1/Pre-train-Imbalance
原文摘要 · Abstract (English)
Long-tailed learning has garnered increasing attention due to its practical significance. Among the various approaches, the fine-tuning paradigm has gained considerable interest with the advent of foundation models. However, most existing methods primarily focus on leveraging knowledge from these models, overlooking the inherent biases introduced by the imbalanced training data they rely on. In this paper, we examine how such imbalances from pre-training affect long-tailed downstream tasks. Specifically, we find the imbalance biases inherited in foundation models on downstream task as parameter imbalance and data imbalance. During fine-tuning, we observe that parameter imbalance plays a more critical role, while data imbalance can be mitigated using existing re-balancing strategies. Moreover, we find that parameter imbalance cannot be effectively addressed by current re-balancing techniques, such as adjusting the logits, during training, unlike data imbalance. To tackle both imbalances simultaneously, we build our method on causal learning and view the incomplete semantic factor as the confounder, which brings spurious correlations between input samples and labels. To resolve the negative effects of this, we propose a novel backdoor adjustment method that learns the true causal effect between input samples and labels, rather than merely fitting the correlations in the data. Notably, we achieve an average performance increase of about $1.67\%$ on each dataset. Code is available: https://github.com/JiahaoChen1/Pre-train-Imbalance
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。