{
  "model_dir": "models/deepIVUS",
  "output_dir": "models/multitask",
  "split_json": "evals/splits/ivus_split_merged_600.json",
  "num_train": 420,
  "num_val": 90,
  "num_test": 90,
  "num_train_with_lumen": 199,
  "num_val_with_lumen": 46,
  "num_test_with_lumen": 48,
  "selected_trainable_variables": [
    "conv2d/kernel:0",
    "conv2d/bias:0",
    "batch_normalization/gamma:0",
    "batch_normalization/beta:0",
    "conv2d_1/kernel:0",
    "conv2d_1/bias:0",
    "batch_normalization_1/gamma:0",
    "batch_normalization_1/beta:0",
    "conv2d_2/kernel:0",
    "conv2d_2/bias:0",
    "batch_normalization_2/gamma:0",
    "batch_normalization_2/beta:0",
    "conv2d_3/kernel:0",
    "conv2d_3/bias:0",
    "conv2d_4/kernel:0",
    "conv2d_4/bias:0",
    "batch_normalization_3/gamma:0",
    "batch_normalization_3/beta:0",
    "conv2d_5/kernel:0",
    "conv2d_5/bias:0",
    "batch_normalization_4/gamma:0",
    "batch_normalization_4/beta:0",
    "conv2d_6/kernel:0",
    "conv2d_6/bias:0",
    "conv2d_7/kernel:0",
    "conv2d_7/bias:0",
    "batch_normalization_5/gamma:0",
    "batch_normalization_5/beta:0",
    "conv2d_8/kernel:0",
    "conv2d_8/bias:0",
    "batch_normalization_6/gamma:0",
    "batch_normalization_6/beta:0",
    "conv2d_9/kernel:0",
    "conv2d_9/bias:0",
    "conv2d_10/kernel:0",
    "conv2d_10/bias:0",
    "batch_normalization_7/gamma:0",
    "batch_normalization_7/beta:0",
    "conv2d_11/kernel:0",
    "conv2d_11/bias:0",
    "batch_normalization_8/gamma:0",
    "batch_normalization_8/beta:0",
    "conv2d_12/kernel:0",
    "conv2d_12/bias:0",
    "conv2d_13/kernel:0",
    "conv2d_13/bias:0",
    "batch_normalization_9/gamma:0",
    "batch_normalization_9/beta:0",
    "conv2d_14/kernel:0",
    "conv2d_14/bias:0",
    "batch_normalization_10/gamma:0",
    "batch_normalization_10/beta:0",
    "conv2d_15/kernel:0",
    "conv2d_15/bias:0",
    "conv2d_16/kernel:0",
    "conv2d_16/bias:0",
    "batch_normalization_11/gamma:0",
    "batch_normalization_11/beta:0",
    "conv2d_17/kernel:0",
    "conv2d_17/bias:0",
    "batch_normalization_12/gamma:0",
    "batch_normalization_12/beta:0",
    "conv2d_18/kernel:0",
    "conv2d_18/bias:0",
    "batch_normalization_13/gamma:0",
    "batch_normalization_13/beta:0",
    "conv2d_19/kernel:0",
    "conv2d_19/bias:0",
    "batch_normalization_14/gamma:0",
    "batch_normalization_14/beta:0",
    "conv2d_20/kernel:0",
    "conv2d_20/bias:0",
    "conv2d_21/kernel:0",
    "conv2d_21/bias:0",
    "batch_normalization_15/gamma:0",
    "batch_normalization_15/beta:0",
    "conv2d_22/kernel:0",
    "conv2d_22/bias:0",
    "batch_normalization_16/gamma:0",
    "batch_normalization_16/beta:0",
    "conv2d_23/kernel:0",
    "conv2d_23/bias:0",
    "batch_normalization_17/gamma:0",
    "batch_normalization_17/beta:0",
    "conv2d_24/kernel:0",
    "conv2d_24/bias:0",
    "batch_normalization_18/gamma:0",
    "batch_normalization_18/beta:0",
    "conv2d_25/kernel:0",
    "conv2d_25/bias:0",
    "conv2d_26/kernel:0",
    "conv2d_26/bias:0",
    "batch_normalization_19/gamma:0",
    "batch_normalization_19/beta:0",
    "conv2d_27/kernel:0",
    "conv2d_27/bias:0",
    "batch_normalization_20/gamma:0",
    "batch_normalization_20/beta:0",
    "conv2d_28/kernel:0",
    "conv2d_28/bias:0",
    "batch_normalization_21/gamma:0",
    "batch_normalization_21/beta:0",
    "conv2d_29/kernel:0",
    "conv2d_29/bias:0",
    "batch_normalization_22/gamma:0",
    "batch_normalization_22/beta:0",
    "conv2d_30/kernel:0",
    "conv2d_30/bias:0",
    "batch_normalization_23/gamma:0",
    "batch_normalization_23/beta:0",
    "conv2d_31/kernel:0",
    "conv2d_31/bias:0",
    "batch_normalization_24/gamma:0",
    "batch_normalization_24/beta:0",
    "conv2d_32/kernel:0",
    "conv2d_32/bias:0"
  ],
  "lumen_class": 1,
  "pos_weight": 40.0,
  "seg_loss_weight": 1.0,
  "cls_loss_weight": 1.0,
  "learning_rate": 1e-05,
  "epochs": 50,
  "early_stop_patience": 10,
  "early_stop_min_delta": 0.0001,
  "best_val_total_loss": 0.4344508498907089,
  "tensorboard_run_dir": "output/tensorboard/multitask_20260223_224823",
  "saved_base_model": "models/multitask/lumen_multitask_base",
  "saved_cls_head": "models/multitask/bifurcation_head.keras",
  "history": [
    {
      "epoch": 1,
      "total_loss": 1.6219867297581263,
      "seg_loss": 0.6216800527203651,
      "cls_loss": 1.0003066670326959,
      "val_total_loss": 1.166487949049991,
      "val_seg_loss": 0.48462160773899243,
      "val_cls_loss": 0.6818663387194924,
      "val_seg_iou": 0.8267602893737925,
      "val_seg_dice": 0.9051656029343657,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.4888888888888889,
      "val_cls_precision": 0.8421052631578947,
      "val_cls_recall": 0.2711864406779661,
      "val_cls_f1": 0.41025641025641024,
      "val_cls_auc": 0.7908693552017212
    },
    {
      "epoch": 2,
      "total_loss": 1.0519415350187393,
      "seg_loss": 0.290034962055229,
      "cls_loss": 0.7619065758727845,
      "val_total_loss": 0.8631932268971982,
      "val_seg_loss": 0.23684684527309044,
      "val_cls_loss": 0.6263463808142621,
      "val_seg_iou": 0.8250022138914308,
      "val_seg_dice": 0.9041109184544913,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8222222222222222,
      "val_cls_precision": 0.9215686274509803,
      "val_cls_recall": 0.7966101694915254,
      "val_cls_f1": 0.8545454545454545,
      "val_cls_auc": 0.8138327598571777
    },
    {
      "epoch": 3,
      "total_loss": 0.834745450814565,
      "seg_loss": 0.18543397166899273,
      "cls_loss": 0.649311478648867,
      "val_total_loss": 0.7018603773220725,
      "val_seg_loss": 0.16327474152912264,
      "val_cls_loss": 0.5385856408139934,
      "val_seg_iou": 0.846050787524255,
      "val_seg_dice": 0.9166061879141435,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8222222222222222,
      "val_cls_precision": 0.9215686274509803,
      "val_cls_recall": 0.7966101694915254,
      "val_cls_f1": 0.8545454545454545,
      "val_cls_auc": 0.8477310538291931
    },
    {
      "epoch": 4,
      "total_loss": 0.7766569486686161,
      "seg_loss": 0.13996965427483832,
      "cls_loss": 0.6366872943582988,
      "val_total_loss": 0.6386936164420584,
      "val_seg_loss": 0.14631453933923141,
      "val_cls_loss": 0.49237907321556756,
      "val_seg_iou": 0.8537190784842085,
      "val_seg_dice": 0.921087869670411,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8222222222222222,
      "val_cls_precision": 0.9215686274509803,
      "val_cls_recall": 0.7966101694915254,
      "val_cls_f1": 0.8545454545454545,
      "val_cls_auc": 0.8510115146636963
    },
    {
      "epoch": 5,
      "total_loss": 0.7333434970605941,
      "seg_loss": 0.1164263075306302,
      "cls_loss": 0.6169171908072063,
      "val_total_loss": 0.5967901336110156,
      "val_seg_loss": 0.11332971699859785,
      "val_cls_loss": 0.48346041855604754,
      "val_seg_iou": 0.8680616997222836,
      "val_seg_dice": 0.9293715511124011,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8555555555555555,
      "val_cls_precision": 0.9107142857142857,
      "val_cls_recall": 0.864406779661017,
      "val_cls_f1": 0.8869565217391304,
      "val_cls_auc": 0.9013121724128723
    },
    {
      "epoch": 6,
      "total_loss": 0.6692844209216889,
      "seg_loss": 0.08745707979514486,
      "cls_loss": 0.581827338110833,
      "val_total_loss": 0.5571767538785934,
      "val_seg_loss": 0.11077517190057298,
      "val_cls_loss": 0.44640157987242157,
      "val_seg_iou": 0.8661160593957511,
      "val_seg_dice": 0.9282552979863425,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8666666666666667,
      "val_cls_precision": 0.9272727272727272,
      "val_cls_recall": 0.864406779661017,
      "val_cls_f1": 0.8947368421052632,
      "val_cls_auc": 0.9122471809387207
    },
    {
      "epoch": 7,
      "total_loss": 0.6274600121236983,
      "seg_loss": 0.07445602903053875,
      "cls_loss": 0.5530039798645746,
      "val_total_loss": 0.5509960029436194,
      "val_seg_loss": 0.1002325456103553,
      "val_cls_loss": 0.45076345879098645,
      "val_seg_iou": 0.8746350232394943,
      "val_seg_dice": 0.9331256616854054,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8666666666666667,
      "val_cls_precision": 0.8615384615384616,
      "val_cls_recall": 0.9491525423728814,
      "val_cls_f1": 0.903225806451613,
      "val_cls_auc": 0.891744077205658
    },
    {
      "epoch": 8,
      "total_loss": 0.6130301696913583,
      "seg_loss": 0.08664557699646269,
      "cls_loss": 0.526384590779032,
      "val_total_loss": 0.5237435011760049,
      "val_seg_loss": 0.0972047365554001,
      "val_cls_loss": 0.42653876089531445,
      "val_seg_iou": 0.874846441854967,
      "val_seg_dice": 0.9332459686558615,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8444444444444444,
      "val_cls_precision": 0.8571428571428571,
      "val_cls_recall": 0.9152542372881356,
      "val_cls_f1": 0.8852459016393444,
      "val_cls_auc": 0.9122471213340759
    },
    {
      "epoch": 9,
      "total_loss": 0.5469670985426222,
      "seg_loss": 0.06870117130733672,
      "cls_loss": 0.478265926809538,
      "val_total_loss": 0.49475600311289664,
      "val_seg_loss": 0.09005594059177068,
      "val_cls_loss": 0.4047000644647557,
      "val_seg_iou": 0.8738963815485469,
      "val_seg_dice": 0.932705127298851,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8555555555555555,
      "val_cls_precision": 0.859375,
      "val_cls_recall": 0.9322033898305084,
      "val_cls_f1": 0.8943089430894309,
      "val_cls_auc": 0.9024056792259216
    },
    {
      "epoch": 10,
      "total_loss": 0.5185818938981919,
      "seg_loss": 0.0611622542142868,
      "cls_loss": 0.4574196387969312,
      "val_total_loss": 0.47969698776369507,
      "val_seg_loss": 0.09069109529904697,
      "val_cls_loss": 0.3890058948941853,
      "val_seg_iou": 0.8817167983190007,
      "val_seg_dice": 0.9371408057861493,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8555555555555555,
      "val_cls_precision": 0.8709677419354839,
      "val_cls_recall": 0.9152542372881356,
      "val_cls_f1": 0.8925619834710744,
      "val_cls_auc": 0.9108802676200867
    },
    {
      "epoch": 11,
      "total_loss": 0.5268871486186981,
      "seg_loss": 0.06076143806179365,
      "cls_loss": 0.4661257118696258,
      "val_total_loss": 0.4448992982506752,
      "val_seg_loss": 0.07935935145486957,
      "val_cls_loss": 0.3655399514929108,
      "val_seg_iou": 0.8856157410877115,
      "val_seg_dice": 0.9393385108005587,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8666666666666667,
      "val_cls_precision": 0.8507462686567164,
      "val_cls_recall": 0.9661016949152542,
      "val_cls_f1": 0.9047619047619047,
      "val_cls_auc": 0.8881903290748596
    },
    {
      "epoch": 12,
      "total_loss": 0.5059011499087016,
      "seg_loss": 0.05725990863783019,
      "cls_loss": 0.44864123818420226,
      "val_total_loss": 0.4382701370379199,
      "val_seg_loss": 0.07930069967456486,
      "val_cls_loss": 0.35896943768729334,
      "val_seg_iou": 0.8856746535702068,
      "val_seg_dice": 0.9393716481189703,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8666666666666667,
      "val_cls_precision": 0.8507462686567164,
      "val_cls_recall": 0.9661016949152542,
      "val_cls_f1": 0.9047619047619047,
      "val_cls_auc": 0.9075998067855835
    },
    {
      "epoch": 13,
      "total_loss": 0.4970441772824242,
      "seg_loss": 0.057231439685537705,
      "cls_loss": 0.439812736426081,
      "val_total_loss": 0.4344508498907089,
      "val_seg_loss": 0.09339732369003088,
      "val_cls_loss": 0.3410535249454172,
      "val_seg_iou": 0.8945803720041263,
      "val_seg_dice": 0.9443572679451131,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8777777777777778,
      "val_cls_precision": 0.875,
      "val_cls_recall": 0.9491525423728814,
      "val_cls_f1": 0.9105691056910569,
      "val_cls_auc": 0.9111536741256714
    },
    {
      "epoch": 14,
      "total_loss": 0.46513107106799173,
      "seg_loss": 0.05708659976011231,
      "cls_loss": 0.40804447230129015,
      "val_total_loss": 0.4582638439276944,
      "val_seg_loss": 0.08206254211456879,
      "val_cls_loss": 0.376201301165249,
      "val_seg_iou": 0.8872420885233893,
      "val_seg_dice": 0.9402525451491841,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8666666666666667,
      "val_cls_precision": 0.8507462686567164,
      "val_cls_recall": 0.9661016949152542,
      "val_cls_f1": 0.9047619047619047,
      "val_cls_auc": 0.9026790857315063
    },
    {
      "epoch": 15,
      "total_loss": 0.4548539487378938,
      "seg_loss": 0.05213610732129642,
      "cls_loss": 0.4027178420906975,
      "val_total_loss": 0.454915103705033,
      "val_seg_loss": 0.09062011280785436,
      "val_cls_loss": 0.36429498603810434,
      "val_seg_iou": 0.8955005118096512,
      "val_seg_dice": 0.9448697124905642,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8666666666666667,
      "val_cls_precision": 0.8507462686567164,
      "val_cls_recall": 0.9661016949152542,
      "val_cls_f1": 0.9047619047619047,
      "val_cls_auc": 0.9013122320175171
    },
    {
      "epoch": 16,
      "total_loss": 0.410304028292497,
      "seg_loss": 0.04845838170676004,
      "cls_loss": 0.3618456463728632,
      "val_total_loss": 0.5145163043685581,
      "val_seg_loss": 0.10101435382081114,
      "val_cls_loss": 0.4135019507097161,
      "val_seg_iou": 0.891431483260684,
      "val_seg_dice": 0.9425998151663669,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8555555555555555,
      "val_cls_precision": 0.859375,
      "val_cls_recall": 0.9322033898305084,
      "val_cls_f1": 0.8943089430894309,
      "val_cls_auc": 0.9051394462585449
    },
    {
      "epoch": 17,
      "total_loss": 0.3699024962882201,
      "seg_loss": 0.0493658354594594,
      "cls_loss": 0.3205366617866925,
      "val_total_loss": 0.49016835832077527,
      "val_seg_loss": 0.09558146333564883,
      "val_cls_loss": 0.39458689255558926,
      "val_seg_iou": 0.8980368589743589,
      "val_seg_dice": 0.9462796833773087,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8444444444444444,
      "val_cls_precision": 0.8461538461538461,
      "val_cls_recall": 0.9322033898305084,
      "val_cls_f1": 0.8870967741935484,
      "val_cls_auc": 0.9138873815536499
    },
    {
      "epoch": 18,
      "total_loss": 0.42403326346760706,
      "seg_loss": 0.05110736340284348,
      "cls_loss": 0.3729258975812367,
      "val_total_loss": 0.48100930387559143,
      "val_seg_loss": 0.10024259045072224,
      "val_cls_loss": 0.38076670921367145,
      "val_seg_iou": 0.8942522003419237,
      "val_seg_dice": 0.9441743820389976,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8555555555555555,
      "val_cls_precision": 0.8382352941176471,
      "val_cls_recall": 0.9661016949152542,
      "val_cls_f1": 0.8976377952755905,
      "val_cls_auc": 0.9045926928520203
    },
    {
      "epoch": 19,
      "total_loss": 0.34200384191104344,
      "seg_loss": 0.0491072195271651,
      "cls_loss": 0.29289662268544947,
      "val_total_loss": 0.5058531544130781,
      "val_seg_loss": 0.10361128624366678,
      "val_cls_loss": 0.4022418714897788,
      "val_seg_iou": 0.9026263111538153,
      "val_seg_dice": 0.9488214326295457,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8555555555555555,
      "val_cls_precision": 0.859375,
      "val_cls_recall": 0.9322033898305084,
      "val_cls_f1": 0.8943089430894309,
      "val_cls_auc": 0.9149808883666992
    },
    {
      "epoch": 20,
      "total_loss": 0.3296657298647222,
      "seg_loss": 0.047296005416484106,
      "cls_loss": 0.2823697240003163,
      "val_total_loss": 0.5930156037211418,
      "val_seg_loss": 0.10862410991736081,
      "val_cls_loss": 0.48439149542347243,
      "val_seg_iou": 0.9005908150758388,
      "val_seg_dice": 0.947695640673611,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8555555555555555,
      "val_cls_precision": 0.8484848484848485,
      "val_cls_recall": 0.9491525423728814,
      "val_cls_f1": 0.896,
      "val_cls_auc": 0.9002187252044678
    },
    {
      "epoch": 21,
      "total_loss": 0.3407677157294183,
      "seg_loss": 0.04778165167995862,
      "cls_loss": 0.29298606470582034,
      "val_total_loss": 0.521953581791857,
      "val_seg_loss": 0.11065525464389635,
      "val_cls_loss": 0.411298333140819,
      "val_seg_iou": 0.9034158153419418,
      "val_seg_dice": 0.949257443444796,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8444444444444444,
      "val_cls_precision": 0.835820895522388,
      "val_cls_recall": 0.9491525423728814,
      "val_cls_f1": 0.888888888888889,
      "val_cls_auc": 0.91361403465271
    },
    {
      "epoch": 22,
      "total_loss": 0.3208419575222901,
      "seg_loss": 0.04237900156350363,
      "cls_loss": 0.27846295549756006,
      "val_total_loss": 0.5469912105928296,
      "val_seg_loss": 0.12718820912034615,
      "val_cls_loss": 0.4198029992859001,
      "val_seg_iou": 0.9005287487132929,
      "val_seg_dice": 0.9476612751298544,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8555555555555555,
      "val_cls_precision": 0.8484848484848485,
      "val_cls_recall": 0.9491525423728814,
      "val_cls_f1": 0.896,
      "val_cls_auc": 0.9212684631347656
    },
    {
      "epoch": 23,
      "total_loss": 0.27995289305136317,
      "seg_loss": 0.04284256480279423,
      "cls_loss": 0.23711032902910595,
      "val_total_loss": 0.54095732404486,
      "val_seg_loss": 0.11155797411566196,
      "val_cls_loss": 0.42939934581923095,
      "val_seg_iou": 0.89917340357889,
      "val_seg_dice": 0.9469102735794911,
      "val_seg_count": 46.0,
      "val_cls_accuracy": 0.8666666666666667,
      "val_cls_precision": 0.8615384615384616,
      "val_cls_recall": 0.9491525423728814,
      "val_cls_f1": 0.903225806451613,
      "val_cls_auc": 0.922088623046875
    }
  ],
  "test_metrics": {
    "total_loss": 0.4740983749861303,
    "seg_loss": 0.149400915140691,
    "cls_loss": 0.32469746322654514,
    "seg_iou": 0.851996586190484,
    "seg_dice": 0.920084402469685,
    "seg_count": 48.0,
    "cls_accuracy": 0.8555555555555555,
    "cls_precision": 0.9259259259259259,
    "cls_recall": 0.847457627118644,
    "cls_f1": 0.8849557522123893,
    "cls_auc": 0.9633679986000061
  }
}