Hugging Face's logo

adarsh-aur
/
rgcn-security-zero-embedding

graph-neural-network
cybersecurity
rgcn
pytorch-geometric
multi-cloud
Model card Files
xet
 Community
adarsh-aur commited on
Commit
1936919
·
verified ·
1 Parent(s): c3e864e

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +264 -0
config.json ADDED
@@ -0,0 +1,264 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "stage": 6,
3
+ "primary_model": "HeteroRGCN",
4
+ "library": "PyTorch Geometric (RGCNConv)",
5
+ "design": "holistic (zero embedding) \u2014 schema-agnostic, any fdim",
6
+ "master_node_types": [
7
+ "User",
8
+ "VM",
9
+ "IP",
10
+ "Role",
11
+ "CVE",
12
+ "Container",
13
+ "CloudAccount",
14
+ "Subnet",
15
+ "Bucket",
16
+ "Function",
17
+ "Cluster",
18
+ "Pod",
19
+ "Database",
20
+ "LoadBalancer",
21
+ "Gateway"
22
+ ],
23
+ "master_edge_types": [
24
+ [
25
+ "IP",
26
+ "EXPLOITS",
27
+ "CVE"
28
+ ],
29
+ [
30
+ "User",
31
+ "CROSS_CLOUD_ACCESS",
32
+ "User"
33
+ ],
34
+ [
35
+ "IP",
36
+ "CONNECTS_TO",
37
+ "VM"
38
+ ],
39
+ [
40
+ "User",
41
+ "EXPLOITS",
42
+ "CVE"
43
+ ],
44
+ [
45
+ "User",
46
+ "ACCESS",
47
+ "VM"
48
+ ],
49
+ [
50
+ "User",
51
+ "ASSUMES_ROLE",
52
+ "Role"
53
+ ],
54
+ [
55
+ "VM",
56
+ "CONNECTS_TO",
57
+ "VM"
58
+ ],
59
+ [
60
+ "User",
61
+ "RESTART_VM",
62
+ "VM"
63
+ ],
64
+ [
65
+ "User",
66
+ "START_VM",
67
+ "VM"
68
+ ],
69
+ [
70
+ "User",
71
+ "STOP_VM",
72
+ "VM"
73
+ ],
74
+ [
75
+ "User",
76
+ "DEPLOYED_ON",
77
+ "CloudAccount"
78
+ ],
79
+ [
80
+ "VM",
81
+ "HAS_VULN",
82
+ "CVE"
83
+ ],
84
+ [
85
+ "VM",
86
+ "LATERAL_MOVEMENT",
87
+ "VM"
88
+ ],
89
+ [
90
+ "Role",
91
+ "GRANTS_ACCESS",
92
+ "CloudAccount"
93
+ ],
94
+ [
95
+ "Container",
96
+ "RUNS_ON",
97
+ "VM"
98
+ ],
99
+ [
100
+ "Container",
101
+ "EXPLOITS",
102
+ "CVE"
103
+ ],
104
+ [
105
+ "User",
106
+ "ACCESS",
107
+ "Bucket"
108
+ ],
109
+ [
110
+ "Function",
111
+ "TRIGGERS",
112
+ "VM"
113
+ ],
114
+ [
115
+ "User",
116
+ "ACCESS",
117
+ "Database"
118
+ ],
119
+ [
120
+ "Cluster",
121
+ "CONTAINS",
122
+ "Pod"
123
+ ]
124
+ ],
125
+ "max_relations": 20,
126
+ "max_fdim": 1024,
127
+ "active_node_types": [
128
+ "User",
129
+ "VM",
130
+ "IP",
131
+ "Role",
132
+ "CVE"
133
+ ],
134
+ "active_edge_types": [
135
+ [
136
+ "IP",
137
+ "EXPLOITS",
138
+ "CVE"
139
+ ],
140
+ [
141
+ "User",
142
+ "CROSS_CLOUD_ACCESS",
143
+ "User"
144
+ ],
145
+ [
146
+ "IP",
147
+ "CONNECTS_TO",
148
+ "VM"
149
+ ],
150
+ [
151
+ "User",
152
+ "EXPLOITS",
153
+ "CVE"
154
+ ],
155
+ [
156
+ "User",
157
+ "ACCESS",
158
+ "VM"
159
+ ],
160
+ [
161
+ "User",
162
+ "ASSUMES_ROLE",
163
+ "Role"
164
+ ],
165
+ [
166
+ "VM",
167
+ "CONNECTS_TO",
168
+ "VM"
169
+ ],
170
+ [
171
+ "User",
172
+ "RESTART_VM",
173
+ "VM"
174
+ ],
175
+ [
176
+ "User",
177
+ "START_VM",
178
+ "VM"
179
+ ],
180
+ [
181
+ "User",
182
+ "STOP_VM",
183
+ "VM"
184
+ ]
185
+ ],
186
+ "active_supervised": [
187
+ "User",
188
+ "Role"
189
+ ],
190
+ "n_nodes": {
191
+ "User": 11,
192
+ "VM": 9,
193
+ "IP": 4,
194
+ "Role": 3,
195
+ "CVE": 6
196
+ },
197
+ "hidden_dim": 256,
198
+ "out_dim": 128,
199
+ "num_layers": 3,
200
+ "adapter_rank": 16,
201
+ "dropout": 0.3,
202
+ "epochs": 80,
203
+ "lr": 0.001,
204
+ "weight_decay": 0.0001,
205
+ "seed": 42,
206
+ "ablation_results": [
207
+ {
208
+ "Model": "RGCN",
209
+ "Type": "PRIMARY",
210
+ "Params": "8,515,599",
211
+ "User_AUC": 0.5,
212
+ "User_F1": 0.0,
213
+ "User_AP": 0.5,
214
+ "Role_AUC": 0.5,
215
+ "Role_F1": 0.0,
216
+ "Role_AP": 0.5
217
+ },
218
+ {
219
+ "Model": "GCN",
220
+ "Type": "BASELINE",
221
+ "Params": "5,219,855",
222
+ "User_AUC": 0.5,
223
+ "User_F1": 0.0,
224
+ "User_AP": 0.5,
225
+ "Role_AUC": 0.5,
226
+ "Role_F1": 0.0,
227
+ "Role_AP": 0.5
228
+ },
229
+ {
230
+ "Model": "GAT",
231
+ "Type": "BASELINE",
232
+ "Params": "5,320,207",
233
+ "User_AUC": 0.5,
234
+ "User_F1": 0.0,
235
+ "User_AP": 0.5,
236
+ "Role_AUC": 0.5,
237
+ "Role_F1": 0.0,
238
+ "Role_AP": 0.5
239
+ },
240
+ {
241
+ "Model": "SAGE",
242
+ "Type": "BASELINE",
243
+ "Params": "5,383,695",
244
+ "User_AUC": 0.5,
245
+ "User_F1": 0.0,
246
+ "User_AP": 0.5,
247
+ "Role_AUC": 0.5,
248
+ "Role_F1": 0.0,
249
+ "Role_AP": 0.5
250
+ }
251
+ ],
252
+ "test_report": {
253
+ "User": {
254
+ "auc": 0.5,
255
+ "f1": 0.0,
256
+ "ap": 0.5
257
+ },
258
+ "Role": {
259
+ "auc": 0.5,
260
+ "f1": 0.0,
261
+ "ap": 0.5
262
+ }
263
+ }
264
+ }