{
  "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 and trained for 24 epochs on two runs",
  "nodes_per_run": 5925,
  "condition_dim": 224,
  "hierarchy_sizes": [
    5925,
    1024,
    128,
    16
  ],
  "history": [
    {
      "epoch": 1,
      "total": 1.4255238771438599,
      "denoise": 1.042484164237976,
      "risk": 1.094399094581604,
      "AP": 0.001461073156131731
    },
    {
      "epoch": 2,
      "total": 1.3716135025024414,
      "denoise": 0.9999887347221375,
      "risk": 1.0617849826812744,
      "AP": 0.007039999423477495
    },
    {
      "epoch": 3,
      "total": 1.3586900234222412,
      "denoise": 0.9929608106613159,
      "risk": 1.0449405908584595,
      "AP": 0.011096485106778672
    },
    {
      "epoch": 4,
      "total": 1.3031413555145264,
      "denoise": 0.9443404078483582,
      "risk": 1.025145411491394,
      "AP": 0.0217848186144266
    },
    {
      "epoch": 5,
      "total": 1.2969038486480713,
      "denoise": 0.9452704787254333,
      "risk": 1.0046666860580444,
      "AP": 0.037762821629466196
    },
    {
      "epoch": 6,
      "total": 1.2666504383087158,
      "denoise": 0.9220626354217529,
      "risk": 0.9845367670059204,
      "AP": 0.06383192800400399
    },
    {
      "epoch": 7,
      "total": 1.2456353902816772,
      "denoise": 0.9067770838737488,
      "risk": 0.9681665301322937,
      "AP": 0.06107925068073174
    },
    {
      "epoch": 8,
      "total": 1.1996923685073853,
      "denoise": 0.8658297061920166,
      "risk": 0.9538933038711548,
      "AP": 0.16804683207715213
    },
    {
      "epoch": 9,
      "total": 1.1838434934616089,
      "denoise": 0.8566049337387085,
      "risk": 0.9349673390388489,
      "AP": 0.16317393438159056
    },
    {
      "epoch": 10,
      "total": 1.1489417552947998,
      "denoise": 0.8234162330627441,
      "risk": 0.9300730228424072,
      "AP": 0.14556540024321263
    },
    {
      "epoch": 11,
      "total": 1.1197205781936646,
      "denoise": 0.7985026240348816,
      "risk": 0.9177656173706055,
      "AP": 0.19788282511039243
    },
    {
      "epoch": 12,
      "total": 1.0724351406097412,
      "denoise": 0.7547782063484192,
      "risk": 0.9075912833213806,
      "AP": 0.25707247432430946
    },
    {
      "epoch": 13,
      "total": 1.0425127744674683,
      "denoise": 0.7323633432388306,
      "risk": 0.8861411213874817,
      "AP": 0.1981552505858875
    },
    {
      "epoch": 14,
      "total": 1.01426100730896,
      "denoise": 0.7053802609443665,
      "risk": 0.8825163245201111,
      "AP": 0.24462444183348403
    },
    {
      "epoch": 15,
      "total": 0.9745813608169556,
      "denoise": 0.6670998334884644,
      "risk": 0.8785187005996704,
      "AP": 0.21146515942321084
    },
    {
      "epoch": 16,
      "total": 0.9450134038925171,
      "denoise": 0.6390492916107178,
      "risk": 0.874183177947998,
      "AP": 0.2305927281652205
    },
    {
      "epoch": 17,
      "total": 0.9044433832168579,
      "denoise": 0.6007190942764282,
      "risk": 0.8677837252616882,
      "AP": 0.2240875856461619
    },
    {
      "epoch": 18,
      "total": 0.8655245900154114,
      "denoise": 0.5653916001319885,
      "risk": 0.8575228452682495,
      "AP": 0.24529588231708033
    },
    {
      "epoch": 19,
      "total": 0.8460791110992432,
      "denoise": 0.5425108075141907,
      "risk": 0.8673381209373474,
      "AP": 0.15846513801840806
    },
    {
      "epoch": 20,
      "total": 0.820114016532898,
      "denoise": 0.5202183127403259,
      "risk": 0.8568448424339294,
      "AP": 0.19158497374821068
    },
    {
      "epoch": 21,
      "total": 0.7967633008956909,
      "denoise": 0.5018002390861511,
      "risk": 0.8427517414093018,
      "AP": 0.24979710365444746
    },
    {
      "epoch": 22,
      "total": 0.7744503021240234,
      "denoise": 0.4796822667121887,
      "risk": 0.8421944975852966,
      "AP": 0.18911616991448565
    },
    {
      "epoch": 23,
      "total": 0.7701451778411865,
      "denoise": 0.47912827134132385,
      "risk": 0.8314768075942993,
      "AP": 0.24675653593955163
    },
    {
      "epoch": 24,
      "total": 0.7464787364006042,
      "denoise": 0.45679381489753723,
      "risk": 0.827671229839325,
      "AP": 0.2854615107531819
    }
  ],
  "best_loss": 0.7464787364006042,
  "best_epoch": 24,
  "metrics": [
    {
      "run": "16h",
      "MAE_mm": 8.203474044799805,
      "RMSE_mm": 10.455013275146484,
      "CAD_correspondent_Chamfer_mm": 6.050500392913818,
      "registered_scan_Chamfer_mm": 13.713613510131836,
      "risk": {
        "AP": 0.5141943734015345,
        "IoU_at_0.5": 0.0008438818565400844,
        "Dice_at_0.5": 0.0016863406408094434
      },
      "mean_stochastic_std_mm": 14.17358112335205,
      "max_stochastic_std_mm": 29.913127899169922
    },
    {
      "run": "18h01",
      "MAE_mm": 8.400187492370605,
      "RMSE_mm": 10.712625503540039,
      "CAD_correspondent_Chamfer_mm": 6.1764044761657715,
      "registered_scan_Chamfer_mm": 16.041446685791016,
      "risk": {
        "AP": 0.2684651032103203,
        "IoU_at_0.5": 0.0018565400843881857,
        "Dice_at_0.5": 0.003706199460916442
      },
      "mean_stochastic_std_mm": 14.112639427185059,
      "max_stochastic_std_mm": 29.20965576171875
    }
  ],
  "checkpoint": "gen_models/models/graph_unet_6000.pt",
  "production_valid": false,
  "limitation": "two weak/unvalidated NG runs; no held-out generalization result"
}