arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2401.09191cs.LGmath.OCstat.ML

多类分类中计算对抗训练下界的最优传输方法

An Optimal Transport Approach for Computing Adversarial Training Lower Bounds in Multiclass Classification

University of Wisconsin-Madison(威斯康星大学麦迪逊分校);UC Santa Barbara(加州大学圣塔芭芭拉分校);University of British Columbia(不列颠哥伦比亚大学);Tufts University(塔夫茨大学)

Nicolas Garcia Trillos, Matt Jacobs, Jakwang Kim, Matthew Werenski

AI总结:

本文利用多边际最优传输与多类对抗训练的联系,提出基于线性规划和Sinkhorn的可处理算法,用于计算最优对抗风险通用下界并识别最优分类器,在MNIST和CIFAR-10上验证了其有效性。

摘要:

尽管基于深度学习的算法取得了成功,但众所周知,神经网络可能缺乏鲁棒性。增强鲁棒性的一种流行范式是对抗训练(AT),然而这会引入许多计算和理论上的困难。近期工作建立了多类分类场景下的对抗训练与多边际最优传输(MOT)之间的联系,为研究该问题开辟了一套新工具。本文利用这一MOT联系,提出了计算上可处理的数值算法,用于计算最优对抗风险的通用下界并识别最优分类器。我们提出了两种主要算法,分别基于线性规划(LP)和熵正则化(Sinkhorn)。我们的核心见解是,可以无害地截断类之间的高阶交互,从而避免MOT问题中通常遇到的组合运行时间。我们通过在MNIST和CIFAR-10上的实验验证了这些结果,证明了所提方法的可处理性。

英文摘要:

Despite the success of deep learning-based algorithms, it is widely known that neural networks may fail to be robust. A popular paradigm to enforce robustness is adversarial training (AT), however, this introduces many computational and theoretical difficulties. Recent works have developed a connection between AT in the multiclass classification setting and multimarginal optimal transport (MOT), unlocking a new set of tools to study this problem. In this paper, we leverage the MOT connection to propose computationally tractable numerical algorithms for computing universal lower bounds on the optimal adversarial risk and identifying optimal classifiers. We propose two main algorithms based on linear programming (LP) and entropic regularization (Sinkhorn). Our key insight is that one can harmlessly truncate the higher order interactions between classes, preventing the combinatorial run times typically encountered in MOT problems. We validate these results with experiments on MNIST and CIFAR-$10$, which demonstrate the tractability of our approach.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑