| { |
| "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 |
| } |
| } |
