Function test_prbcd_modification_attack

examples/GRB/test_attack_defense.py:268–304

()

def test_prbcd_modification_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("PRBCD modification attack...")
    device = "cpu"
    epsilon = 0.3
    n_epoch = 5
    n_mod_ratio = 0.01
    n_node_mod = int(graph.y.shape[0] * n_mod_ratio)
    n_edge_mod = int(graph.to_scipy_csr()[test_mask.cpu()].getnnz() * n_mod_ratio)
    feat_lim_min = 0.0
    feat_lim_max = 1.0
    early_stop_patience = 2
    early_stop_epsilon = 1e-3
    attack = PRBCD(epsilon,
                n_epoch,
                n_node_mod,
                n_edge_mod,
                feat_lim_min,
                feat_lim_max,
                early_stop=True,
                early_stop_patience=early_stop_patience,
                early_stop_epsilon=early_stop_epsilon,
                device=device)
    graph_attack = attack.attack(model_sur, graph)
    print(graph_attack)
    test_score = evaluate(model_sur, 
                        graph_attack,
                        mask=test_mask,
                        device=device)
    print("After attack, test score of surrogate model: {:.4f}".format(test_score))
    test_score = evaluate(model_target, 
                        graph_attack,
                        mask=test_mask,
                        device=device)
    print("After attack, test score of target model: {:.4f}".format(test_score))


def test_fgsm_injection_attack():