pygip.models.defense.atom package

Submodules

pygip.models.defense.atom.ATOM module

class pygip.models.defense.atom.ATOM.ATOM(dataset, attack_node_fraction=0)

Bases: BaseDefense

_abc_impl = <_abc_data object>

_convert_to_networkx()

Convert graph data to NetworkX format, compatible with different graph data structures

_create_default_masks()

Create default train/test masks for datasets without predefined masks

static _evaluate_downstream_task(model, data, test_mask, labels)

Evaluate model performance on downstream task

_load_data_and_model(batch_size=16, seed=0, lamb=0)

Load data and model following BackdoorWM format Use data from self.dataset directly instead of reloading

static _train_gcn_model(model, data, train_mask, labels)

Train GCN model for PyG format

defend()

Execute ATOM defense, following BackdoorWM’s defend method structure

supported_api_types = {'pyg'}

supported_datasets = {'CS', 'CiteSeer', 'Computers', 'Cora', 'Photo', 'Physics', 'PubMed'}

class pygip.models.defense.atom.ATOM.FusionGRU(*args: Any, **kwargs: Any)

Bases: Module

forward(h_it, h_it_m1, hidden_state)

process_sequence(inputs, hidden_state=None)

class pygip.models.defense.atom.ATOM.GCN(*args: Any, **kwargs: Any)

Bases: Module

forward(x, edge_index)

class pygip.models.defense.atom.ATOM.Memory

Bases: object

clear()

store(custom_states, action, log_prob, reward, done, entropy, probs=None, masks=None)

class pygip.models.defense.atom.ATOM.PPOAgent(*args: Any, **kwargs: Any)

Bases: Module

select_action(state)

update(memory)

class pygip.models.defense.atom.ATOM.PolicyNetwork(*args: Any, **kwargs: Any)

Bases: Module

forward(state)

class pygip.models.defense.atom.ATOM.SequencesDataset(*args: Any, **kwargs: Any)

Bases: Dataset

class pygip.models.defense.atom.ATOM.StateTransformMLP(*args: Any, **kwargs: Any)

Bases: Module

forward(prob_factor)

class pygip.models.defense.atom.ATOM.TargetGCN(trained_model, data)

Bases: object

get_embedding()

predict(query_indices)

pygip.models.defense.atom.ATOM.average_pooling_with_neighbors(model, data, node_idx)

pygip.models.defense.atom.ATOM.average_pooling_with_neighbors_batch(model, data, node_indices)

pygip.models.defense.atom.ATOM.build_loaders(csv_path='attack_CiteSeer.csv', batch_size=16, drop_last=True, seed=42)

pygip.models.defense.atom.ATOM.collate_fn_no_pad(batch)

pygip.models.defense.atom.ATOM.compute_embedding_batch(target_model, data, k_core_values_graph, max_k_core, node_indices, lamb=1.0)

pygip.models.defense.atom.ATOM.compute_returns_and_advantages(memory, gamma=0.99, lam=0.95)

pygip.models.defense.atom.ATOM.custom_reward_function(predicted, label, predicted_distribution=None)

pygip.models.defense.atom.ATOM.get_node_embedding(model, data, node_idx)

pygip.models.defense.atom.ATOM.get_one_hop_neighbors(data, node_idx)

pygip.models.defense.atom.ATOM.k_core_decomposition(graph)

pygip.models.defense.atom.ATOM.load_data_and_model(csv_path, batch_size, seed, data_path, lamb)

pygip.models.defense.atom.ATOM.precompute_all_node_embeddings(target_model, data, k_core_values_graph, max_k_core, lamb=1.0)

pygip.models.defense.atom.ATOM.precompute_simple_embeddings(target_model, data)

pygip.models.defense.atom.ATOM.preprocess_sequences(df)

pygip.models.defense.atom.ATOM.set_seed(seed)

pygip.models.defense.atom.ATOM.simple_embedding_batch(target_model, data, node_indices)

pygip.models.defense.atom.ATOM.split_and_adjust(dataset_sequences, seed)

pygip.models.defense.atom.ATOM.test_model(agent, gru, mlp_transform, test_loader, target_model, data, all_embeddings, hidden_size, device)

pygip.models.defense.atom.ATOM.train_gcn(model, data, optimizer, criterion, epochs=200, verbose=True)

Module contents