{ "all_raw_sites_consumed_when_raw_enabled": true, "arm": "raw_plus_all_sidecar", "best_epoch": 12, "best_inner_validation_rmse": 0.29564625910355785, "checkpoint_artifact": { "path": "checkpoint.pt", "sha256": "66439a336847122deb2f13fb96e407aeff4fddd593a2168d1ca580e1a0c28ab5" }, "curve": [ { "epoch": 1, "fit_raw_sites_consumed": 36722254433, "fit_rmse": 0.34087404218659645, "fit_seconds": 15098.081735803004, "inner_raw_sites_consumed": 206175097, "inner_seconds": 45.10423895799613, "inner_validation_mae": 0.20996905645355582, "inner_validation_rmse": 0.3312836728049844, "learning_rate": 0.0002951444302461922, "world_size": 4 }, { "epoch": 2, "fit_raw_sites_consumed": 36722254433, "fit_rmse": 0.31444703204244884, "fit_seconds": 15090.712951476002, "inner_raw_sites_consumed": 206175097, "inner_seconds": 44.60768667497905, "inner_validation_mae": 0.20058076480403542, "inner_validation_rmse": 0.3221521386574506, "learning_rate": 0.0002809086200392825, "world_size": 4 }, { "epoch": 3, "fit_raw_sites_consumed": 36722254433, "fit_rmse": 0.306523782969338, "fit_seconds": 15086.540569486999, "inner_raw_sites_consumed": 206175097, "inner_seconds": 45.390437073016074, "inner_validation_mae": 0.20495079054683446, "inner_validation_rmse": 0.3260795499437604, "learning_rate": 0.000258262716319083, "world_size": 4 }, { "epoch": 4, "fit_raw_sites_consumed": 36722254433, "fit_rmse": 0.3012292016400525, "fit_seconds": 15124.13770476199, "inner_raw_sites_consumed": 206175097, "inner_seconds": 44.548844887991436, "inner_validation_mae": 0.19201846178621054, "inner_validation_rmse": 0.313239853650842, "learning_rate": 0.00022875, "world_size": 4 }, { "epoch": 5, "fit_raw_sites_consumed": 36722254433, "fit_rmse": 0.29665308876240726, "fit_seconds": 15065.592401795002, "inner_raw_sites_consumed": 206175097, "inner_seconds": 44.03431639200426, "inner_validation_mae": 0.18792768684402109, "inner_validation_rmse": 0.3123453922196243, "learning_rate": 0.00019438171392710925, "world_size": 4 }, { "epoch": 6, "fit_raw_sites_consumed": 36722254433, "fit_rmse": 0.292632265712787, "fit_seconds": 15058.629629426025, "inner_raw_sites_consumed": 206175097, "inner_seconds": 43.858921767998254, "inner_validation_mae": 0.18596054547280072, "inner_validation_rmse": 0.30963140134950584, "learning_rate": 0.00015750000000000003, "world_size": 4 }, { "epoch": 7, "fit_raw_sites_consumed": 36722254433, "fit_rmse": 0.2887900081237377, "fit_seconds": 15100.54132378599, "inner_raw_sites_consumed": 206175097, "inner_seconds": 44.61221433701576, "inner_validation_mae": 0.18670305621251465, "inner_validation_rmse": 0.3076383556016324, "learning_rate": 0.00012061828607289084, "world_size": 4 }, { "epoch": 8, "fit_raw_sites_consumed": 36722254433, "fit_rmse": 0.2852680109873503, "fit_seconds": 15105.252510551014, "inner_raw_sites_consumed": 206175097, "inner_seconds": 45.254287110001314, "inner_validation_mae": 0.17927646996453406, "inner_validation_rmse": 0.30296562367939545, "learning_rate": 8.625000000000005e-05, "world_size": 4 }, { "epoch": 9, "fit_raw_sites_consumed": 36722254433, "fit_rmse": 0.28190212940809534, "fit_seconds": 15109.145124823, "inner_raw_sites_consumed": 206175097, "inner_seconds": 45.04117431101622, "inner_validation_mae": 0.17761979158595204, "inner_validation_rmse": 0.29856353093021126, "learning_rate": 5.6737283680917e-05, "world_size": 4 }, { "epoch": 10, "fit_raw_sites_consumed": 36722254433, "fit_rmse": 0.2792003421644127, "fit_seconds": 15115.779493177019, "inner_raw_sites_consumed": 206175097, "inner_seconds": 45.14029743798892, "inner_validation_mae": 0.17527823546901344, "inner_validation_rmse": 0.2984785894159893, "learning_rate": 3.4091379960717494e-05, "world_size": 4 }, { "epoch": 11, "fit_raw_sites_consumed": 36722254433, "fit_rmse": 0.2769203732896985, "fit_seconds": 15059.704184403003, "inner_raw_sites_consumed": 206175097, "inner_seconds": 43.408416548976675, "inner_validation_mae": 0.17450068172067404, "inner_validation_rmse": 0.296249679580655, "learning_rate": 1.9855569753807785e-05, "world_size": 4 }, { "epoch": 12, "fit_raw_sites_consumed": 36722254433, "fit_rmse": 0.2754198010391387, "fit_seconds": 15058.952385816025, "inner_raw_sites_consumed": 206175097, "inner_seconds": 44.07527311699232, "inner_validation_mae": 0.17352580783888696, "inner_validation_rmse": 0.29564625910355785, "learning_rate": 1.4999999999999999e-05, "world_size": 4 } ], "device": "cuda", "direct_64_bin_output": true, "epochs_ran": 12, "evaluation_raw_sites_consumed": 216060598, "evaluation_seconds": 200.81153906905092, "experiment_signature": "3ca82bb9f687729cbfb65b05c31df0ec2bbc45a26a6ba23acd76df6d057797de", "fit_rows": 89996, "fit_seconds": 181073.07001530507, "fixed_epoch_budget": 12, "gpu_name": "NVIDIA GeForce RTX 4090", "hazard_tail_survival_dual_or_class_head": false, "initial_state_sha256": "7ecf840102b35595f25ab8b6f6d6fbe36ad9fed367b184e33444a2f76b85b100", "inner_rows": 500, "inner_validation_seconds": 535.0761086159764, "mechanism_evaluation_artifact": { "path": "mechanism_evaluation.npz", "sha256": "e1b6fb0397de740c75e5396415ed3fc4f97346158b1c80872f286a1ab606b329" }, "mechanism_evaluation_metrics": { "constant_predictor_rmse": 0.402485317024815, "label_spread_strata": { "flat": { "constant_predictor_rmse": 0.15479404880509584, "label_spread_range": [ 0.0, 0.24 ], "mae": 0.09360358197756899, "n_samples": 123, "rmse": 0.1436479484639223 }, "moderate": { "constant_predictor_rmse": 0.31611850295071287, "label_spread_range": [ 0.24, 0.4 ], "mae": 0.15201989890642695, "n_samples": 194, "rmse": 0.2331097103280478 }, "structured": { "constant_predictor_rmse": 0.5661845833240783, "label_spread_range": [ 0.4, null ], "mae": 0.2350760534344638, "n_samples": 183, "rmse": 0.3923640478870658 } }, "mae": 0.16804803749918937, "n_samples": 500, "per_bin_mae": [ 0.20169192790985108, 0.19998913764953613, 0.1942358236312866, 0.18924160385131836, 0.1786602282524109, 0.17287541484832764, 0.16899711894989014, 0.1672422320842743, 0.16411268377304078, 0.1675063877105713, 0.16890448093414306, 0.16627783441543578, 0.16986408948898316, 0.1631735382080078, 0.16132657957077026, 0.14841758346557618, 0.14817850589752196, 0.14446694660186768, 0.14677689170837402, 0.14296780824661254, 0.14180426502227783, 0.14598267555236816, 0.14376418566703797, 0.13781293511390685, 0.13464703631401062, 0.1399448585510254, 0.13249676990509032, 0.13033871173858644, 0.12443238925933837, 0.12065467548370361, 0.1233967113494873, 0.12303443908691407, 0.12051682472229004, 0.12519490718841553, 0.13106551933288574, 0.13314309406280517, 0.13869336271286012, 0.14007183361053466, 0.15588291692733763, 0.16631028079986573, 0.17281942892074584, 0.16888595581054688, 0.16396772623062134, 0.16385178089141847, 0.16056319904327393, 0.16767566108703613, 0.17306171655654906, 0.17412895011901855, 0.17518114471435547, 0.18032619047164916, 0.18147862386703492, 0.18711848640441894, 0.19098834276199342, 0.19500057077407837, 0.20169839763641356, 0.20566725587844847, 0.20789461183547975, 0.2123627061843872, 0.2158780574798584, 0.22555310916900634, 0.23377195405960083, 0.23035997533798216, 0.23742619943618776, 0.2553191456794739 ], "per_bin_rmse": [ 0.2819418485320672, 0.2788502766086884, 0.27289224885126334, 0.26886952732089203, 0.2564628434300029, 0.24817675854227503, 0.24624635338985015, 0.24552712283868594, 0.24589760526499801, 0.26139918240635035, 0.2658251528890732, 0.2569750302248488, 0.26896279384694605, 0.2602885612300516, 0.26046314988515074, 0.23313392036331027, 0.243578525544524, 0.23627639268588047, 0.2403076227445285, 0.2345328921796551, 0.22228622840969348, 0.23039393773078432, 0.23302763388025302, 0.21066365589246772, 0.2174826662653095, 0.23405585285039598, 0.21660457956159523, 0.2027580048724039, 0.19020919911030462, 0.1862559264681172, 0.19260170784657196, 0.1907806120545217, 0.1896077909604853, 0.20735086400679795, 0.21854841086910495, 0.22887007347359797, 0.24033269864515702, 0.24174720823873377, 0.2786224569256669, 0.3059284883238336, 0.31727660047245165, 0.3132094032944006, 0.3017479965731763, 0.2936456178609491, 0.28412538037398877, 0.31277327207613254, 0.31859022773522316, 0.31785024899503606, 0.3257192459064496, 0.3402205853662137, 0.33933019681251325, 0.345368285973525, 0.3567534523623315, 0.36026876463312374, 0.3658580104592019, 0.37470922308683635, 0.37564917149095867, 0.3761344792175013, 0.3804524331850799, 0.3830505445242531, 0.39829760736856207, 0.39236717500465007, 0.39924621899859164, 0.4236096956968575 ], "rmse": 0.28723796994893297 }, "mechanism_evaluation_rows": 500, "mechanism_evaluation_rows_requested": 500, "model_name": "carrier_raw_summary_capacity_matched_direct64_nw_residual_v1", "model_parameters": 1026409, "optimizer_boundaries_sha256": "27140647f34c5befb3c5784fe61165998d9857bdab581758a28ea16f6608df40", "peak_cuda_allocated_bytes": 12197892096, "peak_cuda_reserved_bytes": 12949913600, "publication_benchmark": false, "row_order_sha256": "70b7fc2d0fc249d228ce284133d71ac3a0550a505b289925331708401942c5f7", "schema": "dlcoalsim-carrier-raw-summary-fusion-v1", "seed": 20260728, "segment_reduction_backend": "padded", "selection_access_count": 0, "selection_rows_read": 0, "stage": "pilot", "status": "complete", "summary_only_skips_raw_io": false, "trainable_parameters": 1026409, "training_objective": { "mode_weight_alpha": 0.0, "principal_basis_rows": 0, "spectral_decoupling": 0.0 }, "training_raw_sites_consumed": 133269685380, "training_seconds": 181073.07001530507 }