shawn24 commited on
Commit
2b54c62
·
verified ·
1 Parent(s): 7605810

Add LUMINA cancer classification checkpoints

Browse files

Add linear-probe and full-fine-tuning inference checkpoints with integrity and input metadata.

fullfinetune_mammo_FM_image_encoder/lumina/cancer_classification/README.md ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ # LUMINA cancer classification — Full fine-tuning
2
+
3
+ Mammo-FM EfficientNet-B5 image encoder with a binary cancer-classification head.
4
+
5
+ The checkpoint was selected by validation AUROC from a completed nine-epoch run using the full training split (seed 10, fold 0). Input geometry is 1520 pixels high × 912 pixels wide, with three image channels. The model returns one logit per image; apply sigmoid for its cancer score.
6
+
7
+ Checkpoint: [`mammo_fm_lumina_cancer_full_finetune_seed10_fold0_best_validation_auroc.pth`](checkpoint/mammo_fm_lumina_cancer_full_finetune_seed10_fold0_best_validation_auroc.pth). Integrity and input metadata are in [`release_metadata.json`](release_metadata.json).
8
+
9
+ The checkpoint contains model tensors and simple inference metadata only. It can be loaded with `torch.load(..., weights_only=True)` or the Mammo-FM strict classifier backend. For research use; not a standalone clinical device.
fullfinetune_mammo_FM_image_encoder/lumina/cancer_classification/checkpoint/mammo_fm_lumina_cancer_full_finetune_seed10_fold0_best_validation_auroc.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7d8ace59af336932ac3887f5028ae5923a837764128dba70ebb5d29e80ef5e06
3
+ size 114438790
fullfinetune_mammo_FM_image_encoder/lumina/cancer_classification/release_metadata.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "release_schema_version": 1,
3
+ "dataset": "LUMINA",
4
+ "task": "cancer",
5
+ "method": "full_finetune",
6
+ "model": "Mammo-FM EfficientNet-B5 image encoder with binary classification head",
7
+ "checkpoint_file": "checkpoint/mammo_fm_lumina_cancer_full_finetune_seed10_fold0_best_validation_auroc.pth",
8
+ "checkpoint_sha256": "7d8ace59af336932ac3887f5028ae5923a837764128dba70ebb5d29e80ef5e06",
9
+ "checkpoint_bytes": 114438790,
10
+ "checkpoint_selection": "best validation AUROC",
11
+ "selected_epoch": 4,
12
+ "training_epochs": 9,
13
+ "seed": 10,
14
+ "fold": 0,
15
+ "data_fraction": 1.0,
16
+ "image_size_hw": [
17
+ 1520,
18
+ 912
19
+ ],
20
+ "input_channels": 3,
21
+ "feature_dimension": 2048,
22
+ "output_classes": 1,
23
+ "output_activation": "sigmoid",
24
+ "normalization_mean": 0.3089279,
25
+ "normalization_std": 0.25053555408335154,
26
+ "initialization_checkpoint_filename": "Mammo-FM_BatmanlabTrained_CLIP.tar",
27
+ "model_tensor_count": 854,
28
+ "strict_mammofm_load": "PASS",
29
+ "source_model_tensor_equality": "PASS"
30
+ }
linear_probe_mammo_FM_image_encoder/lumina/cancer_classification/README.md ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ # LUMINA cancer classification — Linear probe
2
+
3
+ Mammo-FM EfficientNet-B5 image encoder with a binary cancer-classification head.
4
+
5
+ The checkpoint was selected by validation AUROC from a completed nine-epoch run using the full training split (seed 10, fold 0). Input geometry is 1520 pixels high × 912 pixels wide, with three image channels. The model returns one logit per image; apply sigmoid for its cancer score.
6
+
7
+ Checkpoint: [`mammo_fm_lumina_cancer_linear_probe_seed10_fold0_best_validation_auroc.pth`](checkpoint/mammo_fm_lumina_cancer_linear_probe_seed10_fold0_best_validation_auroc.pth). Integrity and input metadata are in [`release_metadata.json`](release_metadata.json).
8
+
9
+ The checkpoint contains model tensors and simple inference metadata only. It can be loaded with `torch.load(..., weights_only=True)` or the Mammo-FM strict classifier backend. For research use; not a standalone clinical device.
linear_probe_mammo_FM_image_encoder/lumina/cancer_classification/checkpoint/mammo_fm_lumina_cancer_linear_probe_seed10_fold0_best_validation_auroc.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:416b7eff952bff6101480e1a074188a243a74009cb4d75b13eb5f1f551821fa6
3
+ size 114434220
linear_probe_mammo_FM_image_encoder/lumina/cancer_classification/release_metadata.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "release_schema_version": 1,
3
+ "dataset": "LUMINA",
4
+ "task": "cancer",
5
+ "method": "linear_probe",
6
+ "model": "Mammo-FM EfficientNet-B5 image encoder with binary classification head",
7
+ "checkpoint_file": "checkpoint/mammo_fm_lumina_cancer_linear_probe_seed10_fold0_best_validation_auroc.pth",
8
+ "checkpoint_sha256": "416b7eff952bff6101480e1a074188a243a74009cb4d75b13eb5f1f551821fa6",
9
+ "checkpoint_bytes": 114434220,
10
+ "checkpoint_selection": "best validation AUROC",
11
+ "selected_epoch": 8,
12
+ "training_epochs": 9,
13
+ "seed": 10,
14
+ "fold": 0,
15
+ "data_fraction": 1.0,
16
+ "image_size_hw": [
17
+ 1520,
18
+ 912
19
+ ],
20
+ "input_channels": 3,
21
+ "feature_dimension": 2048,
22
+ "output_classes": 1,
23
+ "output_activation": "sigmoid",
24
+ "normalization_mean": 0.3089279,
25
+ "normalization_std": 0.25053555408335154,
26
+ "initialization_checkpoint_filename": "Mammo-FM_BatmanlabTrained_CLIP.tar",
27
+ "model_tensor_count": 854,
28
+ "strict_mammofm_load": "PASS",
29
+ "source_model_tensor_equality": "PASS"
30
+ }