
DeepCustom: 提升深度学习系统鲁棒性的快速定制化对抗数据生成方法
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
简介:DeepCustom是一种创新的方法,能够为深度学习模型生成定制化的对抗性数据,从而有效提升其在各种环境下的稳定性和准确性。该技术提供了一种高效的途径来增强机器学习系统的鲁棒性,确保它们能更好地应对未知和复杂的数据情况。
通过快速且可定制的对抗性数据生成来提升深度学习系统的鲁棒性。该代码包含七个子项目。我们研究了三个自定义损失函数(LRP、NCE 和可疑性),并在MNIST和CIFAR 数据集上进行了实验。由于DGN架构因使用的数据集不同而异,因此我们将项目分成多个子项目以避免复杂性的增加。对于每个自定义损失,有两个相应的子项目:RobustMNIST 和 RobustCIFAR。因此,我们有六个与基于CAE的对抗性数据生成相关的子项目。
还有一个用于使用IBM鲁棒工具箱生成FGSM和PGD攻击数据及进行FGSM、PGD对抗训练的独立子项目。这六个项目中的每一个都包含相同的Python脚本:
- CustomLosses.py:该文件包含了作为DGN模型损失函数的自定义损失函数。
- train.py:用于训练DGN模型并将生成的数据集整合进现有深度学习框架中进行鲁棒性测试和改进。
全部评论 (0)
还没有任何评论哟~


