pratimassaravanan commited on
Commit
f914ffa
·
verified ·
1 Parent(s): bd8699d

RL v3 training + eval results

Browse files
Files changed (1) hide show
  1. rl_v3_training_results.json +507 -0
rl_v3_training_results.json ADDED
@@ -0,0 +1,507 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "method": "REINFORCE v3 (fresh LoRA, no SFT, heuristic override, reward shaping)",
3
+ "base_model": "Qwen/Qwen3-4B",
4
+ "rl_config": {
5
+ "episodes": 30,
6
+ "max_steps": 50,
7
+ "lr": 5e-05,
8
+ "temperature": 1.0,
9
+ "grad_accum": 4,
10
+ "grad_clip": 1.0,
11
+ "lora_r": 16,
12
+ "lora_alpha": 32
13
+ },
14
+ "gpu": "NVIDIA L40S",
15
+ "vram_gb": 47.7,
16
+ "torch_version": "2.8.0+cu128",
17
+ "rl_episodes": [
18
+ {
19
+ "episode": 0,
20
+ "task": "easy_bench",
21
+ "total_reward": 15.1892,
22
+ "enrolled": 9,
23
+ "target": 80,
24
+ "steps": 50,
25
+ "unique_action_types": 3,
26
+ "positive_rl_steps": 50,
27
+ "action_dist": {
28
+ "screen_patient": 11,
29
+ "recontact": 24,
30
+ "allocate_to_site": 15
31
+ }
32
+ },
33
+ {
34
+ "episode": 1,
35
+ "task": "medium_bench",
36
+ "total_reward": 11.5018,
37
+ "enrolled": 2,
38
+ "target": 120,
39
+ "steps": 50,
40
+ "unique_action_types": 3,
41
+ "positive_rl_steps": 50,
42
+ "action_dist": {
43
+ "screen_patient": 19,
44
+ "recontact": 17,
45
+ "allocate_to_site": 14
46
+ }
47
+ },
48
+ {
49
+ "episode": 2,
50
+ "task": "hard_bench",
51
+ "total_reward": 10.227,
52
+ "enrolled": 0,
53
+ "target": 150,
54
+ "steps": 50,
55
+ "unique_action_types": 3,
56
+ "positive_rl_steps": 50,
57
+ "action_dist": {
58
+ "screen_patient": 9,
59
+ "recontact": 37,
60
+ "allocate_to_site": 4
61
+ }
62
+ },
63
+ {
64
+ "episode": 3,
65
+ "task": "easy_bench",
66
+ "total_reward": 14.3473,
67
+ "enrolled": 8,
68
+ "target": 80,
69
+ "steps": 50,
70
+ "unique_action_types": 3,
71
+ "positive_rl_steps": 50,
72
+ "action_dist": {
73
+ "screen_patient": 10,
74
+ "recontact": 26,
75
+ "allocate_to_site": 14
76
+ }
77
+ },
78
+ {
79
+ "episode": 4,
80
+ "task": "medium_bench",
81
+ "total_reward": 11.5018,
82
+ "enrolled": 2,
83
+ "target": 120,
84
+ "steps": 50,
85
+ "unique_action_types": 3,
86
+ "positive_rl_steps": 50,
87
+ "action_dist": {
88
+ "screen_patient": 19,
89
+ "recontact": 17,
90
+ "allocate_to_site": 14
91
+ }
92
+ },
93
+ {
94
+ "episode": 5,
95
+ "task": "hard_bench",
96
+ "total_reward": 10.227,
97
+ "enrolled": 0,
98
+ "target": 150,
99
+ "steps": 50,
100
+ "unique_action_types": 3,
101
+ "positive_rl_steps": 50,
102
+ "action_dist": {
103
+ "screen_patient": 9,
104
+ "recontact": 37,
105
+ "allocate_to_site": 4
106
+ }
107
+ },
108
+ {
109
+ "episode": 6,
110
+ "task": "easy_bench",
111
+ "total_reward": 14.3473,
112
+ "enrolled": 8,
113
+ "target": 80,
114
+ "steps": 50,
115
+ "unique_action_types": 3,
116
+ "positive_rl_steps": 50,
117
+ "action_dist": {
118
+ "screen_patient": 10,
119
+ "recontact": 26,
120
+ "allocate_to_site": 14
121
+ }
122
+ },
123
+ {
124
+ "episode": 7,
125
+ "task": "medium_bench",
126
+ "total_reward": 11.5018,
127
+ "enrolled": 2,
128
+ "target": 120,
129
+ "steps": 50,
130
+ "unique_action_types": 3,
131
+ "positive_rl_steps": 50,
132
+ "action_dist": {
133
+ "screen_patient": 19,
134
+ "recontact": 17,
135
+ "allocate_to_site": 14
136
+ }
137
+ },
138
+ {
139
+ "episode": 8,
140
+ "task": "hard_bench",
141
+ "total_reward": 10.227,
142
+ "enrolled": 0,
143
+ "target": 150,
144
+ "steps": 50,
145
+ "unique_action_types": 3,
146
+ "positive_rl_steps": 50,
147
+ "action_dist": {
148
+ "screen_patient": 9,
149
+ "recontact": 37,
150
+ "allocate_to_site": 4
151
+ }
152
+ },
153
+ {
154
+ "episode": 9,
155
+ "task": "easy_bench",
156
+ "total_reward": 14.3473,
157
+ "enrolled": 8,
158
+ "target": 80,
159
+ "steps": 50,
160
+ "unique_action_types": 3,
161
+ "positive_rl_steps": 50,
162
+ "action_dist": {
163
+ "screen_patient": 10,
164
+ "recontact": 26,
165
+ "allocate_to_site": 14
166
+ }
167
+ },
168
+ {
169
+ "episode": 10,
170
+ "task": "medium_bench",
171
+ "total_reward": 11.5018,
172
+ "enrolled": 2,
173
+ "target": 120,
174
+ "steps": 50,
175
+ "unique_action_types": 3,
176
+ "positive_rl_steps": 50,
177
+ "action_dist": {
178
+ "screen_patient": 19,
179
+ "recontact": 17,
180
+ "allocate_to_site": 14
181
+ }
182
+ },
183
+ {
184
+ "episode": 11,
185
+ "task": "hard_bench",
186
+ "total_reward": 10.227,
187
+ "enrolled": 0,
188
+ "target": 150,
189
+ "steps": 50,
190
+ "unique_action_types": 3,
191
+ "positive_rl_steps": 50,
192
+ "action_dist": {
193
+ "screen_patient": 9,
194
+ "recontact": 37,
195
+ "allocate_to_site": 4
196
+ }
197
+ },
198
+ {
199
+ "episode": 12,
200
+ "task": "easy_bench",
201
+ "total_reward": 14.3473,
202
+ "enrolled": 8,
203
+ "target": 80,
204
+ "steps": 50,
205
+ "unique_action_types": 3,
206
+ "positive_rl_steps": 50,
207
+ "action_dist": {
208
+ "screen_patient": 10,
209
+ "recontact": 26,
210
+ "allocate_to_site": 14
211
+ }
212
+ },
213
+ {
214
+ "episode": 13,
215
+ "task": "medium_bench",
216
+ "total_reward": 11.5018,
217
+ "enrolled": 2,
218
+ "target": 120,
219
+ "steps": 50,
220
+ "unique_action_types": 3,
221
+ "positive_rl_steps": 50,
222
+ "action_dist": {
223
+ "screen_patient": 19,
224
+ "recontact": 17,
225
+ "allocate_to_site": 14
226
+ }
227
+ },
228
+ {
229
+ "episode": 14,
230
+ "task": "hard_bench",
231
+ "total_reward": 10.227,
232
+ "enrolled": 0,
233
+ "target": 150,
234
+ "steps": 50,
235
+ "unique_action_types": 3,
236
+ "positive_rl_steps": 50,
237
+ "action_dist": {
238
+ "screen_patient": 9,
239
+ "recontact": 37,
240
+ "allocate_to_site": 4
241
+ }
242
+ },
243
+ {
244
+ "episode": 15,
245
+ "task": "easy_bench",
246
+ "total_reward": 14.3473,
247
+ "enrolled": 8,
248
+ "target": 80,
249
+ "steps": 50,
250
+ "unique_action_types": 3,
251
+ "positive_rl_steps": 50,
252
+ "action_dist": {
253
+ "screen_patient": 10,
254
+ "recontact": 26,
255
+ "allocate_to_site": 14
256
+ }
257
+ },
258
+ {
259
+ "episode": 16,
260
+ "task": "medium_bench",
261
+ "total_reward": 11.5018,
262
+ "enrolled": 2,
263
+ "target": 120,
264
+ "steps": 50,
265
+ "unique_action_types": 3,
266
+ "positive_rl_steps": 50,
267
+ "action_dist": {
268
+ "screen_patient": 19,
269
+ "recontact": 17,
270
+ "allocate_to_site": 14
271
+ }
272
+ },
273
+ {
274
+ "episode": 17,
275
+ "task": "hard_bench",
276
+ "total_reward": 10.227,
277
+ "enrolled": 0,
278
+ "target": 150,
279
+ "steps": 50,
280
+ "unique_action_types": 3,
281
+ "positive_rl_steps": 50,
282
+ "action_dist": {
283
+ "screen_patient": 9,
284
+ "recontact": 37,
285
+ "allocate_to_site": 4
286
+ }
287
+ },
288
+ {
289
+ "episode": 18,
290
+ "task": "easy_bench",
291
+ "total_reward": 14.3473,
292
+ "enrolled": 8,
293
+ "target": 80,
294
+ "steps": 50,
295
+ "unique_action_types": 3,
296
+ "positive_rl_steps": 50,
297
+ "action_dist": {
298
+ "screen_patient": 10,
299
+ "recontact": 26,
300
+ "allocate_to_site": 14
301
+ }
302
+ },
303
+ {
304
+ "episode": 19,
305
+ "task": "medium_bench",
306
+ "total_reward": 11.5018,
307
+ "enrolled": 2,
308
+ "target": 120,
309
+ "steps": 50,
310
+ "unique_action_types": 3,
311
+ "positive_rl_steps": 50,
312
+ "action_dist": {
313
+ "screen_patient": 19,
314
+ "recontact": 17,
315
+ "allocate_to_site": 14
316
+ }
317
+ },
318
+ {
319
+ "episode": 20,
320
+ "task": "hard_bench",
321
+ "total_reward": 10.227,
322
+ "enrolled": 0,
323
+ "target": 150,
324
+ "steps": 50,
325
+ "unique_action_types": 3,
326
+ "positive_rl_steps": 50,
327
+ "action_dist": {
328
+ "screen_patient": 9,
329
+ "recontact": 37,
330
+ "allocate_to_site": 4
331
+ }
332
+ },
333
+ {
334
+ "episode": 21,
335
+ "task": "easy_bench",
336
+ "total_reward": 14.3473,
337
+ "enrolled": 8,
338
+ "target": 80,
339
+ "steps": 50,
340
+ "unique_action_types": 3,
341
+ "positive_rl_steps": 50,
342
+ "action_dist": {
343
+ "screen_patient": 10,
344
+ "recontact": 26,
345
+ "allocate_to_site": 14
346
+ }
347
+ },
348
+ {
349
+ "episode": 22,
350
+ "task": "medium_bench",
351
+ "total_reward": 11.5018,
352
+ "enrolled": 2,
353
+ "target": 120,
354
+ "steps": 50,
355
+ "unique_action_types": 3,
356
+ "positive_rl_steps": 50,
357
+ "action_dist": {
358
+ "screen_patient": 19,
359
+ "recontact": 17,
360
+ "allocate_to_site": 14
361
+ }
362
+ },
363
+ {
364
+ "episode": 23,
365
+ "task": "hard_bench",
366
+ "total_reward": 10.227,
367
+ "enrolled": 0,
368
+ "target": 150,
369
+ "steps": 50,
370
+ "unique_action_types": 3,
371
+ "positive_rl_steps": 50,
372
+ "action_dist": {
373
+ "screen_patient": 9,
374
+ "recontact": 37,
375
+ "allocate_to_site": 4
376
+ }
377
+ },
378
+ {
379
+ "episode": 24,
380
+ "task": "easy_bench",
381
+ "total_reward": 14.3473,
382
+ "enrolled": 8,
383
+ "target": 80,
384
+ "steps": 50,
385
+ "unique_action_types": 3,
386
+ "positive_rl_steps": 50,
387
+ "action_dist": {
388
+ "screen_patient": 10,
389
+ "recontact": 26,
390
+ "allocate_to_site": 14
391
+ }
392
+ },
393
+ {
394
+ "episode": 25,
395
+ "task": "medium_bench",
396
+ "total_reward": 11.5018,
397
+ "enrolled": 2,
398
+ "target": 120,
399
+ "steps": 50,
400
+ "unique_action_types": 3,
401
+ "positive_rl_steps": 50,
402
+ "action_dist": {
403
+ "screen_patient": 19,
404
+ "recontact": 17,
405
+ "allocate_to_site": 14
406
+ }
407
+ },
408
+ {
409
+ "episode": 26,
410
+ "task": "hard_bench",
411
+ "total_reward": 10.227,
412
+ "enrolled": 0,
413
+ "target": 150,
414
+ "steps": 50,
415
+ "unique_action_types": 3,
416
+ "positive_rl_steps": 50,
417
+ "action_dist": {
418
+ "screen_patient": 9,
419
+ "recontact": 37,
420
+ "allocate_to_site": 4
421
+ }
422
+ },
423
+ {
424
+ "episode": 27,
425
+ "task": "easy_bench",
426
+ "total_reward": 14.3473,
427
+ "enrolled": 8,
428
+ "target": 80,
429
+ "steps": 50,
430
+ "unique_action_types": 3,
431
+ "positive_rl_steps": 50,
432
+ "action_dist": {
433
+ "screen_patient": 10,
434
+ "recontact": 26,
435
+ "allocate_to_site": 14
436
+ }
437
+ },
438
+ {
439
+ "episode": 28,
440
+ "task": "medium_bench",
441
+ "total_reward": 11.5018,
442
+ "enrolled": 2,
443
+ "target": 120,
444
+ "steps": 50,
445
+ "unique_action_types": 3,
446
+ "positive_rl_steps": 50,
447
+ "action_dist": {
448
+ "screen_patient": 19,
449
+ "recontact": 17,
450
+ "allocate_to_site": 14
451
+ }
452
+ },
453
+ {
454
+ "episode": 29,
455
+ "task": "hard_bench",
456
+ "total_reward": 10.227,
457
+ "enrolled": 0,
458
+ "target": 150,
459
+ "steps": 50,
460
+ "unique_action_types": 3,
461
+ "positive_rl_steps": 50,
462
+ "action_dist": {
463
+ "screen_patient": 9,
464
+ "recontact": 37,
465
+ "allocate_to_site": 4
466
+ }
467
+ }
468
+ ],
469
+ "eval_results": {
470
+ "easy_bench": {
471
+ "steps": 50,
472
+ "total_reward": 14.3473,
473
+ "enrolled": 8,
474
+ "target": 80,
475
+ "json_parse_rate": 1.0,
476
+ "action_dist": {
477
+ "screen_patient": 10,
478
+ "allocate_to_site": 14,
479
+ "recontact": 26
480
+ }
481
+ },
482
+ "medium_bench": {
483
+ "steps": 50,
484
+ "total_reward": 11.5018,
485
+ "enrolled": 2,
486
+ "target": 120,
487
+ "json_parse_rate": 1.0,
488
+ "action_dist": {
489
+ "screen_patient": 19,
490
+ "allocate_to_site": 14,
491
+ "recontact": 17
492
+ }
493
+ },
494
+ "hard_bench": {
495
+ "steps": 50,
496
+ "total_reward": 10.227,
497
+ "enrolled": 0,
498
+ "target": 150,
499
+ "json_parse_rate": 1.0,
500
+ "action_dist": {
501
+ "screen_patient": 9,
502
+ "recontact": 37,
503
+ "allocate_to_site": 4
504
+ }
505
+ }
506
+ }
507
+ }