Function test_fgsm_injection_attack

examples/GRB/test_attack_defense.py:307–335

()

def test_fgsm_injection_attack():
    graph, dataset, test_mask, device, device_ids = init_dataset()
    model_sur = init_surrogate_model(graph, dataset, test_mask, device, device_ids)
    model_target = init_target_model(graph, dataset, test_mask, device, device_ids)
    print("FGSM injection attack...")
    attack = FGSM(epsilon=0.01,
                n_epoch=5,
                n_inject_max=10,
                n_edge_max=20,
                feat_lim_min=-1,
                feat_lim_max=1,
                early_stop=True,
                early_stop_patience=2,
                early_stop_epsilon=1e-4,
                device=device)
    graph_attack = attack.attack(model=model_sur,
                                graph=graph,
                                adj_norm_func=GCNAdjNorm)
    print(graph_attack)
    test_score_sur = evaluate(model_sur,
                            graph_attack,
                            mask=test_mask,
                            device=device)
    print("Test score after attack for surrogate model: {:.4f}.".format(test_score_sur))
    test_score_target_attack = evaluate(model_target,
                                        graph_attack,
                                        mask=test_mask,
                                        device=device)
    print("Test score after attack for target model: {:.4f}.".format(test_score_target_attack))


def test_pgd_injection_attack():