leftyer/mindarmour
1
0
Fork 0
mirror of https://gitee.com/mindspore/mindarmour.git synced 2025-12-06 11:59:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

!538 optimize genetic attack and update version number

Browse Source 
Merge pull request !538 from jxl/r1.9
This commit is contained in:
i-robot
2023-08-15 00:46:21 +00:00
committed by Gitee
parent a76dc69015 3b3fce5e8c
commit d0c1871406
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
mindarmour/adv_robustness/attacks/black/genetic_attack.py
View File

@@ -283,6 +283,7 @@ class GeneticAttack(Attack):
self._bounds[0], self._bounds[1])
pop_preds = self._model.predict(cur_pop)
pop_preds = np.exp(pop_preds) / np.sum(np.exp(pop_preds), axis=1, keepdims=True)
query_times += cur_pop.shape[0]
all_preds = np.argmax(pop_preds, axis=1)
if self._targeted:
@@ -294,9 +295,9 @@ class GeneticAttack(Attack):
target_preds = pop_preds[:, label_i]
others_preds_sum = np.sum(pop_preds, axis=1) - target_preds
if self._targeted:
fit_vals = target_preds - others_preds_sum
fit_vals = np.log(target_preds + 1e-30) - np.log(others_preds_sum + 1e-30)
else:
fit_vals = others_preds_sum - target_preds
fit_vals = np.log(others_preds_sum + 1e-30) - np.log(target_preds + 1e-30)
if is_success:
LOGGER.debug(TAG, 'successfully find one adversarial sample '

2
setup.py
View File

@@ -24,7 +24,7 @@ from setuptools import setup
from setuptools.command.egg_info import egg_info
from setuptools.command.build_py import build_py
version = '1.9.0'
version = '1.9.1'
cur_dir = os.path.dirname(os.path.realpath(__file__))
pkg_dir = os.path.join(cur_dir, 'build')