ehsanroohi's picture
Publish task-preserving HJ checkpoints and model card
e9629cb verified
Raw
History Blame Contribute Delete
8.54 kB
{
"seeds": [
20260904,
20260905,
20260906
],
"arms": {
"hj_restricted": {
"label": "Task-preserving HJ",
"parameters": 268496,
"trainable_parameters": 75891,
"shock_dice": {
"airfoil": {
"mean": 0.9357071988455214,
"sample_sd": 0.004202090216577011,
"values": [
0.9344757083049362,
0.9322584471635962,
0.9403874410680321
]
},
"cylinder": {
"mean": 0.9386199049068382,
"sample_sd": 0.02439132578156726,
"values": [
0.9633139318703084,
0.9380027911530677,
0.9145429916971386
]
}
},
"core_dice": {
"airfoil": {
"mean": 0.8269258635545622,
"sample_sd": 0.04765046704785947,
"values": [
0.8326573417737513,
0.8714513644366197,
0.7766688844533156
]
},
"cylinder": {
"mean": 0.2277874006486062,
"sample_sd": 0.02043299476214628,
"values": [
0.21611023364740484,
0.21587098195068724,
0.25138098634772654
]
}
},
"core_probability_max_drift_by_seed": [
0.0,
0.0,
0.0
],
"core_probability_bit_exact_by_seed": [
true,
true,
true
],
"analytic_vortex_dice": {
"mean": 0.6724153508806848,
"sample_sd": 0.08568023829738895,
"values": [
0.5769273709675609,
0.697739762255559,
0.7425789194189347
]
},
"analytic_centre_recovery": {
"mean": 0.8888888888888888,
"sample_sd": 0.1924500897298753,
"values": [
0.6666666666666666,
1.0,
1.0
]
},
"parallel_shear_core_fraction_by_seed": [
0.0,
0.0,
0.0
],
"su2_ray_coverage": {
"mean": 1.0,
"sample_sd": 0.0,
"values": [
1.0,
1.0,
1.0
]
},
"su2_normal_mae_c": {
"mean": 0.0028370008979655163,
"sample_sd": 0.0009152518358685503,
"values": [
0.002958751756579843,
0.0018669673584372772,
0.0036852835788794285
]
},
"su2_absolute_angle_error_deg": {
"mean": 0.18143140034565305,
"sample_sd": 0.06487893122310609,
"values": [
0.21318896960592185,
0.22431372796385815,
0.10679150346717918
]
}
},
"unet_decoder": {
"label": "Shared-decoder U-Net",
"parameters": 267639,
"trainable_parameters": 107982,
"shock_dice": {
"airfoil": {
"mean": 0.935129593481926,
"sample_sd": 0.010422196951165707,
"values": [
0.9373893805309734,
0.9237628932698358,
0.9442365066449688
]
},
"cylinder": {
"mean": 0.8952466680148627,
"sample_sd": 0.02168587314766801,
"values": [
0.9029934022401697,
0.8707512695811054,
0.9119953322233133
]
}
},
"core_dice": {
"airfoil": {
"mean": 0.0826739706460272,
"sample_sd": 0.07363293735241894,
"values": [
0.031190445940795324,
0.16701517391824225,
0.049816292079044054
]
},
"cylinder": {
"mean": 0.0006778472151364567,
"sample_sd": 0.0010449639082319893,
"values": [
0.0,
0.0018812602293524971,
0.0001522814160568731
]
}
},
"core_probability_max_drift_by_seed": [
0.9986276626586914,
0.9993672966957092,
0.9808923602104187
],
"core_probability_bit_exact_by_seed": [
false,
false,
false
],
"analytic_vortex_dice": {
"mean": 0.08138905594173194,
"sample_sd": 0.06587599798052739,
"values": [
0.06842924398390403,
0.022956100595298717,
0.15278182324599307
]
},
"analytic_centre_recovery": {
"mean": 0.4722222222222222,
"sample_sd": 0.5023094811118091,
"values": [
0.4166666666666667,
0.0,
1.0
]
},
"parallel_shear_core_fraction_by_seed": [
0.0,
0.0,
0.0
],
"su2_ray_coverage": {
"mean": 1.0,
"sample_sd": 0.0,
"values": [
1.0,
1.0,
1.0
]
},
"su2_normal_mae_c": {
"mean": 0.00303275559886716,
"sample_sd": 0.00037233965760421583,
"values": [
0.003356846392082894,
0.0026260457607492488,
0.003115374643769338
]
},
"su2_absolute_angle_error_deg": {
"mean": 0.19599332091759744,
"sample_sd": 0.0681041637364286,
"values": [
0.22607250461241507,
0.2438791543721152,
0.11802830376826208
]
}
},
"hj_joint": {
"label": "Joint HJ control",
"parameters": 268496,
"trainable_parameters": 268496,
"shock_dice": {
"airfoil": {
"mean": 0.9579573129256692,
"sample_sd": 0.0012847180825509754,
"values": [
0.9566206394057329,
0.9580684437295857,
0.9591828556416889
]
},
"cylinder": {
"mean": 0.9550029949711734,
"sample_sd": 0.021041558444334304,
"values": [
0.96236045950239,
0.9312706353897836,
0.9713778900213463
]
}
},
"core_dice": {
"airfoil": {
"mean": 0.38242093516209885,
"sample_sd": 0.3819531226443472,
"values": [
0.7809354789713089,
0.346802921008104,
0.019524405506883606
]
},
"cylinder": {
"mean": 0.10417600929204775,
"sample_sd": 0.08639857863261999,
"values": [
0.1185937184552822,
0.18245874281475988,
0.011475566606101176
]
}
},
"core_probability_max_drift_by_seed": [
0.9097776412963867,
0.9530678987503052,
0.970103919506073
],
"core_probability_bit_exact_by_seed": [
false,
false,
false
],
"analytic_vortex_dice": {
"mean": 0.593029778671725,
"sample_sd": 0.1915973306070832,
"values": [
0.3750270075978114,
0.7346751674685614,
0.6693871609488024
]
},
"analytic_centre_recovery": {
"mean": 0.8611111111111112,
"sample_sd": 0.24056261216234404,
"values": [
0.5833333333333334,
1.0,
1.0
]
},
"parallel_shear_core_fraction_by_seed": [
0.0,
0.0,
0.0
],
"su2_ray_coverage": {
"mean": 1.0,
"sample_sd": 0.0,
"values": [
1.0,
1.0,
1.0
]
},
"su2_normal_mae_c": {
"mean": 0.002473480977439144,
"sample_sd": 0.00021664348876689025,
"values": [
0.002471484269415214,
0.002691115919064767,
0.002257842743837451
]
},
"su2_absolute_angle_error_deg": {
"mean": 0.1494419299895185,
"sample_sd": 0.045027183385208974,
"values": [
0.1930990552509229,
0.10315955775708403,
0.15206717696054853
]
}
}
},
"scope": {
"segmentation": "43 development-test fields per seed; weak corrected targets; thresholds selected on 43 validation fields only",
"analytic_vortex": "48 states from 24 correlated two-time analytic trajectories; no training or calibration",
"su2": "five finite analytical ray segments per seed; exposed-development fields; conditional localization"
}
}