arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.04527cs.LGcs.CV

迈向公平的类别级鲁棒性:类别最优分布对抗训练

Towards Fair Class-wise Robustness: Class Optimal Distribution Adversarial Training

Institute of Information Technology, Information Engineering University(信息工程大学信息技术研究所)

Hongxin Zhi, Hongtao Yu, Shaome Li, Xiuming Zhao, Yiteng Wu

AI总结:

针对对抗训练中类别鲁棒性不均的问题,本文提出CODAT min-max框架,通过分布鲁棒优化和闭式解联合优化类别权重与模型参数,提升鲁棒公平性。

摘要:

对抗训练已被证明是提升深度神经网络抵御对抗攻击鲁棒性的一种高度有效方法。然而,人们观察到它在鲁棒公平性方面存在局限,表现为不同类别之间的鲁棒性存在显著差异。近期缓解该问题的努力转向了类别重加权方法。但是,这些方法缺乏严格的理论分析,并且对权重空间的探索有限,因为它们主要依赖现有启发式算法或直觉来计算权重。此外,由于权重与模型参数的解耦优化,这些方法无法保证优化方向的一致性。它们可能导致次优的权重分配,进而产生次优模型。为解决这些问题,本文提出一种新颖的 min-max 训练框架 Class Optimal Distribution Adversarial Training(CODAT,类别最优分布对抗训练),该框架采用分布鲁棒优化来充分探索类别权重空间,从而能够在具有理论保证的情况下识别最优权重。此外,我们推导了内部最大化问题的闭式最优解,进而得到一个确定性等价目标函数,为权重与模型参数的联合优化提供理论基础。同时,我们提出公平弹性系数,用于从鲁棒性和鲁棒公平性两方面评估算法。在多个数据集上的实验结果表明,所提方法能够有效提升模型的鲁棒公平性,并优于最先进方法。

英文摘要:

Adversarial training has proven to be a highly effective method for improving the robustness of deep neural networks against adversarial attacks. Nonetheless, it has been observed to exhibit a limitation in terms of robust fairness, characterized by a significant disparity in robustness across different classes. Recent efforts to mitigate this problem have turned to class-wise reweighted methods. However, these methods suffer from a lack of rigorous theoretical analysis and are limited in their exploration of the weight space, as they mainly rely on existing heuristic algorithms or intuition to compute weights. In addition, these methods fail to guarantee the consistency of the optimization direction due to the decoupled optimization of weights and the model parameters. They potentially lead to suboptimal weight assignments and consequently, a suboptimal model. To address these problems, this paper proposes a novel min-max training framework, Class Optimal Distribution Adversarial Training (CODAT), which employs distributionally robust optimization to fully explore the class-wise weight space, thus enabling the identification of the optimal weight with theoretical guarantees. Furthermore, we derive a closed-form optimal solution to the internal maximization and then get a deterministic equivalent objective function, which provides a theoretical basis for the joint optimization of weights and model parameters. Meanwhile, we propose a fairness elasticity coefficient for the evaluation of the algorithm with regard to both robustness and robust fairness. Experimental results on various datasets show that the proposed method can effectively improve the robust fairness of the model and outperform the state-of-the-art approaches.

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

↑