GBCA / trainedmodels /Attributes.json
Kalpokoch's picture
Claude Opus 5.5
feat: add BiomedCLIP attribute classifier as stage 2 for Cancer predictions
dce349c
Raw History Blame Contribute Delete
1.09 kB
{
"arch": "BiomedCLIPAttributeClassifier",
"checkpoint": "final_model_full_data.pth",
"source_run": "BioMedCLIP_GBCA_v3_multiseed_maskedloss_grouped_llm",
"base_model": "hf-hub:microsoft/BiomedCLIP-PubMedBERT_256-vit_base_patch16_224",
"timm_model_name": "vit_base_patch16_224",
"image_size": 336,
"embed_dim": 512,
"hidden_dim": 256,
"dropout": 0.3,
"clip_mean": [0.48145466, 0.4578275, 0.40821073],
"clip_std": [0.26862954, 0.26130258, 0.27577711],
"intensity_pct_low": 1.0,
"intensity_pct_high": 99.0,
"ruq_frac": [0.08, 0.78, 0.0, 0.62],
"labels": [
"wall_thickening",
"mass_replacing_gb",
"polypoidal_lesion",
"liver_invasion",
"lymph_node_involvement",
"biliary_obstruction",
"adjacent_organ_involvement"
],
"thresholds": {
"wall_thickening": 0.46,
"mass_replacing_gb": 0.62,
"polypoidal_lesion": 0.56,
"liver_invasion": 0.51,
"lymph_node_involvement": 0.6,
"biliary_obstruction": 0.62,
"adjacent_organ_involvement": 0.5
},
"oof_mean_auc": 0.7273,
"oof_mean_f1": 0.5433,
"tta": false
}