Image Classification
The below input parameters are for different attack types. To start working with the APIs view the Image Classification.
- Data: The processed data, ready to be passed to the model for prediction, should be saved in a folder.
- Label: A CSV file should be created with two columns: "image" and "label." The first column should contain the image name, and the second column should contain the label. The label should be in integer format. Check sample label file attached.
- Model: The model should be saved in either .h5 or TensorFlow format with full architecture. Full architecture is needed when loading the model to the platofrm for assessment either in encrypted or unencrypted. This can be ignored when model is hosted as an API.
Note:
- All files uploaded should be in zipped format. The above files are sample data for the MNIST use case.
- Prerequisite: Only 2-5 % of data is needed. The data should be representative and balanced across all classes.
- For poisoning, check poisoning section to get sample data, label and models.
The below table parameters are common for all attact types such as Extraction, Evasion, and Poisoning.
To see the additional parameter specific for each attact type, such as Extraction and Evasion, refer to the below sections.
Parameter | Data type | Description | Remark |
---|---|---|---|
model_Id | String | Model_id received during model registration. We need to provide this model ID in query parameter in URL. | You have to do model registration only once for a model to perform model analysis. This will help you track the no of api call made, and it's success metric. |
Request Body (Json format) | | | |
normalize_data | String | Model trained on Normalized data. | if model is trained on normalized data, then set this parameter as "yes" else "no". |
input_dimensions | String | Provide input dimension of the image | the parameter should be string in the format "(height, weight, channel)" For example 28*28*1 for MNIST |
number_of_classes | String | Number of prediction classes. | the parameter should be string. Example MNIST : 10 (Range >0 & <=200) |
model_framework | String | Original model is built with tensorflow framework. | curretly supported framework are: tensorflow, scikit-learn, keras. (Option:[tensorflow]) |
Data Poisoning sample Artifact: CleanData , Label, Model, UniversalDataset, UniversalLabel
Model Poisoning Sample Artifact: CleanModel1, CleanModel2, Data, Label, Model_to_test
To access all sample artifacts, please visit Artifacts.
- For specific artifact details, refer
Note: For Image classification, supported attack types are - Extraction, Evasion and Poisoning.