pygip.models package¶
Subpackages¶
- pygip.models.attack package
- Subpackages
- Submodules
- pygip.models.attack.AdvMEA module
- pygip.models.attack.CEGA module
CEGAGCN_dropGcnNetaccuracy()attack0()aug_normalized_adjacency()aug_random_walk()calculate_entropy()compute_distance()compute_rank()convert_pyg_to_dgl()evaluate()find_short_dist()get_current_neighbors_1()get_current_neighbors_dense()get_entropy_contribute()get_max_info_entropy_node_set()get_max_nnd_node_dense()get_receptive_fields_dense()init_mask()load_data()load_data_from_grain()normalize()page_rank()parse_index_file()perturb_avg()perturb_features()quantile_selection()rank_centrality()rank_cluster()rank_density()rank_diversity()rank_entropy()rank_perturb()set_seed()sparse_mx_to_torch_sparse_tensor()update_sub_train_mask()
- pygip.models.attack.DataFreeMEA module
- pygip.models.attack.base module
BaseAttackBaseAttack.supported_api_typesBaseAttack.supported_datasetsBaseAttack.deviceBaseAttack.datasetBaseAttack.graph_datasetBaseAttack.graph_dataBaseAttack.num_nodesBaseAttack.num_featuresBaseAttack.num_classesBaseAttack.attack_node_fractionBaseAttack.model_pathBaseAttack._abc_implBaseAttack._check_dataset_compatibility()BaseAttack._load_model()BaseAttack._train_attack_model()BaseAttack._train_target_model()BaseAttack.attack()BaseAttack.supported_api_typesBaseAttack.supported_datasets
- Module contents
AdvMEACEGADFEATypeIDFEATypeIIDFEATypeIIIModelExtractionAttack0ModelExtractionAttack1ModelExtractionAttack2ModelExtractionAttack3ModelExtractionAttack4ModelExtractionAttack5RealisticAttackRealisticAttack._abc_implRealisticAttack._evaluate_and_update_metrics()RealisticAttack.add_potential_edges()RealisticAttack.attack()RealisticAttack.compute_cosine_similarity()RealisticAttack.generate_candidate_edges()RealisticAttack.simulate_target_model_queries()RealisticAttack.supported_api_typesRealisticAttack.supported_datasetsRealisticAttack.train_edge_predictor()RealisticAttack.train_surrogate_model()
- pygip.models.defense package
- Subpackages
- Submodules
- pygip.models.defense.BackdoorWM module
- pygip.models.defense.ImperceptibleWM module
ImperceptibleWMImperceptibleWM._abc_implImperceptibleWM._load_model()ImperceptibleWM._train_defense_model()ImperceptibleWM._train_surrogate_model()ImperceptibleWM._train_target_model()ImperceptibleWM.defend()ImperceptibleWM.evaluate_model()ImperceptibleWM.supported_api_typesImperceptibleWM.train_target_model()ImperceptibleWM.verify_watermark()
TriggerGeneratorbi_level_optimization()calculate_metrics()generate_trigger_graph()
- pygip.models.defense.ImperceptibleWM2 module
ImperceptibleWM2ImperceptibleWM2._abc_implImperceptibleWM2._calculate_embedding_loss()ImperceptibleWM2._calculate_generation_loss_integrated()ImperceptibleWM2._calculate_imperception_loss()ImperceptibleWM2._calculate_regulation_loss()ImperceptibleWM2._calculate_trigger_loss()ImperceptibleWM2._construct_backdoor_graph()ImperceptibleWM2._create_temp_trigger_graph()ImperceptibleWM2._evaluate_model_on_graph()ImperceptibleWM2._evaluate_with_metrics()ImperceptibleWM2._generate_trigger_graph()ImperceptibleWM2._inner_optimization()ImperceptibleWM2._select_poisoning_nodes()ImperceptibleWM2._train_defense_model()ImperceptibleWM2._train_target_model()ImperceptibleWM2.defend()ImperceptibleWM2.verify_ownership()
TriggerGenerator
- pygip.models.defense.RandomWM module
RandomWMRandomWM._abc_implRandomWM._evaluate_attack_on_watermark()RandomWM._evaluate_watermark()RandomWM._generate_watermark_graph()RandomWM._get_attack_class()RandomWM._test_on_watermark()RandomWM._train_defense_model()RandomWM._train_target_model()RandomWM.defend()RandomWM.evaluate_model()RandomWM.supported_api_typesRandomWM.train_target_model()RandomWM.verify_watermark()
- pygip.models.defense.SurviveWM module
SurviveWMSurviveWM._abc_implSurviveWM._load_model()SurviveWM._to_cpu()SurviveWM._train_watermarked_model()SurviveWM.combine_with_trigger()SurviveWM.compute_metrics()SurviveWM.defend()SurviveWM.evaluate_model()SurviveWM.generate_key_graph()SurviveWM.snn_loss()SurviveWM.supported_api_typesSurviveWM.train_target_model()SurviveWM.train_with_snnl()SurviveWM.verify_watermark()SurviveWM.verify_watermark_model()
- pygip.models.defense.SurviveWM2 module
- pygip.models.defense.base module
- Module contents
BackdoorWMImperceptibleWMImperceptibleWM._abc_implImperceptibleWM._load_model()ImperceptibleWM._train_defense_model()ImperceptibleWM._train_surrogate_model()ImperceptibleWM._train_target_model()ImperceptibleWM.defend()ImperceptibleWM.evaluate_model()ImperceptibleWM.supported_api_typesImperceptibleWM.train_target_model()ImperceptibleWM.verify_watermark()
ImperceptibleWM2ImperceptibleWM2._abc_implImperceptibleWM2._calculate_embedding_loss()ImperceptibleWM2._calculate_generation_loss_integrated()ImperceptibleWM2._calculate_imperception_loss()ImperceptibleWM2._calculate_regulation_loss()ImperceptibleWM2._calculate_trigger_loss()ImperceptibleWM2._construct_backdoor_graph()ImperceptibleWM2._create_temp_trigger_graph()ImperceptibleWM2._evaluate_model_on_graph()ImperceptibleWM2._evaluate_with_metrics()ImperceptibleWM2._generate_trigger_graph()ImperceptibleWM2._inner_optimization()ImperceptibleWM2._select_poisoning_nodes()ImperceptibleWM2._train_defense_model()ImperceptibleWM2._train_target_model()ImperceptibleWM2.defend()ImperceptibleWM2.verify_ownership()
RandomWMRandomWM._abc_implRandomWM._evaluate_attack_on_watermark()RandomWM._evaluate_watermark()RandomWM._generate_watermark_graph()RandomWM._get_attack_class()RandomWM._test_on_watermark()RandomWM._train_defense_model()RandomWM._train_target_model()RandomWM.defend()RandomWM.evaluate_model()RandomWM.supported_api_typesRandomWM.train_target_model()RandomWM.verify_watermark()
RevisitingSurviveWMSurviveWM._abc_implSurviveWM._load_model()SurviveWM._to_cpu()SurviveWM._train_watermarked_model()SurviveWM.combine_with_trigger()SurviveWM.compute_metrics()SurviveWM.defend()SurviveWM.evaluate_model()SurviveWM.generate_key_graph()SurviveWM.snn_loss()SurviveWM.supported_api_typesSurviveWM.train_target_model()SurviveWM.train_with_snnl()SurviveWM.verify_watermark()SurviveWM.verify_watermark_model()
SurviveWM2
- pygip.models.nn package