{
  "variant": "graph_unet_6000",
  "architecture": "paper-inspired TAG U-Net with 3 kd-tree pool/unpool levels, EdgeConv max aggregation, concat skips",
  "paper_reproduction": false,
  "paper_difference": "the paper uses direct scalar-field regression with larger ReLU EdgeConv networks and much more training data; this compact width-32 SiLU backbone is adapted as a conditional DDPM denoiser",
  "nodes_per_run": 6155,
  "condition_dim": 224,
  "hierarchy_sizes": [
    6155,
    1024,
    128,
    16
  ],
  "history": [
    {
      "epoch": 1,
      "total": 1.3832834959030151,
      "denoise": 1.0231504440307617,
      "risk": 1.028951644897461,
      "AP": 0.0019082252782802725
    },
    {
      "epoch": 2,
      "total": 1.3435752391815186,
      "denoise": 0.98824143409729,
      "risk": 1.0152394771575928,
      "AP": 0.004355313562070806
    },
    {
      "epoch": 3,
      "total": 1.3426166772842407,
      "denoise": 0.9902350902557373,
      "risk": 1.0068045854568481,
      "AP": 0.006851552335860032
    },
    {
      "epoch": 4,
      "total": 1.3183937072753906,
      "denoise": 0.9685709476470947,
      "risk": 0.9994937181472778,
      "AP": 0.008845535095387852
    },
    {
      "epoch": 5,
      "total": 1.2545207738876343,
      "denoise": 0.9085962772369385,
      "risk": 0.9883556365966797,
      "AP": 0.04781017297444271
    },
    {
      "epoch": 6,
      "total": 1.270835280418396,
      "denoise": 0.9276677370071411,
      "risk": 0.9804787635803223,
      "AP": 0.04412472683149903
    },
    {
      "epoch": 7,
      "total": 1.2168219089508057,
      "denoise": 0.8744961023330688,
      "risk": 0.9780738353729248,
      "AP": 0.03618512544895219
    },
    {
      "epoch": 8,
      "total": 1.1838499307632446,
      "denoise": 0.8429790735244751,
      "risk": 0.9739168286323547,
      "AP": 0.011679063474048382
    },
    {
      "epoch": 9,
      "total": 1.1558979749679565,
      "denoise": 0.8187035322189331,
      "risk": 0.9634127616882324,
      "AP": 0.14115366333994933
    },
    {
      "epoch": 10,
      "total": 1.15428626537323,
      "denoise": 0.8185247182846069,
      "risk": 0.9593187570571899,
      "AP": 0.028465990604778054
    },
    {
      "epoch": 11,
      "total": 1.0709726810455322,
      "denoise": 0.7373290657997131,
      "risk": 0.9532673954963684,
      "AP": 0.041160603471631134
    },
    {
      "epoch": 12,
      "total": 1.0499616861343384,
      "denoise": 0.7172365784645081,
      "risk": 0.9506430625915527,
      "AP": 0.02886421207507016
    },
    {
      "epoch": 13,
      "total": 1.0210708379745483,
      "denoise": 0.6904327273368835,
      "risk": 0.944680392742157,
      "AP": 0.0724180031649329
    },
    {
      "epoch": 14,
      "total": 0.981753408908844,
      "denoise": 0.6526647210121155,
      "risk": 0.9402534365653992,
      "AP": 0.040535192570228994
    },
    {
      "epoch": 15,
      "total": 0.9525766372680664,
      "denoise": 0.625500500202179,
      "risk": 0.9345031976699829,
      "AP": 0.035389948680470004
    },
    {
      "epoch": 16,
      "total": 0.9162208437919617,
      "denoise": 0.5896498560905457,
      "risk": 0.9330599904060364,
      "AP": 0.02412339177407328
    },
    {
      "epoch": 17,
      "total": 0.8841019868850708,
      "denoise": 0.560329020023346,
      "risk": 0.9250655174255371,
      "AP": 0.14186698395720362
    },
    {
      "epoch": 18,
      "total": 0.8505452871322632,
      "denoise": 0.5251228213310242,
      "risk": 0.9297783970832825,
      "AP": 0.024226723299732952
    },
    {
      "epoch": 19,
      "total": 0.8350903987884521,
      "denoise": 0.5116632580757141,
      "risk": 0.924077570438385,
      "AP": 0.029319271292597825
    },
    {
      "epoch": 20,
      "total": 0.8247806429862976,
      "denoise": 0.5013062953948975,
      "risk": 0.9242123961448669,
      "AP": 0.02068069743364477
    },
    {
      "epoch": 21,
      "total": 0.7972308397293091,
      "denoise": 0.4767487943172455,
      "risk": 0.9156630039215088,
      "AP": 0.14436184842643787
    },
    {
      "epoch": 22,
      "total": 0.7772327661514282,
      "denoise": 0.4589838981628418,
      "risk": 0.9092825055122375,
      "AP": 0.031567111747269436
    },
    {
      "epoch": 23,
      "total": 0.7582919597625732,
      "denoise": 0.44094157218933105,
      "risk": 0.906715452671051,
      "AP": 0.1436925549326705
    },
    {
      "epoch": 24,
      "total": 0.7379555702209473,
      "denoise": 0.41917338967323303,
      "risk": 0.9108061194419861,
      "AP": 0.13887485582038447
    }
  ],
  "best_loss": 0.7379555702209473,
  "best_epoch": 24,
  "metrics": [
    {
      "run": "14h06",
      "MAE_mm": 9.635078430175781,
      "RMSE_mm": 12.06879711151123,
      "CAD_correspondent_Chamfer_mm": 7.652663230895996,
      "registered_scan_Chamfer_mm": 33.372215270996094,
      "risk": {
        "AP": 0.005831526902786479,
        "IoU_at_0.5": 0.0012997562956945572,
        "Dice_at_0.5": 0.0025961382443615124
      },
      "mean_stochastic_std_mm": 17.090410232543945,
      "max_stochastic_std_mm": 30.0
    }
  ],
  "checkpoint": "gen_models/models/graph_unet_6000.pt",
  "production_valid": false,
  "limitation": "1 weak/unvalidated NG run(s); no held-out generalization result"
}