← 最新论文
🤖 machine learning

Taming the Long Tail: Rebalancing Adversarial Training via Adaptive Perturbation

本文提出 RobustLT,这是一种即插即用框架,能够在对抗训练过程中自适应地调整扰动,从而同时提升模型鲁棒性并解决长尾数据集中的类别不平衡问题。

原作者: Lilin Zhang, Yimo Guo, Yue Li, Jiancheng Shi, Xianggen Liu

发布于 2026-05-14
📖 1 分钟阅读☕ 轻松阅读

原作者: Lilin Zhang, Yimo Guo, Yue Li, Jiancheng Shi, Xianggen Liu

原始论文采用 CC BY 4.0 许可(http://creativecommons.org/licenses/by/4.0/)。 这是对下方论文的AI生成解释。它不是由作者撰写或认可的。如需技术准确性,请参阅原始论文。 阅读完整免责声明

想象一下,你正在训练一名安保人员(一个 AI 模型)来识别入侵者。在一个完美的教室里,老师向安保人员展示了 1,000 张猫的照片和 1,000 张狗的照片。安保人员学会了同样出色地识别这两者。

但在现实世界中,数据往往是“长尾”的。想象一下,老师向安保人员展示了 1,000 张猫的照片,却只有 5 张稀有、异域鸟类的照片。安保人员自然成为了猫方面的专家,但在识别鸟类方面却表现糟糕。它们对猫变得“过度自信”,而忽略了鸟类。

现在,想象一名黑客(对抗者)试图通过给照片添加微小的、看不见的划痕来欺骗这名安保人员。这被称为“对抗性攻击”。通常,研究人员会在平衡的教室(猫和狗数量相等)中测试安保人员。但是,当黑客攻击一名已经偏向猫的安保人员时会发生什么?安保人员在识别稀有鸟类方面会失败得更惨。

这篇论文《驯服长尾》提出了一种训练这些安保人员的新方法,使它们不仅能抵御黑客的诡计,还能停止忽略稀有鸟类。

核心问题:倾斜的训练场

作者发现,在处理稀有物品(即“尾部”)时,标准训练存在两个主要缺陷:

  1. 倾斜的目标:训练目标存在偏差。由于猫的数量如此之多,安保人员将所有精力都集中在它们身上,而忽略了鸟类。
  2. 不稳定的混乱:当黑客试图欺骗安保人员时,“诡计”(扰动)会以一种混乱的方式改变数据分布。安保人员感到困惑,训练变得不稳定。

解决方案:"RobustLT"(自适应教练)

作者创建了一个名为RobustLT的框架。把它想象成一位智能教练,根据谁在挣扎来调整训练课程的难度。

RobustLT 不是给每个学生布置同样数量的作业,而是使用两种特殊工具:

1. 类别扰动平衡(CPB)——“公平调节器”

想象黑客正试图欺骗安保人员。

  • 对于常见的猫:教练告诉黑客:“放轻松。不要让划痕太难被发现。”如果猫已经很容易识别,我们就不需要过度训练它们。
  • 对于稀有的鸟类:教练告诉黑客:“加把劲!让划痕非常难以发现。”

通过让稀有类别的“诡计”更难,而让常见类别的“诡计”更容易,安保人员被迫关注鸟类。这就像给 struggling 的学生布置额外、更难的练习题,而给优等生布置较轻的作业以保持他们的专注。这平衡了训练,使安保人员学会像识别猫一样出色地识别鸟类。

2. 对抗迭代加权(AIW)——“稳健之手”

训练 AI 就像学习骑自行车。如果你试图在摇晃的道路上立即骑得太快,你会摔倒。

  • 早期训练:安保人员是新的且不稳定。教练说:“让我们从非常微小、温和的划痕开始。不要压倒安保人员。”
  • 后期训练:随着安保人员变得更好,教练逐渐增加划痕的强度。

这个工具确保训练不会剧烈波动。它稳定了整个过程,确保安保人员在面对最严峻的攻击之前建立起坚实的基础。

结果:均衡的安保人员

这篇论文在几个数据集(如 CIFAR 和 TinyImageNet)上测试了这种方法,其中某些类别有数千张图像,而其他类别则很少。

  • 在 RobustLT 之前:安保人员非常擅长识别常见物品,但在识别稀有物品方面表现糟糕,尤其是当黑客试图欺骗它们时。
  • 在 RobustLT 之后:安保人员在识别稀有物品方面变得更好,同时没有失去识别常见物品的能力。它变得对黑客“稳健”,并且在所有类别之间“均衡”。

主要启示

这篇论文证明,你不需要丢弃“稀有”数据或完全重新设计 AI。你只需要在训练期间调整攻击的强度。通过给稀有类别一个“更艰难”的挑战,而给常见类别一个“更轻松”的挑战,你可以解决不平衡问题,同时使 AI 对黑客更强大。

简而言之:RobustLT 是一个智能的训练计划,它量身定制“黑客攻击”的难度,以确保没有任何类别被抛在后面。

您所在领域的论文太多了?

获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。

试用 Digest →