{
  "experiment_id": "attention_unet_focal_dice_64_inbreast",
  "model": "attention_unet",
  "encoder": "resnet34",
  "loss_function": "focal_dice",
  "patch_size": 64,
  "dataset": "inbreast",
  "positive_ratio": 0.7,
  "best_epoch": 31,
  "best_val_dice": 0.6916026175022125,
  "training_config": {
    "epochs": 100,
    "batch_size": 8,
    "lr": 0.0001,
    "weight_decay": 0.0001,
    "optimizer": "adamw",
    "scheduler": "cosine",
    "warmup_epochs": 5,
    "early_stopping_patience": 25,
    "gradient_clip_norm": 1.0,
    "mixed_precision": true,
    "num_workers": 4,
    "accelerator": "auto"
  },
  "train_history": [
    {
      "epoch": 1,
      "loss": 0.470939945101738,
      "dice": 0.26273016070201993
    },
    {
      "epoch": 2,
      "loss": 0.37831361627578736,
      "dice": 0.45761922996491194
    },
    {
      "epoch": 3,
      "loss": 0.29135782733559606,
      "dice": 0.5510095232725143
    },
    {
      "epoch": 4,
      "loss": 0.2489262210428715,
      "dice": 0.6045896550416946
    },
    {
      "epoch": 5,
      "loss": 0.2410249900817871,
      "dice": 0.6096846392452717
    },
    {
      "epoch": 6,
      "loss": 0.24974131068587302,
      "dice": 0.5826229446530342
    },
    {
      "epoch": 7,
      "loss": 0.24322941079735755,
      "dice": 0.5860968301892281
    },
    {
      "epoch": 8,
      "loss": 0.22245872941613198,
      "dice": 0.6196434313654899
    },
    {
      "epoch": 9,
      "loss": 0.21304077675938607,
      "dice": 0.6353536452651024
    },
    {
      "epoch": 10,
      "loss": 0.20716007693111896,
      "dice": 0.6453983706831932
    },
    {
      "epoch": 11,
      "loss": 0.19373451760411262,
      "dice": 0.6671460281014443
    },
    {
      "epoch": 12,
      "loss": 0.1965045806467533,
      "dice": 0.6603592174053192
    },
    {
      "epoch": 13,
      "loss": 0.1775081458389759,
      "dice": 0.6980668310374022
    },
    {
      "epoch": 14,
      "loss": 0.17040000094473362,
      "dice": 0.7092891335487366
    },
    {
      "epoch": 15,
      "loss": 0.17529901863634587,
      "dice": 0.6988445857167244
    },
    {
      "epoch": 16,
      "loss": 0.19018878361582756,
      "dice": 0.6742907764315605
    },
    {
      "epoch": 17,
      "loss": 0.19597814710438252,
      "dice": 0.6641482664942742
    },
    {
      "epoch": 18,
      "loss": 0.19380615621060132,
      "dice": 0.6673862655162811
    },
    {
      "epoch": 19,
      "loss": 0.18360209114849568,
      "dice": 0.6841707702577114
    },
    {
      "epoch": 20,
      "loss": 0.19063195736706257,
      "dice": 0.6715489913225174
    },
    {
      "epoch": 21,
      "loss": 0.18482155208289625,
      "dice": 0.6808138552606106
    },
    {
      "epoch": 22,
      "loss": 0.1827920549213886,
      "dice": 0.6839834990203381
    },
    {
      "epoch": 23,
      "loss": 0.1712102492749691,
      "dice": 0.7057046221792698
    },
    {
      "epoch": 24,
      "loss": 0.16310413558036088,
      "dice": 0.7188621182143689
    },
    {
      "epoch": 25,
      "loss": 0.17742198172211648,
      "dice": 0.693995748102665
    },
    {
      "epoch": 26,
      "loss": 0.1663977778851986,
      "dice": 0.7095884943902493
    },
    {
      "epoch": 27,
      "loss": 0.16046073221042753,
      "dice": 0.7245720982551574
    },
    {
      "epoch": 28,
      "loss": 0.1550739907771349,
      "dice": 0.7319629094004632
    },
    {
      "epoch": 29,
      "loss": 0.1504477052092552,
      "dice": 0.7422738779187202
    },
    {
      "epoch": 30,
      "loss": 0.146174390360713,
      "dice": 0.7484889486432076
    },
    {
      "epoch": 31,
      "loss": 0.15282664980739355,
      "dice": 0.7368941506147385
    },
    {
      "epoch": 32,
      "loss": 0.14807593605667352,
      "dice": 0.7443547195196152
    },
    {
      "epoch": 33,
      "loss": 0.1523734062090516,
      "dice": 0.7346913783848286
    },
    {
      "epoch": 34,
      "loss": 0.14430243009328841,
      "dice": 0.7523236124515533
    },
    {
      "epoch": 35,
      "loss": 0.1494244988113642,
      "dice": 0.7420375203490257
    },
    {
      "epoch": 36,
      "loss": 0.16579542135447264,
      "dice": 0.7120123512148857
    },
    {
      "epoch": 37,
      "loss": 0.17393783950805664,
      "dice": 0.7001664729118348
    },
    {
      "epoch": 38,
      "loss": 0.16989232221245765,
      "dice": 0.7047005292773246
    },
    {
      "epoch": 39,
      "loss": 0.17886033928394318,
      "dice": 0.692703010559082
    },
    {
      "epoch": 40,
      "loss": 0.16896781631559132,
      "dice": 0.7090354162454605
    },
    {
      "epoch": 41,
      "loss": 0.16362110132724048,
      "dice": 0.7192413275837898
    },
    {
      "epoch": 42,
      "loss": 0.1623893383964896,
      "dice": 0.7206748932451009
    },
    {
      "epoch": 43,
      "loss": 0.16097493302822113,
      "dice": 0.7230039074420929
    },
    {
      "epoch": 44,
      "loss": 0.15658709464222192,
      "dice": 0.7272709524035453
    },
    {
      "epoch": 45,
      "loss": 0.15169794822484256,
      "dice": 0.7378337498009205
    },
    {
      "epoch": 46,
      "loss": 0.155521089553833,
      "dice": 0.7307330233156681
    },
    {
      "epoch": 47,
      "loss": 0.15938615208864212,
      "dice": 0.724573203265667
    },
    {
      "epoch": 48,
      "loss": 0.16057429045438767,
      "dice": 0.7246711131930351
    },
    {
      "epoch": 49,
      "loss": 0.15699513205885887,
      "dice": 0.7289057652950287
    },
    {
      "epoch": 50,
      "loss": 0.1487896761894226,
      "dice": 0.7442537631690502
    },
    {
      "epoch": 51,
      "loss": 0.15075480243563652,
      "dice": 0.7404399904608726
    },
    {
      "epoch": 52,
      "loss": 0.14639006346464156,
      "dice": 0.7452981821000576
    },
    {
      "epoch": 53,
      "loss": 0.1381391460672021,
      "dice": 0.7602012299895287
    },
    {
      "epoch": 54,
      "loss": 0.1478860678523779,
      "dice": 0.7423566768467427
    },
    {
      "epoch": 55,
      "loss": 0.14112884345650673,
      "dice": 0.7551238349676133
    },
    {
      "epoch": 56,
      "loss": 0.14059246511757373,
      "dice": 0.7576151592731476
    }
  ],
  "val_history": [
    {
      "epoch": 1,
      "loss": 0.4722620937973261,
      "dice": 0.375198483816348,
      "iou": 0.2569431306037586
    },
    {
      "epoch": 2,
      "loss": 0.4105376494117081,
      "dice": 0.3910131643060595,
      "iou": 0.2638940478209406
    },
    {
      "epoch": 3,
      "loss": 0.34148135781288147,
      "dice": 0.44641284202225506,
      "iou": 0.31379235768690705
    },
    {
      "epoch": 4,
      "loss": 0.2882224644999951,
      "dice": 0.5480718445032835,
      "iou": 0.40035711508244276
    },
    {
      "epoch": 5,
      "loss": 0.2709959433414042,
      "dice": 0.5957183283753693,
      "iou": 0.4498963290825486
    },
    {
      "epoch": 6,
      "loss": 0.24377023242413998,
      "dice": 0.6004723135847598,
      "iou": 0.45510789996478707
    },
    {
      "epoch": 7,
      "loss": 0.2368394120130688,
      "dice": 0.5823362534865737,
      "iou": 0.4370007130783051
    },
    {
      "epoch": 8,
      "loss": 0.22926044068299234,
      "dice": 0.6020960644818842,
      "iou": 0.45302246743813157
    },
    {
      "epoch": 9,
      "loss": 0.21095539105590433,
      "dice": 0.6208835542201996,
      "iou": 0.4815255478024483
    },
    {
      "epoch": 10,
      "loss": 0.19459088193252683,
      "dice": 0.6560102868825197,
      "iou": 0.5123847951181233
    },
    {
      "epoch": 11,
      "loss": 0.19888133089989424,
      "dice": 0.6380728920921683,
      "iou": 0.48705912148579955
    },
    {
      "epoch": 12,
      "loss": 0.19694718811661005,
      "dice": 0.6470187032828107,
      "iou": 0.5100248856469989
    },
    {
      "epoch": 13,
      "loss": 0.19573024241253734,
      "dice": 0.6543402906972915,
      "iou": 0.5121215091785416
    },
    {
      "epoch": 14,
      "loss": 0.2265079446369782,
      "dice": 0.579700167523697,
      "iou": 0.4455919817555696
    },
    {
      "epoch": 15,
      "loss": 0.21084913797676563,
      "dice": 0.5985642585437745,
      "iou": 0.4577180080814287
    },
    {
      "epoch": 16,
      "loss": 0.20091359689831734,
      "dice": 0.6324195512570441,
      "iou": 0.48685734649188817
    },
    {
      "epoch": 17,
      "loss": 0.18911945959553123,
      "dice": 0.6366590210236609,
      "iou": 0.5008961581625044
    },
    {
      "epoch": 18,
      "loss": 0.18375813006423414,
      "dice": 0.6661872789263725,
      "iou": 0.5117951752617955
    },
    {
      "epoch": 19,
      "loss": 0.2115599401295185,
      "dice": 0.629787772661075,
      "iou": 0.4812599193537608
    },
    {
      "epoch": 20,
      "loss": 0.21355100453365594,
      "dice": 0.608618725091219,
      "iou": 0.46367813646793365
    },
    {
      "epoch": 21,
      "loss": 0.23852951382286847,
      "dice": 0.5918755219900049,
      "iou": 0.4519428496641922
    },
    {
      "epoch": 22,
      "loss": 0.22837820078711957,
      "dice": 0.5749439364299178,
      "iou": 0.4316546272020787
    },
    {
      "epoch": 23,
      "loss": 0.18988335109315813,
      "dice": 0.6462043053470552,
      "iou": 0.5049592396244407
    },
    {
      "epoch": 24,
      "loss": 0.20879121881444007,
      "dice": 0.619439888279885,
      "iou": 0.4798138940241188
    },
    {
      "epoch": 25,
      "loss": 0.2012313223676756,
      "dice": 0.6140758208930492,
      "iou": 0.47312248405069113
    },
    {
      "epoch": 26,
      "loss": 0.2918721192982048,
      "dice": 0.5135783982113935,
      "iou": 0.388231774471933
    },
    {
      "epoch": 27,
      "loss": 0.22775481012649834,
      "dice": 0.5828360152663663,
      "iou": 0.44793501688400283
    },
    {
      "epoch": 28,
      "loss": 0.1736924066208303,
      "dice": 0.6789645468816161,
      "iou": 0.5324512957595289
    },
    {
      "epoch": 29,
      "loss": 0.20803017436992377,
      "dice": 0.6245911605656147,
      "iou": 0.48540836246684194
    },
    {
      "epoch": 30,
      "loss": 0.17230443668086082,
      "dice": 0.6817624578252435,
      "iou": 0.5322406738996506
    },
    {
      "epoch": 31,
      "loss": 0.1679785802261904,
      "dice": 0.6916026175022125,
      "iou": 0.5527996253222227
    },
    {
      "epoch": 32,
      "loss": 0.22350540931802243,
      "dice": 0.5858028032816947,
      "iou": 0.44592265458777547
    },
    {
      "epoch": 33,
      "loss": 0.22755796741694212,
      "dice": 0.5776670063205529,
      "iou": 0.44505863826634595
    },
    {
      "epoch": 34,
      "loss": 0.2039731479017064,
      "dice": 0.6190242629963905,
      "iou": 0.48224261309951544
    },
    {
      "epoch": 35,
      "loss": 0.1902149972738698,
      "dice": 0.6488110097125173,
      "iou": 0.5081155938096344
    },
    {
      "epoch": 36,
      "loss": 0.24101340980269015,
      "dice": 0.5634422744624317,
      "iou": 0.4239583141170442
    },
    {
      "epoch": 37,
      "loss": 0.196726942085661,
      "dice": 0.6253427173942327,
      "iou": 0.49106831499375403
    },
    {
      "epoch": 38,
      "loss": 0.20104754727799445,
      "dice": 0.6286675287410617,
      "iou": 0.4826438052114099
    },
    {
      "epoch": 39,
      "loss": 0.2033290525432676,
      "dice": 0.6225335474591702,
      "iou": 0.47673061362002045
    },
    {
      "epoch": 40,
      "loss": 0.1876503569073975,
      "dice": 0.6509064808487892,
      "iou": 0.5072036287747324
    },
    {
      "epoch": 41,
      "loss": 0.1924115592846647,
      "dice": 0.6480290554463863,
      "iou": 0.5041704243049026
    },
    {
      "epoch": 42,
      "loss": 0.1988064736360684,
      "dice": 0.6258700173348188,
      "iou": 0.48285772325471044
    },
    {
      "epoch": 43,
      "loss": 0.16633984283544123,
      "dice": 0.6884123645722866,
      "iou": 0.5453846706077456
    },
    {
      "epoch": 44,
      "loss": 0.21930227556731552,
      "dice": 0.608573657926172,
      "iou": 0.4689920600503683
    },
    {
      "epoch": 45,
      "loss": 0.2576732428278774,
      "dice": 0.5401117894798517,
      "iou": 0.39974406571127474
    },
    {
      "epoch": 46,
      "loss": 0.24095960578415543,
      "dice": 0.5760849607177079,
      "iou": 0.44460496166720986
    },
    {
      "epoch": 47,
      "loss": 0.19816575571894646,
      "dice": 0.6358740273863077,
      "iou": 0.4859474473632872
    },
    {
      "epoch": 48,
      "loss": 0.1866596353938803,
      "dice": 0.6548553262837231,
      "iou": 0.510943521745503
    },
    {
      "epoch": 49,
      "loss": 0.18324490555096418,
      "dice": 0.6639226735569537,
      "iou": 0.519788057776168
    },
    {
      "epoch": 50,
      "loss": 0.1922026495449245,
      "dice": 0.6380380033515394,
      "iou": 0.4938765794504434
    },
    {
      "epoch": 51,
      "loss": 0.1760057331994176,
      "dice": 0.6733705010265112,
      "iou": 0.5215165796689689
    },
    {
      "epoch": 52,
      "loss": 0.1743937738938257,
      "dice": 0.6726995687931776,
      "iou": 0.5250943107530475
    },
    {
      "epoch": 53,
      "loss": 0.20487687643617392,
      "dice": 0.6292975507676601,
      "iou": 0.4806041973643005
    },
    {
      "epoch": 54,
      "loss": 0.17063764645718038,
      "dice": 0.6565803317353129,
      "iou": 0.5101644461974502
    },
    {
      "epoch": 55,
      "loss": 0.19417141296435148,
      "dice": 0.6430215393193066,
      "iou": 0.49564255122095346
    },
    {
      "epoch": 56,
      "loss": 0.20377497642766684,
      "dice": 0.6205073287710547,
      "iou": 0.4802676655817777
    }
  ]
}