adarsh-aur's picture
Upload config.json with huggingface_hub
1936919 verified
{
"stage": 6,
"primary_model": "HeteroRGCN",
"library": "PyTorch Geometric (RGCNConv)",
"design": "holistic (zero embedding) \u2014 schema-agnostic, any fdim",
"master_node_types": [
"User",
"VM",
"IP",
"Role",
"CVE",
"Container",
"CloudAccount",
"Subnet",
"Bucket",
"Function",
"Cluster",
"Pod",
"Database",
"LoadBalancer",
"Gateway"
],
"master_edge_types": [
[
"IP",
"EXPLOITS",
"CVE"
],
[
"User",
"CROSS_CLOUD_ACCESS",
"User"
],
[
"IP",
"CONNECTS_TO",
"VM"
],
[
"User",
"EXPLOITS",
"CVE"
],
[
"User",
"ACCESS",
"VM"
],
[
"User",
"ASSUMES_ROLE",
"Role"
],
[
"VM",
"CONNECTS_TO",
"VM"
],
[
"User",
"RESTART_VM",
"VM"
],
[
"User",
"START_VM",
"VM"
],
[
"User",
"STOP_VM",
"VM"
],
[
"User",
"DEPLOYED_ON",
"CloudAccount"
],
[
"VM",
"HAS_VULN",
"CVE"
],
[
"VM",
"LATERAL_MOVEMENT",
"VM"
],
[
"Role",
"GRANTS_ACCESS",
"CloudAccount"
],
[
"Container",
"RUNS_ON",
"VM"
],
[
"Container",
"EXPLOITS",
"CVE"
],
[
"User",
"ACCESS",
"Bucket"
],
[
"Function",
"TRIGGERS",
"VM"
],
[
"User",
"ACCESS",
"Database"
],
[
"Cluster",
"CONTAINS",
"Pod"
]
],
"max_relations": 20,
"max_fdim": 1024,
"active_node_types": [
"User",
"VM",
"IP",
"Role",
"CVE"
],
"active_edge_types": [
[
"IP",
"EXPLOITS",
"CVE"
],
[
"User",
"CROSS_CLOUD_ACCESS",
"User"
],
[
"IP",
"CONNECTS_TO",
"VM"
],
[
"User",
"EXPLOITS",
"CVE"
],
[
"User",
"ACCESS",
"VM"
],
[
"User",
"ASSUMES_ROLE",
"Role"
],
[
"VM",
"CONNECTS_TO",
"VM"
],
[
"User",
"RESTART_VM",
"VM"
],
[
"User",
"START_VM",
"VM"
],
[
"User",
"STOP_VM",
"VM"
]
],
"active_supervised": [
"User",
"Role"
],
"n_nodes": {
"User": 11,
"VM": 9,
"IP": 4,
"Role": 3,
"CVE": 6
},
"hidden_dim": 256,
"out_dim": 128,
"num_layers": 3,
"adapter_rank": 16,
"dropout": 0.3,
"epochs": 80,
"lr": 0.001,
"weight_decay": 0.0001,
"seed": 42,
"ablation_results": [
{
"Model": "RGCN",
"Type": "PRIMARY",
"Params": "8,515,599",
"User_AUC": 0.5,
"User_F1": 0.0,
"User_AP": 0.5,
"Role_AUC": 0.5,
"Role_F1": 0.0,
"Role_AP": 0.5
},
{
"Model": "GCN",
"Type": "BASELINE",
"Params": "5,219,855",
"User_AUC": 0.5,
"User_F1": 0.0,
"User_AP": 0.5,
"Role_AUC": 0.5,
"Role_F1": 0.0,
"Role_AP": 0.5
},
{
"Model": "GAT",
"Type": "BASELINE",
"Params": "5,320,207",
"User_AUC": 0.5,
"User_F1": 0.0,
"User_AP": 0.5,
"Role_AUC": 0.5,
"Role_F1": 0.0,
"Role_AP": 0.5
},
{
"Model": "SAGE",
"Type": "BASELINE",
"Params": "5,383,695",
"User_AUC": 0.5,
"User_F1": 0.0,
"User_AP": 0.5,
"Role_AUC": 0.5,
"Role_F1": 0.0,
"Role_AP": 0.5
}
],
"test_report": {
"User": {
"auc": 0.5,
"f1": 0.0,
"ap": 0.5
},
"Role": {
"auc": 0.5,
"f1": 0.0,
"ap": 0.5
}
}
}