FetalVision / models /seed2026 /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_seed2026",
"device": "mps",
"parameter_count": 777986,
"best_epoch": 10,
"epochs_completed": 15,
"training_seconds": 935.708518124884,
"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_seed2026",
"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": 2026,
"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": -2.4396317107642924e-17,
"reconstruction_scale": 0.9999999999999994,
"classifier_center": -4.182426768029708,
"classifier_scale": 2.764643149539139,
"threshold": 1.2869621288438537,
"validation_auc": 0.9585107712821368,
"validation_f1": 0.847723704866562,
"reconstruction_model": {
"component_names": [
"mse",
"mae",
"ssim",
"gradient"
],
"center": [
0.005860817135019875,
0.04767355534523853,
0.3397925037682824,
0.024900084672407785
],
"scale": [
0.002498879624278239,
0.012349472025245112,
0.05600078434992182,
0.0061834939655143325
],
"weights": [
1.0,
0.0,
0.0,
0.0
],
"threshold": -0.5639049616176401,
"validation_auc": 0.6708807311415753,
"validation_f1": 0.5643363728470111
}
},
"validation_metrics": {
"roc_auc": 0.9585107712821368,
"average_precision": 0.9363763181699387,
"threshold": 1.2869621288438537,
"accuracy": 0.8911946158160404,
"precision": 0.823170731707317,
"recall": 0.8737864077669902,
"specificity": 0.9004291845493563,
"f1": 0.8477237048665619,
"tp": 540,
"tn": 1049,
"fp": 116,
"fn": 78
},
"test_metrics": {
"roc_auc": 0.9753556650246306,
"average_precision": 0.9630265423503459,
"threshold": 1.2869621288438537,
"accuracy": 0.9131850244167119,
"precision": 0.8374455732946299,
"recall": 0.9232,
"specificity": 0.9080459770114943,
"f1": 0.878234398782344,
"tp": 577,
"tn": 1106,
"fp": 112,
"fn": 48
},
"test_per_class": {
"Fetal abdomen": {
"count": 106,
"mean_anomaly_score": 0.8289849242845742,
"predicted_other_rate": 0.2169811320754717
},
"Fetal brain": {
"count": 471,
"mean_anomaly_score": -0.39380829337040435,
"predicted_other_rate": 0.01910828025477707
},
"Fetal femur": {
"count": 159,
"mean_anomaly_score": 0.8886497961974562,
"predicted_other_rate": 0.2641509433962264
},
"Fetal thorax": {
"count": 262,
"mean_anomaly_score": 0.40608574766650485,
"predicted_other_rate": 0.12213740458015267
},
"Maternal cervix": {
"count": 220,
"mean_anomaly_score": -0.56403287603023,
"predicted_other_rate": 0.02727272727272727
},
"Other": {
"count": 625,
"mean_anomaly_score": 3.6597532778300024,
"predicted_other_rate": 0.9232
}
},
"task_limitation": "Other is a proxy for non-standard/OOD views, not fetal pathology."
}