👋TorchEI, 发音为*/ˈtôrCHər/*(like torture), 是Pytorch Error Injection的缩写, 一个围绕DNN Reliability 研究的高速工具箱. TorchEI 使您能够快速简单地将错误注入 DNN，收集您需要的信息并强化您的 DNN。
- 完善的类型提示和文档支持
- 包含来自 DNN 可靠性论文的方法
- 高度定制化
在这里，我们将向您展示一个简单的示例，或者您可以尝试 interactive demo 和online editor
你可以使用 pip3 install torchei 安装或 下载
初始化故障模型
import torch
from torchvision import models
import torchei
model = models.resnet18(pretrained=True)
data = torch.load('data/ilsvrc_valid8.pt')
fault_model = torchei.fault_model(model,data)使用emat方法计算可靠性
fault_model.emat_attack(10,1e-3)使用SERN方法计算可靠性
fault_model.sern_calc(output_class=1000)使用ODR方法加固DNN
fault_model.outlierDR_protection()
fault_model.emat_attack(10,1e-3)
如果您发现🧐任何错误或有🖐️任何建议，请告诉我们。
这个 repo 欢迎所有想要一起维护的人。
You can helps us with follow things:
- PR your implemented methods in your or others' papers
- Complete our project
- Translate our docs to your language
- Other
我们希望将 TorchEI 构建为 DNN 可靠性方面的最佳工具箱，用于位翻转、对抗性攻击等。
Our paper is under delivering.
MIT License. Copyright:copyright:2022/5/23-present, Hao Zheng.