{ "status": "four_conditional_diffusion_weak_label_diagnostic_completed", "production_valid": false, "runs": [ "14h06" ], "variants": [ { "variant": "image_2d", "representation": { "source": "data_processed/images/cad_multiview_128.npz", "representation": "six CAD orthographic views with explicit 2-D pixel-neighbor message aggregation", "nodes": 6000, "denoiser_adapter": "4 nearest selected image-grid pixels in the same view; not a CNN" }, "nodes_per_run": 6000, "condition_dim": 194, "history": [ { "epoch": 1, "total": 1.4718353748321533, "denoise": 0.9902293086051941, "risk": 1.3760172128677368, "AP": 0.013667851049641138 }, { "epoch": 2, "total": 1.4835400581359863, "denoise": 1.0065678358078003, "risk": 1.3627779483795166, "AP": 0.08253952923461665 }, { "epoch": 3, "total": 1.4426686763763428, "denoise": 0.9711874127388, "risk": 1.347089409828186, "AP": 0.10824244859650929 }, { "epoch": 4, "total": 1.4430391788482666, "denoise": 0.9779464602470398, "risk": 1.3288365602493286, "AP": 0.11517286826664472 }, { "epoch": 5, "total": 1.4455012083053589, "denoise": 0.9886595010757446, "risk": 1.3052620887756348, "AP": 0.11545961178254954 }, { "epoch": 6, "total": 1.4061307907104492, "denoise": 0.9594659805297852, "risk": 1.2761850357055664, "AP": 0.11639531636485896 }, { "epoch": 7, "total": 1.4286023378372192, "denoise": 0.9942415356636047, "risk": 1.2410309314727783, "AP": 0.1163492739247827 }, { "epoch": 8, "total": 1.3934481143951416, "denoise": 0.9750063419342041, "risk": 1.1955478191375732, "AP": 0.11611015473110267 }, { "epoch": 9, "total": 1.396803617477417, "denoise": 0.9984453320503235, "risk": 1.1381665468215942, "AP": 0.11645905448444398 }, { "epoch": 10, "total": 1.3528152704238892, "denoise": 0.972769021987915, "risk": 1.0858464241027832, "AP": 0.1170685120784878 }, { "epoch": 11, "total": 1.319230079650879, "denoise": 0.9576271176338196, "risk": 1.0331515073776245, "AP": 0.11890729755463834 }, { "epoch": 12, "total": 1.3115036487579346, "denoise": 0.9699121117591858, "risk": 0.9759759902954102, "AP": 0.12174934534750728 }, { "epoch": 13, "total": 1.2804713249206543, "denoise": 0.9503675103187561, "risk": 0.943153977394104, "AP": 0.1269248071923359 }, { "epoch": 14, "total": 1.2458186149597168, "denoise": 0.9302628636360168, "risk": 0.901587963104248, "AP": 0.12889273501214557 }, { "epoch": 15, "total": 1.248363971710205, "denoise": 0.9410349726676941, "risk": 0.8780826926231384, "AP": 0.13277338518132142 }, { "epoch": 16, "total": 1.258558750152588, "denoise": 0.9488471746444702, "risk": 0.8848904371261597, "AP": 0.13197833603761586 }, { "epoch": 17, "total": 1.2363718748092651, "denoise": 0.9372745752334595, "risk": 0.8545637726783752, "AP": 0.1322068997307376 }, { "epoch": 18, "total": 1.2411634922027588, "denoise": 0.9484814405441284, "risk": 0.8362345099449158, "AP": 0.1300520665926755 }, { "epoch": 19, "total": 1.2055455446243286, "denoise": 0.9140351414680481, "risk": 0.8328869342803955, "AP": 0.12744543130311273 }, { "epoch": 20, "total": 1.2047481536865234, "denoise": 0.9221686124801636, "risk": 0.8073699474334717, "AP": 0.12522836597620984 }, { "epoch": 21, "total": 1.1621931791305542, "denoise": 0.8821758031845093, "risk": 0.8000497817993164, "AP": 0.12476349359399563 }, { "epoch": 22, "total": 1.1662580966949463, "denoise": 0.8922224044799805, "risk": 0.7829592227935791, "AP": 0.1259358030460032 }, { "epoch": 23, "total": 1.1696476936340332, "denoise": 0.9006295800209045, "risk": 0.7686232328414917, "AP": 0.1274240941442041 }, { "epoch": 24, "total": 1.1510701179504395, "denoise": 0.8839195966720581, "risk": 0.7632872462272644, "AP": 0.12877263059725338 }, { "epoch": 25, "total": 1.1417897939682007, "denoise": 0.8770748972892761, "risk": 0.7563282251358032, "AP": 0.13019588396604487 }, { "epoch": 26, "total": 1.1293225288391113, "denoise": 0.8658654093742371, "risk": 0.7527347803115845, "AP": 0.1322317800651904 }, { "epoch": 27, "total": 1.1072356700897217, "denoise": 0.8449776768684387, "risk": 0.7493085861206055, "AP": 0.13615178223004032 }, { "epoch": 28, "total": 1.0976572036743164, "denoise": 0.8368334770202637, "risk": 0.7452108263969421, "AP": 0.13927289089961925 }, { "epoch": 29, "total": 1.098901629447937, "denoise": 0.839140772819519, "risk": 0.7421738505363464, "AP": 0.14175249244628638 }, { "epoch": 30, "total": 1.0882368087768555, "denoise": 0.8302797079086304, "risk": 0.7370201945304871, "AP": 0.14453290828903595 }, { "epoch": 31, "total": 1.0709478855133057, "denoise": 0.8159412145614624, "risk": 0.7285904884338379, "AP": 0.14571536991452166 }, { "epoch": 32, "total": 1.0742210149765015, "denoise": 0.8205258846282959, "risk": 0.7248432636260986, "AP": 0.14496674451827368 }, { "epoch": 33, "total": 1.0588524341583252, "denoise": 0.8026541471481323, "risk": 0.7319951057434082, "AP": 0.14485788444310324 }, { "epoch": 34, "total": 1.0509700775146484, "denoise": 0.780388355255127, "risk": 0.7730908393859863, "AP": 0.1451794276522736 }, { "epoch": 35, "total": 1.0189911127090454, "denoise": 0.7525423765182495, "risk": 0.7612821459770203, "AP": 0.1453313154957198 }, { "epoch": 36, "total": 0.9993460178375244, "denoise": 0.750098466873169, "risk": 0.712135910987854, "AP": 0.14509985108209156 }, { "epoch": 37, "total": 0.9816952347755432, "denoise": 0.7202211022377014, "risk": 0.7470690011978149, "AP": 0.145106626335661 }, { "epoch": 38, "total": 1.0027656555175781, "denoise": 0.7496258020401001, "risk": 0.7232569456100464, "AP": 0.14555582641558013 }, { "epoch": 39, "total": 1.047140121459961, "denoise": 0.7999752759933472, "risk": 0.7061851620674133, "AP": 0.14572523787480257 }, { "epoch": 40, "total": 1.0251989364624023, "denoise": 0.7753177881240845, "risk": 0.7139459848403931, "AP": 0.1455977891972928 }, { "epoch": 41, "total": 0.9418255686759949, "denoise": 0.6979974508285522, "risk": 0.6966517567634583, "AP": 0.1469716207913364 }, { "epoch": 42, "total": 0.9630658626556396, "denoise": 0.715262234210968, "risk": 0.7080104351043701, "AP": 0.14692483109043245 }, { "epoch": 43, "total": 0.9280459880828857, "denoise": 0.6843795776367188, "risk": 0.6961897015571594, "AP": 0.14676845019695328 }, { "epoch": 44, "total": 0.9589401483535767, "denoise": 0.7151589393615723, "risk": 0.6965177655220032, "AP": 0.1480556288441579 }, { "epoch": 45, "total": 0.9261332750320435, "denoise": 0.6818899512290955, "risk": 0.6978380680084229, "AP": 0.1484960306233204 }, { "epoch": 46, "total": 0.9420628547668457, "denoise": 0.7017595171928406, "risk": 0.6865809559822083, "AP": 0.1492325885809365 }, { "epoch": 47, "total": 0.8943822383880615, "denoise": 0.6521181464195251, "risk": 0.6921831965446472, "AP": 0.14918637818703406 }, { "epoch": 48, "total": 0.8874256610870361, "denoise": 0.6497570872306824, "risk": 0.6790531277656555, "AP": 0.1498041773137646 }, { "epoch": 49, "total": 0.8784026503562927, "denoise": 0.639793336391449, "risk": 0.6817408800125122, "AP": 0.15058751750518756 }, { "epoch": 50, "total": 0.8501132130622864, "denoise": 0.6142223477363586, "risk": 0.6739739179611206, "AP": 0.15075568969401418 }, { "epoch": 51, "total": 0.8652151823043823, "denoise": 0.6305760145187378, "risk": 0.6703976392745972, "AP": 0.1509793412010743 }, { "epoch": 52, "total": 0.8506553173065186, "denoise": 0.6156002879142761, "risk": 0.6715858578681946, "AP": 0.1520153083658428 }, { "epoch": 53, "total": 0.8420814871788025, "denoise": 0.6096522212028503, "risk": 0.6640836596488953, "AP": 0.15581912259936606 }, { "epoch": 54, "total": 0.8358715176582336, "denoise": 0.603508472442627, "risk": 0.6638944745063782, "AP": 0.15651317820768793 }, { "epoch": 55, "total": 0.8189959526062012, "denoise": 0.5861862897872925, "risk": 0.6651705503463745, "AP": 0.15736736506623292 }, { "epoch": 56, "total": 0.8245144486427307, "denoise": 0.5940862894058228, "risk": 0.6583662033081055, "AP": 0.1572228017075275 }, { "epoch": 57, "total": 0.8223952054977417, "denoise": 0.5929749011993408, "risk": 0.6554866433143616, "AP": 0.15772997961840182 }, { "epoch": 58, "total": 0.8496524095535278, "denoise": 0.6184465885162354, "risk": 0.6605880260467529, "AP": 0.1586413509624185 }, { "epoch": 59, "total": 0.8484728336334229, "denoise": 0.6162909269332886, "risk": 0.6633768081665039, "AP": 0.15900315546977206 }, { "epoch": 60, "total": 0.8519440293312073, "denoise": 0.624722957611084, "risk": 0.6492031216621399, "AP": 0.16039904201344476 }, { "epoch": 61, "total": 0.7837828397750854, "denoise": 0.5583200454711914, "risk": 0.6441795229911804, "AP": 0.16192800669907256 }, { "epoch": 62, "total": 0.7639881372451782, "denoise": 0.5364375114440918, "risk": 0.6501446962356567, "AP": 0.16323907178811106 }, { "epoch": 63, "total": 0.7765469551086426, "denoise": 0.5491892695426941, "risk": 0.6495932936668396, "AP": 0.16329623280665204 }, { "epoch": 64, "total": 0.7596136331558228, "denoise": 0.5346341729164124, "risk": 0.6427984833717346, "AP": 0.16395229616712473 }, { "epoch": 65, "total": 0.7264940142631531, "denoise": 0.5036266446113586, "risk": 0.6367639303207397, "AP": 0.16510125319417618 }, { "epoch": 66, "total": 0.7555314302444458, "denoise": 0.5321931838989258, "risk": 0.6381093263626099, "AP": 0.165044356292693 }, { "epoch": 67, "total": 0.7567762136459351, "denoise": 0.5331441760063171, "risk": 0.6389487385749817, "AP": 0.16518226050099896 }, { "epoch": 68, "total": 0.717340886592865, "denoise": 0.4964255094528198, "risk": 0.6311867833137512, "AP": 0.1665546431632383 }, { "epoch": 69, "total": 0.7447202205657959, "denoise": 0.5252679586410522, "risk": 0.6270065307617188, "AP": 0.16817397263133035 }, { "epoch": 70, "total": 0.7327861189842224, "denoise": 0.5132440328598022, "risk": 0.6272631287574768, "AP": 0.16855698437284125 }, { "epoch": 71, "total": 0.7119765877723694, "denoise": 0.49177706241607666, "risk": 0.6291415691375732, "AP": 0.17009406958089018 }, { "epoch": 72, "total": 0.7107897400856018, "denoise": 0.4894798696041107, "risk": 0.6323139071464539, "AP": 0.17075805500570673 }, { "epoch": 73, "total": 0.7182679176330566, "denoise": 0.49819302558898926, "risk": 0.6287854909896851, "AP": 0.17314673447973697 }, { "epoch": 74, "total": 0.7299298644065857, "denoise": 0.5121610760688782, "risk": 0.6221964955329895, "AP": 0.1728237061558461 }, { "epoch": 75, "total": 0.6729649305343628, "denoise": 0.4575810134410858, "risk": 0.6153825521469116, "AP": 0.17264734348330327 }, { "epoch": 76, "total": 0.6741083860397339, "denoise": 0.4591074287891388, "risk": 0.6142885088920593, "AP": 0.17546994810636635 }, { "epoch": 77, "total": 0.6978290677070618, "denoise": 0.4809930920600891, "risk": 0.6195313930511475, "AP": 0.17608719215185883 }, { "epoch": 78, "total": 0.7089216113090515, "denoise": 0.4883851408958435, "risk": 0.6301042437553406, "AP": 0.17967971573131525 }, { "epoch": 79, "total": 0.6782518029212952, "denoise": 0.44686833024024963, "risk": 0.6610956788063049, "AP": 0.1804019974155708 }, { "epoch": 80, "total": 0.6784766316413879, "denoise": 0.4524761438369751, "risk": 0.6457157135009766, "AP": 0.18401403165279065 }, { "epoch": 81, "total": 0.7025994658470154, "denoise": 0.48035404086112976, "risk": 0.6349869966506958, "AP": 0.1834424609989459 }, { "epoch": 82, "total": 0.6705830693244934, "denoise": 0.4573911130428314, "risk": 0.6091198921203613, "AP": 0.19542976491434516 }, { "epoch": 83, "total": 0.6625417470932007, "denoise": 0.45151570439338684, "risk": 0.6029316186904907, "AP": 0.19839776149133379 }, { "epoch": 84, "total": 0.6757927536964417, "denoise": 0.46124130487442017, "risk": 0.6130040884017944, "AP": 0.19584931330900243 }, { "epoch": 85, "total": 0.6549682021141052, "denoise": 0.4413692355155945, "risk": 0.6102827191352844, "AP": 0.1942493906267407 }, { "epoch": 86, "total": 0.6617079973220825, "denoise": 0.4513738751411438, "risk": 0.6009546518325806, "AP": 0.19376590974231048 }, { "epoch": 87, "total": 0.6825348734855652, "denoise": 0.4732055366039276, "risk": 0.5980837941169739, "AP": 0.19772681748255794 }, { "epoch": 88, "total": 0.6593219041824341, "denoise": 0.4482690393924713, "risk": 0.6030081510543823, "AP": 0.19851466455268388 }, { "epoch": 89, "total": 0.6534088253974915, "denoise": 0.44161126017570496, "risk": 0.6051359176635742, "AP": 0.19851449386116934 }, { "epoch": 90, "total": 0.6588258743286133, "denoise": 0.45079275965690613, "risk": 0.5943803787231445, "AP": 0.20438416963482017 }, { "epoch": 91, "total": 0.6420347690582275, "denoise": 0.43449369072914124, "risk": 0.5929744243621826, "AP": 0.2033822851142593 }, { "epoch": 92, "total": 0.6182634234428406, "denoise": 0.40786856412887573, "risk": 0.6011281609535217, "AP": 0.20006907908193428 }, { "epoch": 93, "total": 0.6473782062530518, "denoise": 0.4385462701320648, "risk": 0.5966625809669495, "AP": 0.18183945143360516 }, { "epoch": 94, "total": 0.6357333660125732, "denoise": 0.4296287000179291, "risk": 0.5888705849647522, "AP": 0.18161900582991597 }, { "epoch": 95, "total": 0.6204533576965332, "denoise": 0.4155697524547577, "risk": 0.5853817462921143, "AP": 0.17316828287448272 }, { "epoch": 96, "total": 0.6260461807250977, "denoise": 0.4205712378025055, "risk": 0.5870711803436279, "AP": 0.1688904612419913 }, { "epoch": 97, "total": 0.6062655448913574, "denoise": 0.4007149934768677, "risk": 0.5872873663902283, "AP": 0.16879159079273703 }, { "epoch": 98, "total": 0.6590864658355713, "denoise": 0.45437246561050415, "risk": 0.5848972201347351, "AP": 0.16234930417238289 }, { "epoch": 99, "total": 0.6256868243217468, "denoise": 0.42101144790649414, "risk": 0.5847867727279663, "AP": 0.16296146957563745 }, { "epoch": 100, "total": 0.6331099271774292, "denoise": 0.4287329912185669, "risk": 0.5839340686798096, "AP": 0.1573738486645062 }, { "epoch": 101, "total": 0.6418925523757935, "denoise": 0.4379611015319824, "risk": 0.5826613306999207, "AP": 0.15768776702765516 }, { "epoch": 102, "total": 0.6484778523445129, "denoise": 0.4465652406215668, "risk": 0.576893150806427, "AP": 0.15312576437775238 }, { "epoch": 103, "total": 0.6117595434188843, "denoise": 0.41096755862236023, "risk": 0.5736913084983826, "AP": 0.152013696822107 }, { "epoch": 104, "total": 0.6411639451980591, "denoise": 0.44082126021385193, "risk": 0.5724077224731445, "AP": 0.15000272961229372 }, { "epoch": 105, "total": 0.6234807968139648, "denoise": 0.42344143986701965, "risk": 0.5715411305427551, "AP": 0.14661842716810675 }, { "epoch": 106, "total": 0.6019824147224426, "denoise": 0.40235498547554016, "risk": 0.5703641176223755, "AP": 0.1459355264304413 }, { "epoch": 107, "total": 0.5975226759910583, "denoise": 0.39810076355934143, "risk": 0.5697769522666931, "AP": 0.1408875286000419 }, { "epoch": 108, "total": 0.6086756587028503, "denoise": 0.4079911410808563, "risk": 0.5733842849731445, "AP": 0.1428600761500438 }, { "epoch": 109, "total": 0.6202412247657776, "denoise": 0.4135388135910034, "risk": 0.5905783772468567, "AP": 0.13603636312264067 }, { "epoch": 110, "total": 0.6544554829597473, "denoise": 0.42065751552581787, "risk": 0.6679941415786743, "AP": 0.1468645240067743 }, { "epoch": 111, "total": 0.6320899724960327, "denoise": 0.39865633845329285, "risk": 0.6669531464576721, "AP": 0.12920770268845486 }, { "epoch": 112, "total": 0.6474089026451111, "denoise": 0.397075891494751, "risk": 0.7152372002601624, "AP": 0.14841776847548552 }, { "epoch": 113, "total": 0.6065918207168579, "denoise": 0.4100758135318756, "risk": 0.5614742636680603, "AP": 0.1403760001076299 }, { "epoch": 114, "total": 0.6590617895126343, "denoise": 0.40618303418159485, "risk": 0.7225107550621033, "AP": 0.1304499286518811 }, { "epoch": 115, "total": 0.710283637046814, "denoise": 0.42721855640411377, "risk": 0.808757483959198, "AP": 0.15501401987673066 }, { "epoch": 116, "total": 0.6325072050094604, "denoise": 0.39521166682243347, "risk": 0.677987277507782, "AP": 0.15305042151695647 }, { "epoch": 117, "total": 0.6851169466972351, "denoise": 0.41354793310165405, "risk": 0.7759114503860474, "AP": 0.1335879525497929 }, { "epoch": 118, "total": 0.6251971125602722, "denoise": 0.4282369911670685, "risk": 0.5627432465553284, "AP": 0.1467805423385268 }, { "epoch": 119, "total": 0.6375710964202881, "denoise": 0.39984461665153503, "risk": 0.6792185306549072, "AP": 0.15736791622996898 }, { "epoch": 120, "total": 0.6068252921104431, "denoise": 0.3961517810821533, "risk": 0.6019243001937866, "AP": 0.15610068042809958 } ], "best_loss": 0.5975226759910583, "best_epoch": 107, "metrics": [ { "run": "14h06", "MAE_mm": 5.738138198852539, "RMSE_mm": 7.550040245056152, "CAD_correspondent_Chamfer_mm": 5.469223976135254, "registered_scan_Chamfer_mm": 32.22509765625, "risk": { "AP": 0.1428600761500438, "IoU_at_0.5": 0.0547008547008547, "Dice_at_0.5": 0.10372771474878444 }, "mean_stochastic_std_mm": 8.586365699768066, "max_stochastic_std_mm": 25.067598342895508 } ], "heater_projection_valid": [ false ], "checkpoint": "gen_models/models/image_2d.pt" }, { "variant": "voxel_3d", "representation": { "source": "data_processed/voxels/cad_surface_sparse_64.npz", "representation": "64-grid sparse CAD surface voxels with 3-D voxel-neighbor message aggregation", "nodes": 2592, "denoiser_adapter": "6 nearest sparse voxel-grid neighbours; not a dense 3-D CNN" }, "nodes_per_run": 2592, "condition_dim": 184, "history": [ { "epoch": 1, "total": 1.4969472885131836, "denoise": 1.013253927230835, "risk": 1.3819811344146729, "AP": 0.005513425550034996 }, { "epoch": 2, "total": 1.5145291090011597, "denoise": 1.0381022691726685, "risk": 1.3612195253372192, "AP": 0.06064400725429359 }, { "epoch": 3, "total": 1.4721039533615112, "denoise": 1.0008957386016846, "risk": 1.3463091850280762, "AP": 0.099424516471032 }, { "epoch": 4, "total": 1.4611756801605225, "denoise": 0.9958838820457458, "risk": 1.3294050693511963, "AP": 0.11223992422670313 }, { "epoch": 5, "total": 1.4425057172775269, "denoise": 0.9844507575035095, "risk": 1.308728575706482, "AP": 0.12449867811819933 }, { "epoch": 6, "total": 1.427966833114624, "denoise": 0.9786120653152466, "risk": 1.2838705778121948, "AP": 0.13167811536883173 }, { "epoch": 7, "total": 1.3866651058197021, "denoise": 0.9470096230506897, "risk": 1.256158471107483, "AP": 0.13515943284855164 }, { "epoch": 8, "total": 1.436253547668457, "denoise": 1.0073038339614868, "risk": 1.2255704402923584, "AP": 0.14028867826347646 }, { "epoch": 9, "total": 1.4192925691604614, "denoise": 1.0005773305892944, "risk": 1.1963292360305786, "AP": 0.14578046580323584 }, { "epoch": 10, "total": 1.4353926181793213, "denoise": 1.0224320888519287, "risk": 1.179887056350708, "AP": 0.14819212081762936 }, { "epoch": 11, "total": 1.3322757482528687, "denoise": 0.9217199683189392, "risk": 1.1730164289474487, "AP": 0.14903167428293168 }, { "epoch": 12, "total": 1.339016318321228, "denoise": 0.9327783584594727, "risk": 1.1606799364089966, "AP": 0.1497456322675061 }, { "epoch": 13, "total": 1.3740776777267456, "denoise": 0.9734724760055542, "risk": 1.144586205482483, "AP": 0.15546899889063184 }, { "epoch": 14, "total": 1.3468362092971802, "denoise": 0.9570531249046326, "risk": 1.1136659383773804, "AP": 0.15763813917496344 }, { "epoch": 15, "total": 1.2920401096343994, "denoise": 0.9142871499061584, "risk": 1.0792940855026245, "AP": 0.15794575105993078 }, { "epoch": 16, "total": 1.2797157764434814, "denoise": 0.9138386249542236, "risk": 1.045363187789917, "AP": 0.1581052465847162 }, { "epoch": 17, "total": 1.2966358661651611, "denoise": 0.9419649839401245, "risk": 1.0133452415466309, "AP": 0.1586697229640422 }, { "epoch": 18, "total": 1.2486839294433594, "denoise": 0.9059056639671326, "risk": 0.9793664216995239, "AP": 0.1582404679995774 }, { "epoch": 19, "total": 1.2624576091766357, "denoise": 0.9318151473999023, "risk": 0.9446929097175598, "AP": 0.15791951605018278 }, { "epoch": 20, "total": 1.250940203666687, "denoise": 0.9338876008987427, "risk": 0.9058646559715271, "AP": 0.15767379466420234 }, { "epoch": 21, "total": 1.2243355512619019, "denoise": 0.9214529991149902, "risk": 0.8653786778450012, "AP": 0.15795531631148282 }, { "epoch": 22, "total": 1.1824969053268433, "denoise": 0.8943243026733398, "risk": 0.8233502507209778, "AP": 0.1584387706430097 }, { "epoch": 23, "total": 1.1799356937408447, "denoise": 0.9057449698448181, "risk": 0.7834022641181946, "AP": 0.15805692542816432 }, { "epoch": 24, "total": 1.215828537940979, "denoise": 0.954155445098877, "risk": 0.7476373910903931, "AP": 0.1556016502227013 }, { "epoch": 25, "total": 1.1155363321304321, "denoise": 0.863362729549408, "risk": 0.7204961180686951, "AP": 0.15343921952130157 }, { "epoch": 26, "total": 1.1372523307800293, "denoise": 0.8605913519859314, "risk": 0.7904598116874695, "AP": 0.11886343971465224 }, { "epoch": 27, "total": 1.287498116493225, "denoise": 0.8979659676551819, "risk": 1.1129491329193115, "AP": 0.12957359974820154 }, { "epoch": 28, "total": 1.0862598419189453, "denoise": 0.833051323890686, "risk": 0.7234527468681335, "AP": 0.1283276387445374 }, { "epoch": 29, "total": 1.1847918033599854, "denoise": 0.8337994813919067, "risk": 1.0028352737426758, "AP": 0.11868982959556254 }, { "epoch": 30, "total": 1.1115210056304932, "denoise": 0.8631356358528137, "risk": 0.7096724510192871, "AP": 0.13420782845473664 }, { "epoch": 31, "total": 1.1235010623931885, "denoise": 0.8619526028633118, "risk": 0.7472813725471497, "AP": 0.1681416562432813 }, { "epoch": 32, "total": 1.1419014930725098, "denoise": 0.8506941795349121, "risk": 0.8320209980010986, "AP": 0.17039372602493505 }, { "epoch": 33, "total": 1.097809076309204, "denoise": 0.850545346736908, "risk": 0.7064679861068726, "AP": 0.17376528193004454 }, { "epoch": 34, "total": 1.0803778171539307, "denoise": 0.8435521721839905, "risk": 0.6766448020935059, "AP": 0.1798443664067434 }, { "epoch": 35, "total": 1.122327208518982, "denoise": 0.8498924970626831, "risk": 0.7783848643302917, "AP": 0.208845908158144 }, { "epoch": 36, "total": 1.0445747375488281, "denoise": 0.7990099191665649, "risk": 0.7016138434410095, "AP": 0.21511761960787368 }, { "epoch": 37, "total": 1.042030930519104, "denoise": 0.8102373480796814, "risk": 0.6622673869132996, "AP": 0.2135167409595423 }, { "epoch": 38, "total": 1.0674238204956055, "denoise": 0.8170891404151917, "risk": 0.7152420878410339, "AP": 0.21267205835790032 }, { "epoch": 39, "total": 1.0606675148010254, "denoise": 0.8080031871795654, "risk": 0.7218981385231018, "AP": 0.2129945974393907 }, { "epoch": 40, "total": 1.0620887279510498, "denoise": 0.8294333815574646, "risk": 0.6647295355796814, "AP": 0.2131616064313138 }, { "epoch": 41, "total": 1.0358004570007324, "denoise": 0.807881772518158, "risk": 0.6511962413787842, "AP": 0.20997293074932416 }, { "epoch": 42, "total": 1.0143252611160278, "denoise": 0.768628716468811, "risk": 0.7019901275634766, "AP": 0.21054389390636447 }, { "epoch": 43, "total": 0.9840487241744995, "denoise": 0.7525882720947266, "risk": 0.6613155603408813, "AP": 0.2101121360136911 }, { "epoch": 44, "total": 0.9366557002067566, "denoise": 0.7176375389099121, "risk": 0.6257661581039429, "AP": 0.21014102778779822 }, { "epoch": 45, "total": 0.9681411385536194, "denoise": 0.7391074895858765, "risk": 0.6543818712234497, "AP": 0.21383722564595906 }, { "epoch": 46, "total": 0.9627876877784729, "denoise": 0.7376930117607117, "risk": 0.6431276202201843, "AP": 0.2098657068217812 }, { "epoch": 47, "total": 0.9178267121315002, "denoise": 0.706302285194397, "risk": 0.60435551404953, "AP": 0.2107684805702909 }, { "epoch": 48, "total": 0.8911339044570923, "denoise": 0.6728032231330872, "risk": 0.6238019466400146, "AP": 0.21002037824000838 }, { "epoch": 49, "total": 0.9184176325798035, "denoise": 0.7017846703529358, "risk": 0.6189513802528381, "AP": 0.20942926041573365 }, { "epoch": 50, "total": 0.8582228422164917, "denoise": 0.6545680165290833, "risk": 0.581870973110199, "AP": 0.2101908935558669 }, { "epoch": 51, "total": 0.8922300338745117, "denoise": 0.6821738481521606, "risk": 0.6001604795455933, "AP": 0.2104193331459801 }, { "epoch": 52, "total": 0.8753014206886292, "denoise": 0.6703207492828369, "risk": 0.5856590867042542, "AP": 0.20934669772862124 }, { "epoch": 53, "total": 0.8576699495315552, "denoise": 0.660634458065033, "risk": 0.5629584789276123, "AP": 0.19821734841363567 }, { "epoch": 54, "total": 0.9557367563247681, "denoise": 0.7513405680656433, "risk": 0.5839890837669373, "AP": 0.14836806547193465 }, { "epoch": 55, "total": 0.9688800573348999, "denoise": 0.7761549949645996, "risk": 0.5506430268287659, "AP": 0.16688367831997086 }, { "epoch": 56, "total": 0.8277304768562317, "denoise": 0.623639702796936, "risk": 0.5831164717674255, "AP": 0.20333241010004682 }, { "epoch": 57, "total": 0.9455270767211914, "denoise": 0.7490316033363342, "risk": 0.5614156126976013, "AP": 0.1961515807732115 }, { "epoch": 58, "total": 0.7926274538040161, "denoise": 0.6018453240394592, "risk": 0.5450918078422546, "AP": 0.1420472688221222 }, { "epoch": 59, "total": 0.9008901715278625, "denoise": 0.705522894859314, "risk": 0.558192253112793, "AP": 0.13690001585080017 }, { "epoch": 60, "total": 0.7680805325508118, "denoise": 0.5828964114189148, "risk": 0.5290974974632263, "AP": 0.20290348270623415 }, { "epoch": 61, "total": 0.8288444876670837, "denoise": 0.6386324763298035, "risk": 0.5434629321098328, "AP": 0.20910022817571564 }, { "epoch": 62, "total": 0.789745569229126, "denoise": 0.6057345867156982, "risk": 0.5257457494735718, "AP": 0.2095054972262734 }, { "epoch": 63, "total": 0.8028594255447388, "denoise": 0.6217544674873352, "risk": 0.5174427032470703, "AP": 0.20190453186846544 }, { "epoch": 64, "total": 0.7877272367477417, "denoise": 0.6042625308036804, "risk": 0.5241848230361938, "AP": 0.1716428540286464 }, { "epoch": 65, "total": 0.7074911594390869, "denoise": 0.5304787158966064, "risk": 0.5057499408721924, "AP": 0.2074117104185537 }, { "epoch": 66, "total": 0.7791626453399658, "denoise": 0.5994712114334106, "risk": 0.5134040713310242, "AP": 0.21142874646694085 }, { "epoch": 67, "total": 0.7416664958000183, "denoise": 0.5673656463623047, "risk": 0.49800240993499756, "AP": 0.18347131968415595 }, { "epoch": 68, "total": 0.7143570780754089, "denoise": 0.5399942994117737, "risk": 0.4981793463230133, "AP": 0.17316720445961747 }, { "epoch": 69, "total": 0.7682361006736755, "denoise": 0.5966330766677856, "risk": 0.49029436707496643, "AP": 0.17409469656296486 }, { "epoch": 70, "total": 0.723753035068512, "denoise": 0.5542735457420349, "risk": 0.4842270612716675, "AP": 0.17457833414377963 }, { "epoch": 71, "total": 0.7114286422729492, "denoise": 0.5429192781448364, "risk": 0.48145538568496704, "AP": 0.17573391100955396 }, { "epoch": 72, "total": 0.7318010330200195, "denoise": 0.5669583678245544, "risk": 0.47097906470298767, "AP": 0.1710329245054079 }, { "epoch": 73, "total": 0.6962764859199524, "denoise": 0.5312442183494568, "risk": 0.4715207517147064, "AP": 0.17152781713659587 }, { "epoch": 74, "total": 0.6972377300262451, "denoise": 0.5356334447860718, "risk": 0.46172648668289185, "AP": 0.1724416869652543 }, { "epoch": 75, "total": 0.69074547290802, "denoise": 0.5298189520835876, "risk": 0.45979011058807373, "AP": 0.1727273121889774 }, { "epoch": 76, "total": 0.6877732276916504, "denoise": 0.5293176174163818, "risk": 0.4527302384376526, "AP": 0.1762986494227477 }, { "epoch": 77, "total": 0.6514995694160461, "denoise": 0.4940429627895355, "risk": 0.449876070022583, "AP": 0.17728034289075748 }, { "epoch": 78, "total": 0.671188235282898, "denoise": 0.5149269700050354, "risk": 0.4464607238769531, "AP": 0.1772708025007051 }, { "epoch": 79, "total": 0.6567543148994446, "denoise": 0.5023221373558044, "risk": 0.441234827041626, "AP": 0.17755217390320172 }, { "epoch": 80, "total": 0.6186112761497498, "denoise": 0.4648759663105011, "risk": 0.43924373388290405, "AP": 0.18006191358522414 }, { "epoch": 81, "total": 0.6433549523353577, "denoise": 0.49141979217529297, "risk": 0.4341004490852356, "AP": 0.17968260165432617 }, { "epoch": 82, "total": 0.6385584473609924, "denoise": 0.4877927005290985, "risk": 0.4307592809200287, "AP": 0.17979013695423354 }, { "epoch": 83, "total": 0.6079041361808777, "denoise": 0.4576652944087982, "risk": 0.42925381660461426, "AP": 0.17830027470777812 }, { "epoch": 84, "total": 0.6266117691993713, "denoise": 0.47822877764701843, "risk": 0.4239513874053955, "AP": 0.18104285516525453 }, { "epoch": 85, "total": 0.6259083151817322, "denoise": 0.47874727845191956, "risk": 0.42046013474464417, "AP": 0.18301682136525643 }, { "epoch": 86, "total": 0.6214589476585388, "denoise": 0.47507670521736145, "risk": 0.4182349443435669, "AP": 0.18285321978524557 }, { "epoch": 87, "total": 0.5800138711929321, "denoise": 0.4345996081829071, "risk": 0.41546931862831116, "AP": 0.1854230886548485 }, { "epoch": 88, "total": 0.5734130144119263, "denoise": 0.4295222759246826, "risk": 0.41111648082733154, "AP": 0.18637930135239994 }, { "epoch": 89, "total": 0.6032962799072266, "denoise": 0.460582971572876, "risk": 0.40775227546691895, "AP": 0.185608581074485 }, { "epoch": 90, "total": 0.6370022296905518, "denoise": 0.49495160579681396, "risk": 0.40585893392562866, "AP": 0.18687700973981086 }, { "epoch": 91, "total": 0.7011162042617798, "denoise": 0.5600152015686035, "risk": 0.40314581990242004, "AP": 0.18563502475004962 }, { "epoch": 92, "total": 0.8702249526977539, "denoise": 0.7303932905197144, "risk": 0.3995189964771271, "AP": 0.18566957507157153 }, { "epoch": 93, "total": 0.7345631122589111, "denoise": 0.5928881168365479, "risk": 0.40478578209877014, "AP": 0.18571655847440477 }, { "epoch": 94, "total": 0.6232099533081055, "denoise": 0.4784429967403412, "risk": 0.41361990571022034, "AP": 0.1888656720460001 }, { "epoch": 95, "total": 0.7193301916122437, "denoise": 0.5723949074745178, "risk": 0.4198151230812073, "AP": 0.18238423128936218 }, { "epoch": 96, "total": 0.592594563961029, "denoise": 0.45351383090019226, "risk": 0.3973734676837921, "AP": 0.18433541355981403 }, { "epoch": 97, "total": 0.679103672504425, "denoise": 0.5218023061752319, "risk": 0.44943252205848694, "AP": 0.1811352799477746 }, { "epoch": 98, "total": 0.6346865892410278, "denoise": 0.4541058838367462, "risk": 0.5159448981285095, "AP": 0.2249573180041263 }, { "epoch": 99, "total": 0.6333386898040771, "denoise": 0.4688662886619568, "risk": 0.4699210524559021, "AP": 0.2287322887426136 }, { "epoch": 100, "total": 0.6461944580078125, "denoise": 0.4765394330024719, "risk": 0.48472854495048523, "AP": 0.1833587797084288 }, { "epoch": 101, "total": 0.5865598320960999, "denoise": 0.4413124918937683, "risk": 0.41499245166778564, "AP": 0.23106527530771148 }, { "epoch": 102, "total": 0.6033739447593689, "denoise": 0.4477779269218445, "risk": 0.4445600211620331, "AP": 0.23412661667189683 }, { "epoch": 103, "total": 0.5806222558021545, "denoise": 0.4436233937740326, "risk": 0.39142537117004395, "AP": 0.19350826838802154 }, { "epoch": 104, "total": 0.6194502115249634, "denoise": 0.461141973733902, "risk": 0.4523092210292816, "AP": 0.18253774492756042 }, { "epoch": 105, "total": 0.6008618474006653, "denoise": 0.4471878111362457, "risk": 0.4390687346458435, "AP": 0.23224659300989728 }, { "epoch": 106, "total": 0.6363060474395752, "denoise": 0.4746946096420288, "risk": 0.46174687147140503, "AP": 0.23529844757428567 }, { "epoch": 107, "total": 0.573140025138855, "denoise": 0.4374234676361084, "risk": 0.3877616226673126, "AP": 0.19610403861693299 }, { "epoch": 108, "total": 0.6062776446342468, "denoise": 0.43632668256759644, "risk": 0.4855741262435913, "AP": 0.18375350985554206 }, { "epoch": 109, "total": 0.6121693849563599, "denoise": 0.45233097672462463, "risk": 0.45668119192123413, "AP": 0.2320561476050164 }, { "epoch": 110, "total": 0.5878684520721436, "denoise": 0.40638864040374756, "risk": 0.5185137391090393, "AP": 0.2329045379584056 }, { "epoch": 111, "total": 0.5987398624420166, "denoise": 0.4611021876335144, "risk": 0.393250435590744, "AP": 0.20756494016964297 }, { "epoch": 112, "total": 0.6285727620124817, "denoise": 0.4275887608528137, "risk": 0.5742400288581848, "AP": 0.17463774252170064 }, { "epoch": 113, "total": 0.5639793872833252, "denoise": 0.407604843378067, "risk": 0.4467844069004059, "AP": 0.23342648902350055 }, { "epoch": 114, "total": 0.6370601058006287, "denoise": 0.43347516655921936, "risk": 0.5816712975502014, "AP": 0.23308538077667867 }, { "epoch": 115, "total": 0.5994999408721924, "denoise": 0.4398154616355896, "risk": 0.4562413692474365, "AP": 0.23384991148222437 }, { "epoch": 116, "total": 0.5702406167984009, "denoise": 0.4205118715763092, "risk": 0.4277963638305664, "AP": 0.19369976894281393 }, { "epoch": 117, "total": 0.5919268131256104, "denoise": 0.4352331757545471, "risk": 0.4476962089538574, "AP": 0.19398863632217017 }, { "epoch": 118, "total": 0.5476773381233215, "denoise": 0.399777352809906, "risk": 0.4225713610649109, "AP": 0.2347320702225249 }, { "epoch": 119, "total": 0.6075854301452637, "denoise": 0.4331755042076111, "risk": 0.4983140528202057, "AP": 0.23467333127356058 }, { "epoch": 120, "total": 0.5927420854568481, "denoise": 0.4487384855747223, "risk": 0.41143882274627686, "AP": 0.23567163421869908 } ], "best_loss": 0.5476773381233215, "best_epoch": 118, "metrics": [ { "run": "14h06", "MAE_mm": 6.102814674377441, "RMSE_mm": 7.919643878936768, "CAD_correspondent_Chamfer_mm": 5.854639530181885, "registered_scan_Chamfer_mm": 32.36278533935547, "risk": { "AP": 0.23467333127356058, "IoU_at_0.5": 0.05660377358490566, "Dice_at_0.5": 0.10714285714285714 }, "mean_stochastic_std_mm": 9.036323547363281, "max_stochastic_std_mm": 25.99744415283203 } ], "heater_projection_valid": [ false ], "checkpoint": "gen_models/models/voxel_3d.pt" }, { "variant": "point_6000", "representation": { "source": "data_processed/points/cad_surface_points.npz", "representation": "deterministic 6,000 CAD surface points with PointNet-style local/global pooling", "nodes": 6000, "denoiser_adapter": "8-nearest local pooling plus per-run global max/mean context" }, "nodes_per_run": 6000, "condition_dim": 186, "history": [ { "epoch": 1, "total": 1.4303454160690308, "denoise": 1.0033766031265259, "risk": 1.2199108600616455, "AP": 0.0024512377724582772 }, { "epoch": 2, "total": 1.4988347291946411, "denoise": 1.0749220848083496, "risk": 1.2111788988113403, "AP": 0.005540204044438533 }, { "epoch": 3, "total": 1.432555913925171, "denoise": 1.0090886354446411, "risk": 1.2099064588546753, "AP": 0.016427928738660984 }, { "epoch": 4, "total": 1.4167020320892334, "denoise": 0.9938035607337952, "risk": 1.20828115940094, "AP": 0.05978004811395533 }, { "epoch": 5, "total": 1.4143028259277344, "denoise": 0.9919900894165039, "risk": 1.206607699394226, "AP": 0.0631473398160655 }, { "epoch": 6, "total": 1.3726447820663452, "denoise": 0.9508572816848755, "risk": 1.2051072120666504, "AP": 0.06753666677609853 }, { "epoch": 7, "total": 1.4127120971679688, "denoise": 0.991544246673584, "risk": 1.2033367156982422, "AP": 0.06718376618404595 }, { "epoch": 8, "total": 1.4269111156463623, "denoise": 1.0066962242126465, "risk": 1.2006138563156128, "AP": 0.06447200574622855 }, { "epoch": 9, "total": 1.4181573390960693, "denoise": 0.9990720748901367, "risk": 1.1973865032196045, "AP": 0.06407400435605007 }, { "epoch": 10, "total": 1.4116754531860352, "denoise": 0.9938108921051025, "risk": 1.1938986778259277, "AP": 0.06339060304636207 }, { "epoch": 11, "total": 1.3875826597213745, "denoise": 0.9712775945663452, "risk": 1.1894429922103882, "AP": 0.06176489103009306 }, { "epoch": 12, "total": 1.375313401222229, "denoise": 0.9612364768981934, "risk": 1.1830769777297974, "AP": 0.060402153699908964 }, { "epoch": 13, "total": 1.3732942342758179, "denoise": 0.9618743658065796, "risk": 1.175485372543335, "AP": 0.057999133652140454 }, { "epoch": 14, "total": 1.3998439311981201, "denoise": 0.9911143183708191, "risk": 1.1677991151809692, "AP": 0.0568704966188954 }, { "epoch": 15, "total": 1.3565126657485962, "denoise": 0.9510322213172913, "risk": 1.1585156917572021, "AP": 0.0548008034077921 }, { "epoch": 16, "total": 1.3476743698120117, "denoise": 0.947157084941864, "risk": 1.1443349123001099, "AP": 0.05301731840201929 }, { "epoch": 17, "total": 1.3058432340621948, "denoise": 0.9113146066665649, "risk": 1.1272246837615967, "AP": 0.05127921063561416 }, { "epoch": 18, "total": 1.3144911527633667, "denoise": 0.9242801666259766, "risk": 1.1148884296417236, "AP": 0.05000616393295953 }, { "epoch": 19, "total": 1.3451403379440308, "denoise": 0.9633813500404358, "risk": 1.0907399654388428, "AP": 0.04906662417344544 }, { "epoch": 20, "total": 1.288265347480774, "denoise": 0.9134887456893921, "risk": 1.070790410041809, "AP": 0.04862446502369561 }, { "epoch": 21, "total": 1.274186134338379, "denoise": 0.9066075086593628, "risk": 1.0502246618270874, "AP": 0.04628693837349535 }, { "epoch": 22, "total": 1.256981611251831, "denoise": 0.8992335796356201, "risk": 1.0221374034881592, "AP": 0.042636768827659335 }, { "epoch": 23, "total": 1.2555859088897705, "denoise": 0.906475841999054, "risk": 0.9974575042724609, "AP": 0.03933569680605828 }, { "epoch": 24, "total": 1.2484989166259766, "denoise": 0.9071022272109985, "risk": 0.9754191040992737, "AP": 0.037647321630369664 }, { "epoch": 25, "total": 1.2208528518676758, "denoise": 0.8863961696624756, "risk": 0.9555904269218445, "AP": 0.03465507849402457 }, { "epoch": 26, "total": 1.2053747177124023, "denoise": 0.8769931793212891, "risk": 0.9382330179214478, "AP": 0.03315817467634964 }, { "epoch": 27, "total": 1.207515001296997, "denoise": 0.8819360136985779, "risk": 0.9302257299423218, "AP": 0.031086954593503417 }, { "epoch": 28, "total": 1.2112396955490112, "denoise": 0.8759841918945312, "risk": 0.9578728675842285, "AP": 0.022514326946050812 }, { "epoch": 29, "total": 1.3116750717163086, "denoise": 0.9359742999076843, "risk": 1.0734307765960693, "AP": 0.02483383271232693 }, { "epoch": 30, "total": 1.1763184070587158, "denoise": 0.85281902551651, "risk": 0.9242839217185974, "AP": 0.021836761245454733 }, { "epoch": 31, "total": 1.2240043878555298, "denoise": 0.8680593967437744, "risk": 1.0169856548309326, "AP": 0.023372109559246963 }, { "epoch": 32, "total": 1.1535980701446533, "denoise": 0.8401039242744446, "risk": 0.8956975936889648, "AP": 0.03232469815367805 }, { "epoch": 33, "total": 1.1780623197555542, "denoise": 0.847967267036438, "risk": 0.943128764629364, "AP": 0.04680755373892921 }, { "epoch": 34, "total": 1.1434557437896729, "denoise": 0.8078530430793762, "risk": 0.9588650465011597, "AP": 0.06493736401833684 }, { "epoch": 35, "total": 1.1400690078735352, "denoise": 0.8331466317176819, "risk": 0.8769208788871765, "AP": 0.07304399348617208 }, { "epoch": 36, "total": 1.1123251914978027, "denoise": 0.7934041619300842, "risk": 0.9112029075622559, "AP": 0.06601387366784223 }, { "epoch": 37, "total": 1.130905270576477, "denoise": 0.79312664270401, "risk": 0.9650816917419434, "AP": 0.07217452967122509 }, { "epoch": 38, "total": 1.1220331192016602, "denoise": 0.808255672454834, "risk": 0.8965070843696594, "AP": 0.0868614972505507 }, { "epoch": 39, "total": 1.0780190229415894, "denoise": 0.7749812602996826, "risk": 0.8658220767974854, "AP": 0.08828697956749847 }, { "epoch": 40, "total": 1.0821735858917236, "denoise": 0.7630963325500488, "risk": 0.911649227142334, "AP": 0.09254793966836111 }, { "epoch": 41, "total": 1.0863499641418457, "denoise": 0.7714077234268188, "risk": 0.8998349905014038, "AP": 0.08670320394333711 }, { "epoch": 42, "total": 1.050689458847046, "denoise": 0.7519939541816711, "risk": 0.8534159064292908, "AP": 0.08075373187829915 }, { "epoch": 43, "total": 1.0056997537612915, "denoise": 0.7081338763237, "risk": 0.8501882553100586, "AP": 0.07641024224229377 }, { "epoch": 44, "total": 1.0438601970672607, "denoise": 0.739401638507843, "risk": 0.8698814511299133, "AP": 0.06628193438285006 }, { "epoch": 45, "total": 1.0059372186660767, "denoise": 0.7024017572402954, "risk": 0.8672442436218262, "AP": 0.06583304375065285 }, { "epoch": 46, "total": 0.9803600907325745, "denoise": 0.6908277869224548, "risk": 0.8272351622581482, "AP": 0.07154692893928245 }, { "epoch": 47, "total": 0.9666829109191895, "denoise": 0.6764408946037292, "risk": 0.8292628526687622, "AP": 0.07322319859333394 }, { "epoch": 48, "total": 0.9813739061355591, "denoise": 0.6896573305130005, "risk": 0.8334758877754211, "AP": 0.07104938987209322 }, { "epoch": 49, "total": 0.9564456939697266, "denoise": 0.6702752113342285, "risk": 0.8176300525665283, "AP": 0.0681342936549268 }, { "epoch": 50, "total": 0.9050177335739136, "denoise": 0.6252714395523071, "risk": 0.7992752194404602, "AP": 0.06274239760394486 }, { "epoch": 51, "total": 1.0120043754577637, "denoise": 0.7323212623596191, "risk": 0.7990947961807251, "AP": 0.06173483397358042 }, { "epoch": 52, "total": 1.168778419494629, "denoise": 0.8688541054725647, "risk": 0.856926679611206, "AP": 0.057632115056770544 }, { "epoch": 53, "total": 1.0155012607574463, "denoise": 0.737895667552948, "risk": 0.793159008026123, "AP": 0.06238090581337858 }, { "epoch": 54, "total": 0.9676533937454224, "denoise": 0.6806559562683105, "risk": 0.8199927806854248, "AP": 0.06244719431013089 }, { "epoch": 55, "total": 0.9392995834350586, "denoise": 0.6663766503334045, "risk": 0.779779851436615, "AP": 0.0585489777251636 }, { "epoch": 56, "total": 0.9642559289932251, "denoise": 0.6938263177871704, "risk": 0.7726560235023499, "AP": 0.05759460445815081 }, { "epoch": 57, "total": 0.8736177682876587, "denoise": 0.6041725873947144, "risk": 0.7698433995246887, "AP": 0.055199353881513466 }, { "epoch": 58, "total": 0.9468649625778198, "denoise": 0.678102433681488, "risk": 0.7678928971290588, "AP": 0.05425507671121142 }, { "epoch": 59, "total": 0.8485587239265442, "denoise": 0.5798720121383667, "risk": 0.7676762938499451, "AP": 0.05296931304447527 }, { "epoch": 60, "total": 0.865363359451294, "denoise": 0.5978623628616333, "risk": 0.7642884850502014, "AP": 0.05335013810556353 }, { "epoch": 61, "total": 0.8990310430526733, "denoise": 0.6339797973632812, "risk": 0.7572892904281616, "AP": 0.056240490050250336 }, { "epoch": 62, "total": 0.8358898162841797, "denoise": 0.5709647536277771, "risk": 0.7569288015365601, "AP": 0.05871013654775619 }, { "epoch": 63, "total": 0.8442751169204712, "denoise": 0.579319179058075, "risk": 0.7570170164108276, "AP": 0.059177808018360416 }, { "epoch": 64, "total": 0.858479380607605, "denoise": 0.596734344959259, "risk": 0.7478430867195129, "AP": 0.05952242795428412 }, { "epoch": 65, "total": 0.8334137797355652, "denoise": 0.5741318464279175, "risk": 0.7408055663108826, "AP": 0.05827951447203665 }, { "epoch": 66, "total": 0.8293554782867432, "denoise": 0.5693208575248718, "risk": 0.7429559826850891, "AP": 0.056106648515338226 }, { "epoch": 67, "total": 0.8410713076591492, "denoise": 0.582412838935852, "risk": 0.7390242218971252, "AP": 0.055618518429504764 }, { "epoch": 68, "total": 0.7963957786560059, "denoise": 0.5417574644088745, "risk": 0.7275379300117493, "AP": 0.0556523352350687 }, { "epoch": 69, "total": 0.803899884223938, "denoise": 0.550584614276886, "risk": 0.7237580418586731, "AP": 0.05738175248348607 }, { "epoch": 70, "total": 0.809363603591919, "denoise": 0.5563755035400391, "risk": 0.7228230834007263, "AP": 0.05560588936323973 }, { "epoch": 71, "total": 0.7925573587417603, "denoise": 0.5424483418464661, "risk": 0.714597225189209, "AP": 0.05293854406677464 }, { "epoch": 72, "total": 0.7760789394378662, "denoise": 0.5288936495780945, "risk": 0.7062437534332275, "AP": 0.05184804320721608 }, { "epoch": 73, "total": 0.7722304463386536, "denoise": 0.5250722765922546, "risk": 0.7061662673950195, "AP": 0.050429001288968864 }, { "epoch": 74, "total": 0.7468852400779724, "denoise": 0.5009888410568237, "risk": 0.7025611400604248, "AP": 0.04941973114376442 }, { "epoch": 75, "total": 0.7536944150924683, "denoise": 0.5115852952003479, "risk": 0.6917403936386108, "AP": 0.04965633601381015 }, { "epoch": 76, "total": 0.7412203550338745, "denoise": 0.4994766414165497, "risk": 0.6906962990760803, "AP": 0.05041165249788542 }, { "epoch": 77, "total": 0.7221927642822266, "denoise": 0.4810144007205963, "risk": 0.6890811324119568, "AP": 0.05022291003441778 }, { "epoch": 78, "total": 0.7147496342658997, "denoise": 0.4781278967857361, "risk": 0.6760621070861816, "AP": 0.049796940221842494 }, { "epoch": 79, "total": 0.6967272162437439, "denoise": 0.4588814973831177, "risk": 0.6795591711997986, "AP": 0.04937180540518061 }, { "epoch": 80, "total": 0.7192787528038025, "denoise": 0.4844149649143219, "risk": 0.6710394620895386, "AP": 0.05019998341047696 }, { "epoch": 81, "total": 0.6868427991867065, "denoise": 0.4551480710506439, "risk": 0.6619850397109985, "AP": 0.05105806430248188 }, { "epoch": 82, "total": 0.6998499631881714, "denoise": 0.4675399959087372, "risk": 0.6637428402900696, "AP": 0.05177477755879561 }, { "epoch": 83, "total": 0.6781710982322693, "denoise": 0.4513966739177704, "risk": 0.6479269862174988, "AP": 0.052099355628540754 }, { "epoch": 84, "total": 0.6721903085708618, "denoise": 0.4441952407360077, "risk": 0.6514144539833069, "AP": 0.05231532563402879 }, { "epoch": 85, "total": 0.6914429664611816, "denoise": 0.4689948856830597, "risk": 0.635565996170044, "AP": 0.05328682166921006 }, { "epoch": 86, "total": 0.6654066443443298, "denoise": 0.4433601498603821, "risk": 0.6344186067581177, "AP": 0.05450793709623312 }, { "epoch": 87, "total": 0.668498158454895, "denoise": 0.4501931071281433, "risk": 0.6237286925315857, "AP": 0.054658214356982204 }, { "epoch": 88, "total": 0.6642098426818848, "denoise": 0.4472787380218506, "risk": 0.619803249835968, "AP": 0.05540531515678849 }, { "epoch": 89, "total": 0.667754054069519, "denoise": 0.45346125960350037, "risk": 0.6122651100158691, "AP": 0.05637875226304041 }, { "epoch": 90, "total": 0.6685429811477661, "denoise": 0.45587989687919617, "risk": 0.6076087355613708, "AP": 0.05890245803377962 }, { "epoch": 91, "total": 0.6519207954406738, "denoise": 0.4416085183620453, "risk": 0.600892186164856, "AP": 0.059200817843078214 }, { "epoch": 92, "total": 0.636655330657959, "denoise": 0.4262009263038635, "risk": 0.6012983322143555, "AP": 0.0601444307711039 }, { "epoch": 93, "total": 0.630137026309967, "denoise": 0.42354288697242737, "risk": 0.5902690291404724, "AP": 0.06131065086934679 }, { "epoch": 94, "total": 0.6446961164474487, "denoise": 0.44007861614227295, "risk": 0.5846214294433594, "AP": 0.062355288024586175 }, { "epoch": 95, "total": 0.6410250663757324, "denoise": 0.43607237935066223, "risk": 0.5855790972709656, "AP": 0.062314676048348144 }, { "epoch": 96, "total": 0.6356361508369446, "denoise": 0.43352824449539185, "risk": 0.5774511098861694, "AP": 0.06310380142732824 }, { "epoch": 97, "total": 0.6484822034835815, "denoise": 0.4504840075969696, "risk": 0.5657091736793518, "AP": 0.062488804121907604 }, { "epoch": 98, "total": 0.6617113351821899, "denoise": 0.46519312262535095, "risk": 0.5614806413650513, "AP": 0.06299629450241138 }, { "epoch": 99, "total": 0.658574104309082, "denoise": 0.4607103765010834, "risk": 0.5653249621391296, "AP": 0.06161944383032153 }, { "epoch": 100, "total": 0.630305290222168, "denoise": 0.43176546692848206, "risk": 0.5672566890716553, "AP": 0.060776940397425504 }, { "epoch": 101, "total": 0.6191713213920593, "denoise": 0.4203258454799652, "risk": 0.5681299567222595, "AP": 0.06075339867948704 }, { "epoch": 102, "total": 0.6115893125534058, "denoise": 0.41800224781036377, "risk": 0.5531060099601746, "AP": 0.06009500245902595 }, { "epoch": 103, "total": 0.6363639831542969, "denoise": 0.44699445366859436, "risk": 0.5410557985305786, "AP": 0.059369732803345204 }, { "epoch": 104, "total": 0.6078284978866577, "denoise": 0.42092186212539673, "risk": 0.5340189933776855, "AP": 0.05817751301834011 }, { "epoch": 105, "total": 0.615562379360199, "denoise": 0.430131196975708, "risk": 0.5298033356666565, "AP": 0.057174537377220815 }, { "epoch": 106, "total": 0.6013708114624023, "denoise": 0.4171910285949707, "risk": 0.5262280106544495, "AP": 0.05662752688320539 }, { "epoch": 107, "total": 0.6317708492279053, "denoise": 0.4488755762577057, "risk": 0.5225579738616943, "AP": 0.05657734843929001 }, { "epoch": 108, "total": 0.5930920839309692, "denoise": 0.4112316370010376, "risk": 0.5196012258529663, "AP": 0.05696258272663374 }, { "epoch": 109, "total": 0.5839780569076538, "denoise": 0.4021531641483307, "risk": 0.5194995999336243, "AP": 0.05691440891255065 }, { "epoch": 110, "total": 0.6242893934249878, "denoise": 0.43997424840927124, "risk": 0.5266146063804626, "AP": 0.05760422560815143 }, { "epoch": 111, "total": 0.576433539390564, "denoise": 0.38645288348197937, "risk": 0.5428019762039185, "AP": 0.05692909757245594 }, { "epoch": 112, "total": 0.6511035561561584, "denoise": 0.4365576505661011, "risk": 0.6129883527755737, "AP": 0.057655521830424085 }, { "epoch": 113, "total": 0.6535975933074951, "denoise": 0.41770467162132263, "risk": 0.6739798784255981, "AP": 0.058398811244858365 }, { "epoch": 114, "total": 0.6629513502120972, "denoise": 0.4205568730831146, "risk": 0.6925557255744934, "AP": 0.06037023501760153 }, { "epoch": 115, "total": 0.5977075695991516, "denoise": 0.4202733635902405, "risk": 0.5069549083709717, "AP": 0.06167820926344928 }, { "epoch": 116, "total": 0.6518492698669434, "denoise": 0.4096759557723999, "risk": 0.6919238567352295, "AP": 0.06208095155176118 }, { "epoch": 117, "total": 0.6214041113853455, "denoise": 0.41065865755081177, "risk": 0.6021298170089722, "AP": 0.060907342538932055 }, { "epoch": 118, "total": 0.621854305267334, "denoise": 0.4148244261741638, "risk": 0.5915140509605408, "AP": 0.059662277359114016 }, { "epoch": 119, "total": 0.6204346418380737, "denoise": 0.4173619747161865, "risk": 0.5802077054977417, "AP": 0.05856950956607201 }, { "epoch": 120, "total": 0.5959979295730591, "denoise": 0.40819767117500305, "risk": 0.536572277545929, "AP": 0.057974212602282665 } ], "best_loss": 0.576433539390564, "best_epoch": 111, "metrics": [ { "run": "14h06", "MAE_mm": 5.645156383514404, "RMSE_mm": 7.401700019836426, "CAD_correspondent_Chamfer_mm": 5.1742753982543945, "registered_scan_Chamfer_mm": 33.930809020996094, "risk": { "AP": 0.057655521830424085, "IoU_at_0.5": 0.011126564673157162, "Dice_at_0.5": 0.02200825309491059 }, "mean_stochastic_std_mm": 8.314737319946289, "max_stochastic_std_mm": 24.16046142578125 } ], "heater_projection_valid": [ false ], "checkpoint": "gen_models/models/point_6000.pt" }, { "variant": "graph_6000", "representation": { "source": "data_processed/graphs/graph_6000.npz", "representation": "CAD graph nodes (6,155 nodes) with edge-index message passing at each denoise step", "nodes": 6155, "denoiser_adapter": "prepared graph edge-index, up to 8 adjacent CAD nodes per message" }, "nodes_per_run": 6155, "condition_dim": 224, "history": [ { "epoch": 1, "total": 1.3622430562973022, "denoise": 1.0575857162475586, "risk": 0.8704496026039124, "AP": 0.006109439654888764 }, { "epoch": 2, "total": 1.2721385955810547, "denoise": 0.9818516969680786, "risk": 0.8293911814689636, "AP": 0.006079638318960563 }, { "epoch": 3, "total": 1.2797516584396362, "denoise": 1.007378101348877, "risk": 0.7782102227210999, "AP": 0.009812777937762219 }, { "epoch": 4, "total": 1.2651751041412354, "denoise": 1.0153805017471313, "risk": 0.7136988639831543, "AP": 0.015796882977318377 }, { "epoch": 5, "total": 1.2432048320770264, "denoise": 1.0170047283172607, "risk": 0.6462860703468323, "AP": 0.023787395518558112 }, { "epoch": 6, "total": 1.2061634063720703, "denoise": 0.9988086223602295, "risk": 0.5924420952796936, "AP": 0.0422785922226739 }, { "epoch": 7, "total": 1.1902024745941162, "denoise": 0.9895399212837219, "risk": 0.5733214616775513, "AP": 0.04660588020007793 }, { "epoch": 8, "total": 1.2377469539642334, "denoise": 1.0388740301132202, "risk": 0.5682083368301392, "AP": 0.056101223811278134 }, { "epoch": 9, "total": 1.1379653215408325, "denoise": 0.949887216091156, "risk": 0.5373659729957581, "AP": 0.05652690205636373 }, { "epoch": 10, "total": 1.143058180809021, "denoise": 0.9717174768447876, "risk": 0.4895447790622711, "AP": 0.06494215792638809 }, { "epoch": 11, "total": 1.1213443279266357, "denoise": 0.9660161137580872, "risk": 0.4437948763370514, "AP": 0.06380007061298923 }, { "epoch": 12, "total": 1.116740107536316, "denoise": 0.969942033290863, "risk": 0.4194229543209076, "AP": 0.07091091451618221 }, { "epoch": 13, "total": 1.099217176437378, "denoise": 0.9546094536781311, "risk": 0.413164883852005, "AP": 0.07540616563714972 }, { "epoch": 14, "total": 1.1113845109939575, "denoise": 0.9706748723983765, "risk": 0.40202754735946655, "AP": 0.07618157736537069 }, { "epoch": 15, "total": 1.0819576978683472, "denoise": 0.9515623450279236, "risk": 0.3725580871105194, "AP": 0.07032906382324725 }, { "epoch": 16, "total": 1.0287379026412964, "denoise": 0.9115642309188843, "risk": 0.3347819745540619, "AP": 0.07161348618564638 }, { "epoch": 17, "total": 1.0627973079681396, "denoise": 0.9534679651260376, "risk": 0.31236955523490906, "AP": 0.07031230246780254 }, { "epoch": 18, "total": 1.0283257961273193, "denoise": 0.9181848764419556, "risk": 0.31468823552131653, "AP": 0.06932365450126074 }, { "epoch": 19, "total": 1.0484142303466797, "denoise": 0.9363691210746765, "risk": 0.320128858089447, "AP": 0.06634203590446079 }, { "epoch": 20, "total": 1.0284558534622192, "denoise": 0.9217967987060547, "risk": 0.30474013090133667, "AP": 0.06647191929937254 }, { "epoch": 21, "total": 1.035487413406372, "denoise": 0.9346731901168823, "risk": 0.28804048895835876, "AP": 0.05319815811693129 }, { "epoch": 22, "total": 1.033353328704834, "denoise": 0.9335987567901611, "risk": 0.2850131690502167, "AP": 0.050008703208269896 }, { "epoch": 23, "total": 0.9957700371742249, "denoise": 0.8945755362510681, "risk": 0.2891271412372589, "AP": 0.046962836421359615 }, { "epoch": 24, "total": 1.0120247602462769, "denoise": 0.9142363667488098, "risk": 0.27939552068710327, "AP": 0.04626729814480833 }, { "epoch": 25, "total": 0.9634511470794678, "denoise": 0.8713772892951965, "risk": 0.2630682587623596, "AP": 0.044601441503354655 }, { "epoch": 26, "total": 0.953665554523468, "denoise": 0.8637270331382751, "risk": 0.2569672763347626, "AP": 0.042872105664852084 }, { "epoch": 27, "total": 0.9505267143249512, "denoise": 0.8599738478660583, "risk": 0.2587225139141083, "AP": 0.042180610235903554 }, { "epoch": 28, "total": 0.9845650792121887, "denoise": 0.8950294256210327, "risk": 0.2558162212371826, "AP": 0.04153245897272824 }, { "epoch": 29, "total": 0.9537997245788574, "denoise": 0.8685200810432434, "risk": 0.243656188249588, "AP": 0.04051233017313343 }, { "epoch": 30, "total": 0.9700037837028503, "denoise": 0.8866367340087891, "risk": 0.23819153010845184, "AP": 0.040670640320963354 }, { "epoch": 31, "total": 0.9432147145271301, "denoise": 0.8582944273948669, "risk": 0.24262937903404236, "AP": 0.04160828560368053 }, { "epoch": 32, "total": 0.9460359215736389, "denoise": 0.8606322407722473, "risk": 0.24401046335697174, "AP": 0.04044372313154357 }, { "epoch": 33, "total": 0.9299391508102417, "denoise": 0.8464394807815552, "risk": 0.23857052624225616, "AP": 0.039527704199398515 }, { "epoch": 34, "total": 0.9227151274681091, "denoise": 0.8424409627914429, "risk": 0.22935475409030914, "AP": 0.03831771447146459 }, { "epoch": 35, "total": 0.9026635885238647, "denoise": 0.8233137726783752, "risk": 0.22671380639076233, "AP": 0.037812280490463125 }, { "epoch": 36, "total": 0.9151114225387573, "denoise": 0.8352468609809875, "risk": 0.22818449139595032, "AP": 0.03624446734832788 }, { "epoch": 37, "total": 0.8693845868110657, "denoise": 0.7901346683502197, "risk": 0.22642827033996582, "AP": 0.03523231230557129 }, { "epoch": 38, "total": 0.9002676606178284, "denoise": 0.8233981728553772, "risk": 0.2196270376443863, "AP": 0.034763350485234726 }, { "epoch": 39, "total": 0.8753076791763306, "denoise": 0.8015879392623901, "risk": 0.2106277495622635, "AP": 0.03495358771682598 }, { "epoch": 40, "total": 0.8776082992553711, "denoise": 0.8045738935470581, "risk": 0.2086697518825531, "AP": 0.03522035632675843 }, { "epoch": 41, "total": 0.8085145950317383, "denoise": 0.7362778186798096, "risk": 0.20639073848724365, "AP": 0.0358195059177709 }, { "epoch": 42, "total": 0.814893901348114, "denoise": 0.7451919913291931, "risk": 0.19914834201335907, "AP": 0.035979454119869386 }, { "epoch": 43, "total": 0.8088994026184082, "denoise": 0.74062579870224, "risk": 0.19506752490997314, "AP": 0.036291668820868574 }, { "epoch": 44, "total": 0.8266016840934753, "denoise": 0.757524847984314, "risk": 0.19736236333847046, "AP": 0.036122062506730696 }, { "epoch": 45, "total": 0.8334083557128906, "denoise": 0.7655918598175049, "risk": 0.19376149773597717, "AP": 0.036222184985994804 }, { "epoch": 46, "total": 0.7765746712684631, "denoise": 0.7114235758781433, "risk": 0.18614594638347626, "AP": 0.036844901853398365 }, { "epoch": 47, "total": 0.7507259249687195, "denoise": 0.6844138503074646, "risk": 0.1894630640745163, "AP": 0.03880024660924934 }, { "epoch": 48, "total": 0.7826325297355652, "denoise": 0.718755841255188, "risk": 0.18250475823879242, "AP": 0.03954142190035575 }, { "epoch": 49, "total": 0.8460189700126648, "denoise": 0.7834591865539551, "risk": 0.1787421703338623, "AP": 0.03997050548245716 }, { "epoch": 50, "total": 0.7777506113052368, "denoise": 0.7156882882118225, "risk": 0.1773209422826767, "AP": 0.05150816404986549 }, { "epoch": 51, "total": 0.7322203516960144, "denoise": 0.6712017059326172, "risk": 0.1743389368057251, "AP": 0.051794669778692 }, { "epoch": 52, "total": 0.7993050813674927, "denoise": 0.7385315895080566, "risk": 0.1736384779214859, "AP": 0.051968634957446934 }, { "epoch": 53, "total": 0.727616012096405, "denoise": 0.6669477820396423, "risk": 0.17333781719207764, "AP": 0.05201123710552524 }, { "epoch": 54, "total": 0.7452256083488464, "denoise": 0.6847506165504456, "risk": 0.1727856993675232, "AP": 0.07255938094040343 }, { "epoch": 55, "total": 0.7127585411071777, "denoise": 0.6522677540779114, "risk": 0.17283084988594055, "AP": 0.09383528487927068 }, { "epoch": 56, "total": 0.7361831665039062, "denoise": 0.6761139035224915, "risk": 0.17162641882896423, "AP": 0.09412669710868467 }, { "epoch": 57, "total": 0.7160223722457886, "denoise": 0.656462550163269, "risk": 0.17017099261283875, "AP": 0.07296395718426765 }, { "epoch": 58, "total": 0.7096529006958008, "denoise": 0.6503483057022095, "risk": 0.16944172978401184, "AP": 0.07296423535322402 }, { "epoch": 59, "total": 0.7248417735099792, "denoise": 0.6659504175186157, "risk": 0.1682610660791397, "AP": 0.07340947030798456 }, { "epoch": 60, "total": 0.6402223110198975, "denoise": 0.5820029973983765, "risk": 0.1663409322500229, "AP": 0.07306992737206727 }, { "epoch": 61, "total": 0.6894484758377075, "denoise": 0.6312021017074585, "risk": 0.16641826927661896, "AP": 0.06274396481286271 }, { "epoch": 62, "total": 0.6568679809570312, "denoise": 0.5995588302612305, "risk": 0.1637403517961502, "AP": 0.0632469736022914 }, { "epoch": 63, "total": 0.6549729704856873, "denoise": 0.5979580879211426, "risk": 0.1628996729850769, "AP": 0.06375225171490381 }, { "epoch": 64, "total": 0.673527181148529, "denoise": 0.6168970465660095, "risk": 0.1618003249168396, "AP": 0.06411467147865209 }, { "epoch": 65, "total": 0.6235966086387634, "denoise": 0.5674193501472473, "risk": 0.16050638258457184, "AP": 0.06457873908004086 }, { "epoch": 66, "total": 0.651392936706543, "denoise": 0.5954408645629883, "risk": 0.15986303985118866, "AP": 0.06495196931814441 }, { "epoch": 67, "total": 0.6212344169616699, "denoise": 0.5658469200134277, "risk": 0.15824995934963226, "AP": 0.06531769334183421 }, { "epoch": 68, "total": 0.6048212051391602, "denoise": 0.5494900345802307, "risk": 0.15808908641338348, "AP": 0.06629950528532405 }, { "epoch": 69, "total": 0.6237319707870483, "denoise": 0.5691360831260681, "risk": 0.15598823130130768, "AP": 0.07725225558881335 }, { "epoch": 70, "total": 0.6592633128166199, "denoise": 0.6050338745117188, "risk": 0.15494124591350555, "AP": 0.07711814416558063 }, { "epoch": 71, "total": 0.6304014325141907, "denoise": 0.5762821435928345, "risk": 0.15462654829025269, "AP": 0.07817218356316381 }, { "epoch": 72, "total": 0.5737220644950867, "denoise": 0.5200382471084595, "risk": 0.15338227152824402, "AP": 0.08037429445859504 }, { "epoch": 73, "total": 0.5847928524017334, "denoise": 0.5313599705696106, "risk": 0.15266533195972443, "AP": 0.08157486361751835 }, { "epoch": 74, "total": 0.5905607342720032, "denoise": 0.537835955619812, "risk": 0.15064218640327454, "AP": 0.08115964709049815 }, { "epoch": 75, "total": 0.6065616011619568, "denoise": 0.55390465259552, "risk": 0.1504485011100769, "AP": 0.08075703040636463 }, { "epoch": 76, "total": 0.574176013469696, "denoise": 0.521716833114624, "risk": 0.1498834192752838, "AP": 0.0844869672559257 }, { "epoch": 77, "total": 0.5387800335884094, "denoise": 0.4863760769367218, "risk": 0.14972557127475739, "AP": 0.08491582440235015 }, { "epoch": 78, "total": 0.5580302476882935, "denoise": 0.5065231919288635, "risk": 0.1471630334854126, "AP": 0.10448680208787245 }, { "epoch": 79, "total": 0.5590803623199463, "denoise": 0.5068230032920837, "risk": 0.14930680394172668, "AP": 0.1019164556343623 }, { "epoch": 80, "total": 0.5392817854881287, "denoise": 0.48861709237098694, "risk": 0.14475630223751068, "AP": 0.10537608680059118 }, { "epoch": 81, "total": 0.5302267670631409, "denoise": 0.4787941575050354, "risk": 0.14695028960704803, "AP": 0.10867390817894575 }, { "epoch": 82, "total": 0.5278490781784058, "denoise": 0.4777948260307312, "risk": 0.1430121809244156, "AP": 0.1068229134169132 }, { "epoch": 83, "total": 0.5439032316207886, "denoise": 0.49348023533821106, "risk": 0.14406578242778778, "AP": 0.10299224158543475 }, { "epoch": 84, "total": 0.5134992599487305, "denoise": 0.46415576338768005, "risk": 0.14098136126995087, "AP": 0.1068661392303655 }, { "epoch": 85, "total": 0.5114679336547852, "denoise": 0.46171414852142334, "risk": 0.14215372502803802, "AP": 0.11204316728827471 }, { "epoch": 86, "total": 0.5350593328475952, "denoise": 0.4864580035209656, "risk": 0.13886098563671112, "AP": 0.11153768934304407 }, { "epoch": 87, "total": 0.5327900648117065, "denoise": 0.4841173589229584, "risk": 0.13906489312648773, "AP": 0.17386936895218327 }, { "epoch": 88, "total": 0.5519800186157227, "denoise": 0.5040207505226135, "risk": 0.13702639937400818, "AP": 0.17806248158653956 }, { "epoch": 89, "total": 0.682606041431427, "denoise": 0.6339329481124878, "risk": 0.13906599581241608, "AP": 0.18301809046652873 }, { "epoch": 90, "total": 0.667876124382019, "denoise": 0.6190845370292664, "risk": 0.13940447568893433, "AP": 0.18579082505865951 }, { "epoch": 91, "total": 0.5016557574272156, "denoise": 0.45219525694847107, "risk": 0.14131566882133484, "AP": 0.19669478092848777 }, { "epoch": 92, "total": 0.5808471441268921, "denoise": 0.5317724943161011, "risk": 0.14021329581737518, "AP": 0.1947857984436302 }, { "epoch": 93, "total": 0.4977581799030304, "denoise": 0.44759464263916016, "risk": 0.1433243602514267, "AP": 0.19035512239409558 }, { "epoch": 94, "total": 0.565030038356781, "denoise": 0.5148932337760925, "risk": 0.14324799180030823, "AP": 0.19168230133231104 }, { "epoch": 95, "total": 0.5124223828315735, "denoise": 0.4616490602493286, "risk": 0.14506658911705017, "AP": 0.20084948008802248 }, { "epoch": 96, "total": 0.5035750865936279, "denoise": 0.4526267349720001, "risk": 0.1455666422843933, "AP": 0.1914590111674134 }, { "epoch": 97, "total": 0.545097291469574, "denoise": 0.49545052647590637, "risk": 0.14184796810150146, "AP": 0.17082433268359737 }, { "epoch": 98, "total": 0.5145318508148193, "denoise": 0.4654983878135681, "risk": 0.14009569585323334, "AP": 0.16978148772858748 }, { "epoch": 99, "total": 0.5201123952865601, "denoise": 0.4712304472923279, "risk": 0.1396627575159073, "AP": 0.16865063621482995 }, { "epoch": 100, "total": 0.5167859196662903, "denoise": 0.4685807526111603, "risk": 0.1377289742231369, "AP": 0.16950137394834608 }, { "epoch": 101, "total": 0.48568564653396606, "denoise": 0.4381343424320221, "risk": 0.1358608603477478, "AP": 0.17144559277359886 }, { "epoch": 102, "total": 0.4871044158935547, "denoise": 0.4401411712169647, "risk": 0.1341806948184967, "AP": 0.17198712102261843 }, { "epoch": 103, "total": 0.49141713976860046, "denoise": 0.4452956020832062, "risk": 0.13177581131458282, "AP": 0.17115540015654468 }, { "epoch": 104, "total": 0.47930702567100525, "denoise": 0.4332461357116699, "risk": 0.13160254061222076, "AP": 0.17049850119322904 }, { "epoch": 105, "total": 0.4839414358139038, "denoise": 0.43852338194847107, "risk": 0.12976588308811188, "AP": 0.17388585852530836 }, { "epoch": 106, "total": 0.4676992893218994, "denoise": 0.4221382141113281, "risk": 0.13017454743385315, "AP": 0.178228428383852 }, { "epoch": 107, "total": 0.46427270770072937, "denoise": 0.41899481415748596, "risk": 0.12936539947986603, "AP": 0.17978523814187974 }, { "epoch": 108, "total": 0.46348148584365845, "denoise": 0.41790059208869934, "risk": 0.130231112241745, "AP": 0.1773889809828862 }, { "epoch": 109, "total": 0.46907371282577515, "denoise": 0.42406150698661804, "risk": 0.12860625982284546, "AP": 0.1766241687671134 }, { "epoch": 110, "total": 0.4538276791572571, "denoise": 0.4092784523963928, "risk": 0.12728354334831238, "AP": 0.17709513419816914 }, { "epoch": 111, "total": 0.4297528862953186, "denoise": 0.3852289319038391, "risk": 0.12721134722232819, "AP": 0.17777765743018137 }, { "epoch": 112, "total": 0.4418400228023529, "denoise": 0.39828944206237793, "risk": 0.12443020939826965, "AP": 0.17699639616153692 }, { "epoch": 113, "total": 0.45386457443237305, "denoise": 0.4106704592704773, "risk": 0.12341180443763733, "AP": 0.17655251137752626 }, { "epoch": 114, "total": 0.439879834651947, "denoise": 0.39774197340011597, "risk": 0.12039393186569214, "AP": 0.17764398895960368 }, { "epoch": 115, "total": 0.4448937177658081, "denoise": 0.40317338705062866, "risk": 0.11920090019702911, "AP": 0.18094030716080184 }, { "epoch": 116, "total": 0.43346941471099854, "denoise": 0.3924432694911957, "risk": 0.11721757799386978, "AP": 0.18089405473937648 }, { "epoch": 117, "total": 0.4391760230064392, "denoise": 0.39824408292770386, "risk": 0.11694836616516113, "AP": 0.18237111763312783 }, { "epoch": 118, "total": 0.43646249175071716, "denoise": 0.3963719606399536, "risk": 0.1145443543791771, "AP": 0.19122200291406885 }, { "epoch": 119, "total": 0.4315056800842285, "denoise": 0.39135056734085083, "risk": 0.11472892016172409, "AP": 0.21893142109509645 }, { "epoch": 120, "total": 0.4308333098888397, "denoise": 0.3911997377872467, "risk": 0.11323878169059753, "AP": 0.20665452159712328 } ], "best_loss": 0.4297528862953186, "best_epoch": 111, "metrics": [ { "run": "14h06", "MAE_mm": 5.913818359375, "RMSE_mm": 7.648746013641357, "CAD_correspondent_Chamfer_mm": 5.111356258392334, "registered_scan_Chamfer_mm": 32.72827911376953, "risk": { "AP": 0.17699639616153692, "IoU_at_0.5": 0.0392156862745098, "Dice_at_0.5": 0.07547169811320754 }, "mean_stochastic_std_mm": 9.21276569366455, "max_stochastic_std_mm": 24.149744033813477 } ], "heater_projection_valid": [ false ], "checkpoint": "gen_models/models/graph_6000.pt" }, { "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" } ], "common_contract": "CAD/process conditioning -> signed-normal deformation diffusion -> p=CAD+n*d plus weak risk overlay", "defect_type": { "value": [ "multiple_unvalidated" ], "status": "unvalidated_run_level_labels" }, "critical_limitations": [ "registration_independently_validated=false", "defect extent/identity not fully reviewed", "heater_projection_valid=false; local values zero-imputed with explicit mask", "only 1 NG run(s) and one CAD; no held-out result" ], "output_directories": { "processed_inputs": "gen_models/processed_inputs", "models": "gen_models/models", "outputs": "gen_models/outputs", "feature_attributions": "gen_models/feature_attributions" }, "tag_graph_unet_comparison": { "baseline": "graph_6000", "candidate": "graph_unet_6000", "fixed_view": "14h06 CAD X-Z; identical deformation/risk scales", "artifact": "gen_models/outputs/graph_unet_6000_14h06_comparison.png", "status": "weak-label fit only; not a held-out comparison" } }