Upload trained FetalVision ensemble and model card

README.md

@@ -0,0 +1,44 @@
+---
+library_name: pytorch
+pipeline_tag: image-classification
+tags:
+- medical-imaging
+- ultrasound
+- fetal-ultrasound
+- anomaly-detection
+- out-of-distribution-detection
+- autoencoder
+license: other
+---
+
+# FetalVision
+
+PyTorch hybrid autoencoder + latent classifier for detecting non-standard/OOD fetal ultrasound views. The production ensemble combines three independently trained seeds.
+
+## Results
+
+| Model | ROC-AUC | AP | F1 | Accuracy |
+|---|---:|---:|---:|---:|
+| Best single model (seed 2026) | 0.9754 | 0.9630 | 0.8782 | 0.9132 |
+| Three-seed ensemble | **0.9796** | **0.9680** | **0.8887** | **0.9219** |
+
+The ensemble uses weights `0.30 / 0.35 / 0.35` for seeds `42 / 1337 / 2026`, selected only on the validation set.
+
+## Files
+
+- `models/seed42/best_model.pt`
+- `models/seed1337/best_model.pt`
+- `models/seed2026/best_model.pt`
+- `ensemble/summary.json`
+- Per-run summaries under each model directory
+- Training and score-distribution plots for the best single run
+
+Architecture and preprocessing configuration are stored inside every checkpoint and summary file. Full source code and reproducible training commands are available in [GitHub FetalVision](https://github.com/Tiendat88/FetalVision).
+
+## Intended use and limitations
+
+`Other` is a proxy for non-standard/out-of-distribution views. It is **not** a fetal pathology label, and this model must not be interpreted as a diagnostic system. Clinical deployment requires external validation, calibration and appropriate regulatory review.
+
+## Dataset attribution
+
+Trained using FETAL_PLANES_DB by Burgos-Artizzu et al., DOI: [10.5281/zenodo.3904280](https://doi.org/10.5281/zenodo.3904280). The source record does not state a machine-readable license, so this model card uses `license: other` rather than inventing a license.

ensemble/summary.json

@@ -0,0 +1,88 @@
+{
+  "selection_rule": "Weights and threshold selected only on validation; test used once.",
+  "run_dirs": [
+    "results/fetal_hybrid/multitask_latent32",
+    "results/fetal_hybrid/multitask_latent32_seed1337",
+    "results/fetal_hybrid/multitask_latent32_seed2026"
+  ],
+  "weights": [
+    0.3,
+    0.35,
+    0.35
+  ],
+  "validation_transforms": [
+    {
+      "center": -4.879263421528585e-17,
+      "scale": 0.8994131898406392
+    },
+    {
+      "center": 1.8297237830732194e-17,
+      "scale": 0.9074870790939894
+    },
+    {
+      "center": -1.2808066481512535e-16,
+      "scale": 0.911522702986242
+    }
+  ],
+  "validation_metrics": {
+    "roc_auc": 0.9655443976832369,
+    "average_precision": 0.9462326287165863,
+    "threshold": 1.402689945385293,
+    "accuracy": 0.9040942232192933,
+    "precision": 0.8465116279069768,
+    "recall": 0.883495145631068,
+    "specificity": 0.9150214592274678,
+    "f1": 0.8646080760095013,
+    "tp": 546,
+    "tn": 1066,
+    "fp": 99,
+    "fn": 72
+  },
+  "test_metrics": {
+    "roc_auc": 0.9796,
+    "average_precision": 0.9680119784953686,
+    "threshold": 1.402689945385293,
+    "accuracy": 0.9218665219750407,
+    "precision": 0.85949177877429,
+    "recall": 0.92,
+    "specificity": 0.922824302134647,
+    "f1": 0.8887171561051004,
+    "tp": 575,
+    "tn": 1124,
+    "fp": 94,
+    "fn": 50
+  },
+  "test_per_class": {
+    "Fetal abdomen": {
+      "count": 106,
+      "mean_ensemble_score": 0.9559957136424069,
+      "predicted_other_rate": 0.22641509433962265
+    },
+    "Fetal brain": {
+      "count": 471,
+      "mean_ensemble_score": -0.45007270602595306,
+      "predicted_other_rate": 0.008492569002123142
+    },
+    "Fetal femur": {
+      "count": 159,
+      "mean_ensemble_score": 0.9591376994201026,
+      "predicted_other_rate": 0.27044025157232704
+    },
+    "Fetal thorax": {
+      "count": 262,
+      "mean_ensemble_score": 0.4298571036400613,
+      "predicted_other_rate": 0.08396946564885496
+    },
+    "Maternal cervix": {
+      "count": 220,
+      "mean_ensemble_score": -0.6689015967861202,
+      "predicted_other_rate": 0.004545454545454545
+    },
+    "Other": {
+      "count": 625,
+      "mean_ensemble_score": 3.679058146636188,
+      "predicted_other_rate": 0.92
+    }
+  },
+  "task_limitation": "Other is a proxy for non-standard/OOD views, not fetal pathology."
+}

models/seed1337/best_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9f6a781db18a5c050ccb572ce3e8d9da0b74c83ea41b2c7f2cd45dcccd06b895
+size 3132021

models/seed1337/summary.json

@@ -0,0 +1,145 @@
+{
+  "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."
+}

models/seed2026/best_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:842a4e5f01f554a317088fc4f5d6a8fed77dd89806340370246d9f8241742421
+size 3132021

models/seed2026/summary.json

@@ -0,0 +1,145 @@
+{
+  "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."
+}

models/seed42/best_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bbad934b5d174570fd77c5d9b7a5f2b9709cd448775bcf912a3b625bcf8148da
+size 3131957

models/seed42/summary.json

@@ -0,0 +1,143 @@
+{
+  "run_name": "multitask_latent32",
+  "device": "mps",
+  "parameter_count": 777986,
+  "best_epoch": 16,
+  "epochs_completed": 18,
+  "training_seconds": 1112.1197183751501,
+  "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",
+    "epochs": 18,
+    "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.5,
+    "noise_std": 0.02,
+    "mask_prob": 0.1,
+    "learning_rate": 0.001,
+    "weight_decay": 0.0001,
+    "seed": 42,
+    "device": "auto",
+    "max_train_samples": null,
+    "max_eval_samples": null
+  },
+  "ensemble": {
+    "classifier_weight": 0.9,
+    "reconstruction_weight": 0.09999999999999998,
+    "reconstruction_center": 2.0126961613805414e-16,
+    "reconstruction_scale": 1.0,
+    "classifier_center": -5.662346615594344,
+    "classifier_scale": 4.1553970784992575,
+    "threshold": 1.1421258880475764,
+    "validation_auc": 0.9512326902509827,
+    "validation_f1": 0.8403100775193798,
+    "reconstruction_model": {
+      "component_names": [
+        "mse",
+        "mae",
+        "ssim",
+        "gradient"
+      ],
+      "center": [
+        0.004555416000255507,
+        0.04290066549930194,
+        0.30361355169648263,
+        0.024029070083771396
+      ],
+      "scale": [
+        0.0020560742402685656,
+        0.011212009145435244,
+        0.0526615634960164,
+        0.006134039633209833
+      ],
+      "weights": [
+        1.0,
+        0.0,
+        0.0,
+        0.0
+      ],
+      "threshold": -0.49046945466388703,
+      "validation_auc": 0.6264136005666903,
+      "validation_f1": 0.525067385444744
+    }
+  },
+  "validation_metrics": {
+    "roc_auc": 0.9512326902509827,
+    "average_precision": 0.9231059731189711,
+    "threshold": 1.1421258880475764,
+    "accuracy": 0.8844643858665171,
+    "precision": 0.8065476190476191,
+    "recall": 0.8770226537216829,
+    "specificity": 0.8884120171673819,
+    "f1": 0.8403100775193799,
+    "tp": 542,
+    "tn": 1035,
+    "fp": 130,
+    "fn": 76
+  },
+  "test_metrics": {
+    "roc_auc": 0.9699619047619048,
+    "average_precision": 0.9525875374458892,
+    "threshold": 1.1421258880475764,
+    "accuracy": 0.9028757460661965,
+    "precision": 0.8140845070422535,
+    "recall": 0.9248,
+    "specificity": 0.8916256157635468,
+    "f1": 0.8659176029962546,
+    "tp": 578,
+    "tn": 1086,
+    "fp": 132,
+    "fn": 47
+  },
+  "test_per_class": {
+    "Fetal abdomen": {
+      "count": 106,
+      "mean_anomaly_score": 0.8079648503537046,
+      "predicted_other_rate": 0.2641509433962264
+    },
+    "Fetal brain": {
+      "count": 471,
+      "mean_anomaly_score": -0.45864318004349613,
+      "predicted_other_rate": 0.021231422505307854
+    },
+    "Fetal femur": {
+      "count": 159,
+      "mean_anomaly_score": 0.8756956074685064,
+      "predicted_other_rate": 0.3333333333333333
+    },
+    "Fetal thorax": {
+      "count": 262,
+      "mean_anomaly_score": 0.3491010159813523,
+      "predicted_other_rate": 0.13740458015267176
+    },
+    "Maternal cervix": {
+      "count": 220,
+      "mean_anomaly_score": -0.6068094958164075,
+      "predicted_other_rate": 0.022727272727272728
+    },
+    "Other": {
+      "count": 625,
+      "mean_anomaly_score": 2.927662350913958,
+      "predicted_other_rate": 0.9248
+    }
+  },
+  "task_limitation": "Other is a proxy for non-standard/OOD views, not fetal pathology."
+}