Upload CHE-Master checkpoint manifest

chemaster_main_moe_hasam_best_manifest.json

@@ -0,0 +1,89 @@
+{
+  "name": "CHE-Master main MoE-LoRA + HASAM checkpoint",
+  "artifact_name": "chemaster_main_moe_hasam_best.pth",
+  "checkpoint_role": "public EMA evaluation weights for the main CHE-Master model",
+  "format": "PyTorch checkpoint with model weights and minimal public config",
+  "size_bytes": 1217897294,
+  "sha256": "9de3756c2e1b0895de4f9adcf8b5894fd94734920d980e7cf085f2cf26f72f27",
+  "verified_at": "2026-06-19",
+  "training_config": {
+    "trainer": "convnext_tqn_moe_hasam",
+    "model_name": "convnext_base",
+    "train_file": "datasets/pretrain_final/train_fullres.json",
+    "eval_dir": "datasets/eqtd_eval",
+    "image_size": 512,
+    "batch_size": 128,
+    "epochs": 20,
+    "lr": 0.0001,
+    "num_entities": 56,
+    "use_moe_lora": true,
+    "moe_lora_rank": 8,
+    "moe_lora_alpha": 8.0,
+    "moe_lora_top_k": 2,
+    "moe_lora_target_mode": "late_pwconv2",
+    "moe_smooth_transition": true,
+    "phase2_start_epoch": 4,
+    "moe_label_route_mix_alpha_final": 1.0,
+    "hasam_rho": 0.0001,
+    "hasam_lambda": 0.02,
+    "use_sample_selection": true,
+    "sample_selection_mix_ratio": 0.65,
+    "sample_selection_rare_weight": 0.35,
+    "sample_selection_positive_weight": 0.2,
+    "sample_selection_domain_weight": 0.2,
+    "sample_selection_group_weight": 0.25,
+    "sample_selection_cross_group_weight": 0.15,
+    "use_model_ema": true,
+    "model_ema_decay": 0.9998
+  },
+  "paper_metric_alignment": {
+    "paper_table": "Table 2 adult CXR in-domain and out-of-domain classification performance",
+    "metric": "AUROC",
+    "rounding": "paper values are reported to three decimals",
+    "adult_benchmark_auroc": {
+      "chestxray14": 0.839,
+      "chexpert": 0.923,
+      "vindr_cxr": 0.918,
+      "shenzhen": 0.968,
+      "openi": 0.778,
+      "siim_acr_pneumothorax": 0.935
+    },
+    "adult_six_benchmark_mean_auroc": 0.894
+  },
+  "reviewer_rerun": {
+    "verified_at": "2026-06-19",
+    "device": "NVIDIA A100 80GB PCIe",
+    "batch_size": 64,
+    "environment": {
+      "torch": "2.4.0+cu121",
+      "timm": "1.0.7",
+      "transformers": "4.51.3"
+    },
+    "checkpoint_sha256": "9de3756c2e1b0895de4f9adcf8b5894fd94734920d980e7cf085f2cf26f72f27",
+    "adult_benchmark_auroc_exact": {
+      "chestxray14": 0.8391567221879616,
+      "chexpert": 0.9230505273334062,
+      "vindr_cxr": 0.9180842018955298,
+      "shenzhen": 0.9675175284837861,
+      "openi": 0.7783835181534597,
+      "siim_acr_pneumothorax": 0.9350151766125333
+    },
+    "adult_six_benchmark_mean_auroc_exact": 0.8935346124444461,
+    "rounds_to_paper_table": true
+  },
+  "evaluation_files": {
+    "chestxray14": "datasets/eqtd_eval/chestxray14.json",
+    "chexpert": "datasets/eqtd_eval/chexpert.json",
+    "vindr_cxr": "datasets/eqtd_eval/vindr.json",
+    "shenzhen": "datasets/eqtd_eval/shenzhen.json",
+    "openi": "datasets/eqtd_eval/openi.json",
+    "siim_acr_pneumothorax": "datasets/eqtd_eval/siim.json"
+  },
+  "distribution": {
+    "git_history": "not committed",
+    "reason": "checkpoint is a multi-GB binary artifact and should not be stored in the reviewer source repository",
+    "recommended_channel": "Hugging Face model repository",
+    "download_url": "https://huggingface.co/frankzhang/CHE-Master/resolve/main/chemaster_main_moe_hasam_best.pth",
+    "verification_command": "sha256sum checkpoints/chemaster_main_moe_hasam_best.pth"
+  }
+}