chest2vec/chest2vec_labeler
015
1{2 "model_type": "chest2vec_labeler",3 "architectures": [4 "Chest2VecLabelerModel"5 ],6 "auto_map": {7 "AutoConfig": "modeling_chest2vec_labeler.Chest2VecLabelerConfig",8 "AutoModel": "modeling_chest2vec_labeler.Chest2VecLabelerModel"9 },10 "base_model": "Qwen/Qwen3-Embedding-0.6B",11 "encoder_config": {12 "vocab_size": 151669,13 "max_position_embeddings": 32768,14 "hidden_size": 1024,15 "intermediate_size": 3072,16 "num_hidden_layers": 28,17 "num_attention_heads": 16,18 "use_sliding_window": false,19 "sliding_window": null,20 "max_window_layers": 28,21 "num_key_value_heads": 8,22 "head_dim": 128,23 "hidden_act": "silu",24 "initializer_range": 0.02,25 "rms_norm_eps": 1e-06,26 "use_cache": true,27 "rope_theta": 1000000,28 "rope_scaling": null,29 "attention_bias": false,30 "attention_dropout": 0.0,31 "layer_types": [32 "full_attention",33 "full_attention",34 "full_attention",35 "full_attention",36 "full_attention",37 "full_attention",38 "full_attention",39 "full_attention",40 "full_attention",41 "full_attention",42 "full_attention",43 "full_attention",44 "full_attention",45 "full_attention",46 "full_attention",47 "full_attention",48 "full_attention",49 "full_attention",50 "full_attention",51 "full_attention",52 "full_attention",53 "full_attention",54 "full_attention",55 "full_attention",56 "full_attention",57 "full_attention",58 "full_attention",59 "full_attention"60 ],61 "return_dict": true,62 "output_hidden_states": false,63 "torchscript": false,64 "dtype": "bfloat16",65 "pruned_heads": {},66 "tie_word_embeddings": true,67 "chunk_size_feed_forward": 0,68 "is_encoder_decoder": false,69 "is_decoder": false,70 "cross_attention_hidden_size": null,71 "add_cross_attention": false,72 "tie_encoder_decoder": false,73 "architectures": [74 "Qwen3ForCausalLM"75 ],76 "finetuning_task": null,77 "id2label": {78 "0": "LABEL_0",79 "1": "LABEL_1"80 },81 "label2id": {82 "LABEL_0": 0,83 "LABEL_1": 184 },85 "task_specific_params": null,86 "problem_type": null,87 "tokenizer_class": null,88 "prefix": null,89 "bos_token_id": 151643,90 "pad_token_id": null,91 "eos_token_id": 151643,92 "sep_token_id": null,93 "decoder_start_token_id": null,94 "max_length": 20,95 "min_length": 0,96 "do_sample": false,97 "early_stopping": false,98 "num_beams": 1,99 "temperature": 1.0,100 "top_k": 50,101 "top_p": 1.0,102 "typical_p": 1.0,103 "repetition_penalty": 1.0,104 "length_penalty": 1.0,105 "no_repeat_ngram_size": 0,106 "encoder_no_repeat_ngram_size": 0,107 "bad_words_ids": null,108 "num_return_sequences": 1,109 "output_scores": false,110 "return_dict_in_generate": false,111 "forced_bos_token_id": null,112 "forced_eos_token_id": null,113 "remove_invalid_values": false,114 "exponential_decay_length_penalty": null,115 "suppress_tokens": null,116 "begin_suppress_tokens": null,117 "num_beam_groups": 1,118 "diversity_penalty": 0.0,119 "_name_or_path": "Qwen/Qwen3-Embedding-0.6B",120 "transformers_version": "4.57.3",121 "model_type": "qwen3",122 "tf_legacy_loss": false,123 "use_bfloat16": false,124 "output_attentions": false125 },126 "hidden_size": 1024,127 "num_classes_per_label": 3,128 "labels": [129 "Consolidation",130 "Ground-glass opacity (GGO)",131 "Crazy-paving pattern",132 "Mosaic attenuation / air-trapping",133 "Tree-in-bud",134 "Centrilobular nodules / bronchiolitis pattern",135 "Pulmonary nodule (solid / PSN / GGN)",136 "Pulmonary mass (>3 cm)",137 "Cavitary nodule / mass",138 "Emphysema",139 "Bullae / giant bulla",140 "Pulmonary cysts / cystic lung disease",141 "Reticulation / intralobular thickening",142 "Interlobular septal thickening",143 "Traction bronchiectasis / bronchiolectasis",144 "Honeycombing",145 "Parenchymal scarring / fibrotic band",146 "Tracheal stenosis / malacia",147 "Tracheal / bronchial wall thickening",148 "Bronchiectasis",149 "Mucoid impaction / plugging",150 "Tracheal diverticulum",151 "Endotracheal tube",152 "Tracheostomy tube",153 "Lobar / segmental atelectasis",154 "Subsegmental / linear atelectasis",155 "Post-lobectomy / segmentectomy",156 "Post-pneumonectomy",157 "Lung transplant",158 "Lungs & Airways_others",159 "Pleural effusion",160 "Loculated pleural effusion",161 "Hemothorax",162 "Chest tube / pleural drain",163 "Pneumothorax",164 "Tension pneumothorax",165 "Pleural thickening",166 "Pleural plaques",167 "Pleural nodule / mass",168 "Pleura_others",169 "Mediastinal lymphadenopathy",170 "Hilar lymphadenopathy",171 "Calcified mediastinal / hilar lymph nodes",172 "Anterior mediastinal mass",173 "Middle / posterior mediastinal mass or cyst",174 "Thymic remnant / hyperplasia",175 "Esophageal wall thickening / mass",176 "Hiatal hernia",177 "Esophageal dilation",178 "Nasogastric / orogastric tube",179 "Pneumomediastinum",180 "Mediastinal hematoma / fluid collection",181 "Mediastinum & Hila_others",182 "Cardiomegaly",183 "Pericardial effusion",184 "Pericardial thickening / calcification",185 "Coronary artery calcification",186 "Coronary stent or bypass graft",187 "Thoracic aortic calcification",188 "Thoracic aortic ectasia / dilation (non-aneurysmal)",189 "Thoracic aortic aneurysm",190 "Aortic dissection / intramural hematoma",191 "Main pulmonary artery enlargement",192 "Pulmonary embolism",193 "Aortic valve calcification",194 "Mitral annular calcification",195 "Pacemaker / ICD leads",196 "Central venous catheter / PICC",197 "LVAD / other cardiac assist device",198 "Cardiovascular_others",199 "Chest wall soft tissue edema / hematoma",200 "Subcutaneous emphysema",201 "Chest wall mass",202 "Post-thoracotomy change",203 "Chest wall tumor invasion",204 "Chest Wall_others",205 "Acute rib fracture",206 "Non-acute / healed rib fracture",207 "Sternal fracture",208 "Vertebral compression fracture",209 "Degenerative spine changes",210 "Osteolytic bone lesion",211 "Osteosclerotic bone lesion",212 "Mixed osteolytic-osteosclerotic lesion",213 "Osteopenia",214 "Scoliosis / kyphosis",215 "Vertebral hemangioma",216 "Postoperative spine change / hardware",217 "Bones / Spine_others",218 "Hepatic steatosis",219 "Focal liver lesion (nodule / mass)",220 "Hepatomegaly",221 "Liver contour irregularity / cirrhosis features",222 "Hepatic calcification",223 "Cholelithiasis / gallstones",224 "Post-cholecystectomy (gallbladder operated / absent)",225 "Gallbladder wall thickening",226 "Hydropic gallbladder / distension",227 "Biliary sludge",228 "Biliary stent / catheter / drain",229 "Splenomegaly",230 "Accessory spleen / splenule / polysplenia",231 "Focal splenic lesion (nodule / mass)",232 "Pancreatic mass / focal lesion",233 "Pancreatic lipomatosis",234 "Adrenal nodule / mass",235 "Adrenal thickening / hyperplasia",236 "Adrenal calcification",237 "Simple renal cyst",238 "Complex renal cyst / solid renal mass",239 "Hydronephrosis",240 "Renal calculi / nephrolithiasis",241 "Renal atrophy / decreased renal size",242 "Nephrectomy (kidney absent / operated)",243 "Ascites",244 "Pneumoperitoneum",245 "Bowel wall thickening / inflammation",246 "Diverticulosis",247 "Omental caking / peritoneal carcinomatosis",248 "Abdominal lymphadenopathy",249 "Abdominal aortic aneurysm (partially imaged)",250 "Abdominal aortic calcification / atherosclerosis (partially imaged)",251 "IVC filter",252 "Upper Abdomen_others",253 "Thyroid enlargement (goiter)",254 "Thyroid nodule",255 "Cervical / supraclavicular lymphadenopathy",256 "Neck soft tissue mass",257 "Lower Neck_others",258 "Breast mass / focal asymmetry",259 "Post-lumpectomy / post-mastectomy change",260 "Breast implant (intact or present)",261 "Axillary lymphadenopathy",262 "Motion artifact / suboptimal study",263 "Study limitation / limited evaluation (non-motion)",264 "No significant intrathoracic abnormality",265 "Others_others"266 ],267 "instruction": "Given the following chest CT report, extract the presence/absence of entities",268 "max_len": 512,269 "default_threshold": 0.5,270 "torch_dtype": "float32",271 "n_labels": 137,272 "attn_implementation": "sdpa",273 "label_hierarchy": {274 "Lungs & Airways": {275 "Lung parenchymal attenuation patterns": [276 "Consolidation",277 "Ground-glass opacity (GGO)",278 "Crazy-paving pattern",279 "Mosaic attenuation / air-trapping",280 "Tree-in-bud",281 "Centrilobular nodules / bronchiolitis pattern"282 ],283 "Nodules and masses": [284 "Pulmonary nodule (solid / PSN / GGN)",285 "Pulmonary mass (>3 cm)",286 "Cavitary nodule / mass"287 ],288 "Emphysema & cystic lung disease": [289 "Emphysema",290 "Bullae / giant bulla",291 "Pulmonary cysts / cystic lung disease"292 ],293 "Interstitial / fibrotic lung disease": [294 "Reticulation / intralobular thickening",295 "Interlobular septal thickening",296 "Traction bronchiectasis / bronchiolectasis",297 "Honeycombing",298 "Parenchymal scarring / fibrotic band"299 ],300 "Volume loss / hyperinflation": [301 "Lobar / segmental atelectasis",302 "Subsegmental / linear atelectasis"303 ]304 },305 "Trachea & Airways": {306 "Trachea & Airways": [307 "Tracheal stenosis / malacia",308 "Tracheal / bronchial wall thickening",309 "Bronchiectasis",310 "Mucoid impaction / plugging",311 "Tracheal diverticulum"312 ]313 },314 "Pleura": {315 "Pleural fluid collections": [316 "Pleural effusion",317 "Loculated pleural effusion",318 "Hemothorax"319 ],320 "Pleural air": [321 "Pneumothorax",322 "Tension pneumothorax"323 ],324 "Pleural thickening / masses": [325 "Pleural thickening",326 "Pleural plaques",327 "Pleural nodule / mass"328 ]329 },330 "Mediastinum & Hila": {331 "Lymph nodes (mediastinal)": [332 "Mediastinal lymphadenopathy",333 "Hilar lymphadenopathy",334 "Calcified mediastinal / hilar lymph nodes"335 ],336 "Mediastinal masses / cysts": [337 "Anterior mediastinal mass",338 "Middle / posterior mediastinal mass or cyst",339 "Thymic remnant / hyperplasia"340 ],341 "Esophagus / hiatus": [342 "Esophageal wall thickening / mass",343 "Hiatal hernia",344 "Esophageal dilation"345 ],346 "Air / fluid in mediastinum": [347 "Pneumomediastinum",348 "Mediastinal hematoma / fluid collection"349 ]350 },351 "Cardiovascular": {352 "Cardiac size and morphology": [353 "Cardiomegaly"354 ],355 "Pericardium": [356 "Pericardial effusion",357 "Pericardial thickening / calcification"358 ],359 "Coronary arteries": [360 "Coronary artery calcification",361 "Coronary stent or bypass graft"362 ],363 "Aorta / pulmonary artery": [364 "Thoracic aortic calcification",365 "Thoracic aortic ectasia / dilation (non-aneurysmal)",366 "Thoracic aortic aneurysm",367 "Aortic dissection / intramural hematoma",368 "Main pulmonary artery enlargement",369 "Pulmonary embolism"370 ],371 "Valves and cardiac devices": [372 "Aortic valve calcification",373 "Mitral annular calcification",374 "Pacemaker / ICD leads"375 ]376 },377 "Chest Wall": {378 "Chest wall soft tissues": [379 "Chest wall soft tissue edema / hematoma",380 "Subcutaneous emphysema",381 "Chest wall mass"382 ],383 "Post-surgical / tumor invasion (chest wall)": [384 "Post-thoracotomy change",385 "Post-mastectomy change",386 "Chest wall tumor invasion"387 ]388 },389 "Bones / Spine": {390 "Fractures": [391 "Acute rib fracture",392 "Non-acute / healed rib fracture",393 "Sternal fracture",394 "Vertebral compression fracture"395 ],396 "Degenerative / osseous lesions": [397 "Degenerative spine changes",398 "Osteolytic bone lesion",399 "Osteosclerotic bone lesion",400 "Mixed osteolytic-osteosclerotic lesion",401 "Osteopenia",402 "Scoliosis / kyphosis",403 "Vertebral hemangioma",404 "Postoperative spine change / hardware"405 ]406 },407 "Upper Abdomen": {408 "Liver": [409 "Hepatic steatosis",410 "Focal liver lesion (nodule / mass)",411 "Hepatomegaly",412 "Liver contour irregularity / cirrhosis features",413 "Hepatic calcification"414 ],415 "Gallbladder & biliary": [416 "Cholelithiasis / gallstones",417 "Post-cholecystectomy (gallbladder operated / absent)",418 "Gallbladder wall thickening",419 "Hydropic gallbladder / distension",420 "Biliary sludge",421 "Biliary stent / catheter / drain"422 ],423 "Spleen": [424 "Splenomegaly",425 "Accessory spleen / splenule / polysplenia",426 "Focal splenic lesion (nodule / mass)"427 ],428 "Pancreas": [429 "Pancreatic mass / focal lesion",430 "Pancreatic lipomatosis"431 ],432 "Adrenals": [433 "Adrenal nodule / mass",434 "Adrenal thickening / hyperplasia",435 "Adrenal calcification"436 ],437 "Kidneys / urinary tract": [438 "Simple renal cyst",439 "Complex renal cyst / solid renal mass",440 "Hydronephrosis",441 "Renal calculi / nephrolithiasis",442 "Renal atrophy / decreased renal size",443 "Nephrectomy (kidney absent / operated)"444 ],445 "GI & peritoneum": [446 "Ascites",447 "Pneumoperitoneum",448 "Bowel wall thickening / inflammation",449 "Diverticulosis",450 "Omental caking / peritoneal carcinomatosis"451 ],452 "Lymph nodes (abdominal)": [453 "Abdominal lymphadenopathy"454 ],455 "Upper abdominal vessels": [456 "Abdominal aortic aneurysm (partially imaged)",457 "Abdominal aortic calcification / atherosclerosis (partially imaged)"458 ]459 },460 "Lower Neck": {461 "Thyroid": [462 "Thyroid enlargement (goiter)",463 "Thyroid nodule"464 ],465 "Neck nodes and soft tissues": [466 "Cervical / supraclavicular lymphadenopathy",467 "Neck soft tissue mass"468 ]469 },470 "Others": {471 "Breast": [472 "Breast mass / focal asymmetry",473 "Post-lumpectomy / post-mastectomy change",474 "Breast implant (intact or present)"475 ],476 "Axilla": [477 "Axillary lymphadenopathy"478 ],479 "Tubes, catheters, and support devices": [480 "Endotracheal tube",481 "Tracheostomy tube",482 "Central venous catheter / PICC",483 "Chest tube / pleural drain",484 "Nasogastric / orogastric tube",485 "IVC filter",486 "LVAD / other cardiac assist device"487 ],488 "Post-surgical / transplant status": [489 "Post-lobectomy / segmentectomy",490 "Post-pneumonectomy",491 "Lung transplant"492 ],493 "Study quality / global": [494 "Motion artifact / suboptimal study",495 "Study limitation / limited evaluation (non-motion)",496 "No significant intrathoracic abnormality"497 ]498 }499 }500}