AISafety 对抗攻击的 Python 算法库

我要开发同款
匿名用户2021年09月30日
23阅读
开发技术Python
所属分类人工智能、机器学习/深度学习
授权协议MIT

作品详情

AISafety(启智重明)是一个用于对抗攻击全流程评测算法学习研究的Python库,其主要研究内容为集成对抗攻击和噪声攻击相关的攻击算法、评测算法、加固防御算法,可灵活测试数据集质量、算法训练、评估和部署等算法全生命周期各项指标。

特性集成大量攻击、评测、防御加固算法提供多种可解释性分析工具提供完备的扩展和使用接口 下载及使用STEP1.获取项目Python环境要求

Python3.6.5及以上克隆本项目并安装依赖:

gitclonehttps://git.openi.org.cn/OpenI/AISafety.gitcdAISafety/pipinstallrequirements.txtSTEP2.数据准备重明开源项目的Datasets/中提供了Cifar10和ILSVR2012-ImageNet数据集。用户可使用上述数据集,或按照完整API文档,进行数据集扩展。

用户需要使用所选数据集的训练集,执行模型训练过程。由于空间所限,本项目中统一不提供数据集对应的训练集,仅给出测试集以供测试。

STEP3.快速开始使用cdtest进入test目录。重明开源项目提供了几个示例的算法文件。如测试在FGSM攻击算法下,Resnet20模型的鲁棒性结果:

#使用接口文件默认参数pythontestimport.py#自定义参数调用pythontestimport.py--attack_method"FGSM"--evaluation_method"ALDp"--model_dir""上述命令均将测试ResNet20模型,在FGSM算法攻击下,ALDp指标的评测结果变化。并将结果存在AISafety/test/temp。

有关更多示例和用法(例如,如何扩展模型或算法,如何传入参数),请浏览完备API接口文档。

声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论