File size: 1,540 Bytes
fe9550f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 | {
"name": "toothfairy4-prototype-v1",
"seed": 2026,
"folds": 5,
"preprocess": {
"spacing_mm": [
0.5,
0.5,
0.5
],
"shape_zyx": [
112,
192,
192
],
"clip_percentiles": [
0.5,
99.5
],
"orientation": "LPS",
"dtype": "float16"
},
"prototypes": {
"max_prototypes": 1024,
"min_support": 2,
"embedder": "tfidf",
"linkage_threshold": 0.62,
"max_sentence_words": 60,
"assign_threshold": 0.45,
"tooth_aware": true
},
"encoder": {
"backbone": "slice2d",
"timm_model": "convnext_tiny",
"width": 32,
"dropout": 0.2,
"epochs": 40,
"batch_size": 4,
"accumulate": 2,
"learning_rate": 0.0002,
"weight_decay": 0.02,
"warmup_ratio": 0.1,
"label_smoothing": 0.02,
"focal_gamma_negative": 2.0,
"focal_gamma_positive": 0.0,
"ema_decay": 0.999,
"amp": true,
"num_workers": 8,
"min_head_support": 0,
"seed": 2026
},
"llm": {
"base_model": "Qwen/Qwen2.5-1.5B-Instruct",
"enabled": false,
"lora_rank": 32,
"lora_alpha": 64,
"epochs": 3,
"batch_size": 4,
"learning_rate": 0.0001,
"max_length": 1536,
"max_new_tokens": 512
},
"decode": {
"strategy": "prototype",
"clinical_weight": 0.8,
"captioning_weight": 0.2,
"calibration_rounds": 6,
"refine_top": 250,
"min_sentences": 4,
"max_sentences": 16,
"mbr_candidates": 24
}
}
|