FetalVision / models /seed1337 /summary.json
Tiendat88's picture
Upload trained FetalVision ensemble and model card
1cd4680 verified
Raw
History Blame Contribute Delete
3.81 kB
{
"run_name": "multitask_latent32_seed1337",
"device": "mps",
"parameter_count": 777986,
"best_epoch": 13,
"epochs_completed": 16,
"training_seconds": 1000.590726791881,
"data": {
"train_images": 8290,
"train_standard": 5522,
"train_other": 2768,
"validation_images": 1783,
"test_images": 1843,
"image_size": [
128,
192
]
},
"configuration": {
"manifest_dir": "data/fetal_planes",
"output_root": "results/fetal_hybrid",
"run_name": "multitask_latent32_seed1337",
"epochs": 16,
"patience": 5,
"batch_size": 64,
"num_workers": 0,
"image_height": 128,
"image_width": 192,
"base_channels": 32,
"latent_channels": 32,
"reconstruction_loss": "hybrid",
"classification_weight": 0.75,
"noise_std": 0.02,
"mask_prob": 0.1,
"learning_rate": 0.001,
"weight_decay": 0.0001,
"seed": 1337,
"device": "auto",
"no_progress": false,
"resume": false,
"max_train_samples": null,
"max_eval_samples": null
},
"ensemble": {
"classifier_weight": 0.9,
"reconstruction_weight": 0.09999999999999998,
"reconstruction_center": -4.269355493837512e-17,
"reconstruction_scale": 0.9999999999999998,
"classifier_center": -5.557549904197327,
"classifier_scale": 3.3953835769454943,
"threshold": 1.3463199026148178,
"validation_auc": 0.9569232051335472,
"validation_f1": 0.8465863453815261,
"reconstruction_model": {
"component_names": [
"mse",
"mae",
"ssim",
"gradient"
],
"center": [
0.005649745710246073,
0.0483627147333049,
0.34110302091156464,
0.024708705716826374
],
"scale": [
0.002281657884645313,
0.012843779402067698,
0.057011465334869435,
0.006235540250448552
],
"weights": [
1.0,
0.0,
0.0,
0.0
],
"threshold": -1.0451513311018779,
"validation_auc": 0.6163270691834382,
"validation_f1": 0.5375565610859728
}
},
"validation_metrics": {
"roc_auc": 0.9569232051335472,
"average_precision": 0.9340889725557499,
"threshold": 1.3463199026148178,
"accuracy": 0.8928771733034212,
"precision": 0.8405103668261563,
"recall": 0.8527508090614887,
"specificity": 0.9141630901287554,
"f1": 0.8465863453815261,
"tp": 527,
"tn": 1065,
"fp": 100,
"fn": 91
},
"test_metrics": {
"roc_auc": 0.9732853858784893,
"average_precision": 0.9587582026053482,
"threshold": 1.3463199026148178,
"accuracy": 0.9120998372219208,
"precision": 0.8419497784342689,
"recall": 0.912,
"specificity": 0.9121510673234812,
"f1": 0.8755760368663594,
"tp": 570,
"tn": 1111,
"fp": 107,
"fn": 55
},
"test_per_class": {
"Fetal abdomen": {
"count": 106,
"mean_anomaly_score": 0.9546521653304901,
"predicted_other_rate": 0.27358490566037735
},
"Fetal brain": {
"count": 471,
"mean_anomaly_score": -0.37824112049043424,
"predicted_other_rate": 0.016985138004246284
},
"Fetal femur": {
"count": 159,
"mean_anomaly_score": 0.8448219699689173,
"predicted_other_rate": 0.22641509433962265
},
"Fetal thorax": {
"count": 262,
"mean_anomaly_score": 0.4083387485165997,
"predicted_other_rate": 0.12213740458015267
},
"Maternal cervix": {
"count": 220,
"mean_anomaly_score": -0.648014399132354,
"predicted_other_rate": 0.00909090909090909
},
"Other": {
"count": 625,
"mean_anomaly_score": 3.363634500886128,
"predicted_other_rate": 0.912
}
},
"task_limitation": "Other is a proxy for non-standard/OOD views, not fetal pathology."
}