CCL2026 中文电子病历疾病与手术ICD自动编码 — 推理提交包

• 方案:Qwen3.5-4B(base,≤7B 合规)+ LoRA 微调(V2 重排版 checkpoint-400),vLLM 推理。
• 与 A 榜最佳一致:推理链路 = base+LoRA + 手工 prompt + postprocess_reorder 还原提交格式,与 A 榜实测 0.8850 的 result_ft_reorder_ckpt400.txt 完全相同。

一、目录结构

tijiao/
├── README.md                       # 本文件(使用说明)
├── 环境安装说明.md                  # 环境/依赖安装(从零 or 用vLLM镜像)
├── requirements.txt                # 依赖版本清单(实测可复现)
├── run_infer.sh                    # ★一键推理脚本
│
├── baseline/                       # 代码
│   ├── infer.py                    # 推理入口(读xlsx → 名称→编码 → 后处理 → result.txt)
│   ├── run_baseline.py             # 共享内核(知识库/匹配器/DX2OP/格式; import时加载知识库)
│   └── eval_mtotal.py              # 离线评测(官方M_total, 有金标时用, 可选)
│
├── icd_knowledge_base/             # ICD知识库(必须在 baseline 上一级目录)
│   ├── task_candidates.json        #   18主诊断 + 16主手术 封闭候选集
│   ├── icd10_diseases.json         #   ICD-10 疾病库(名称→编码)
│   ├── icd9cm3_procedures.json     #   ICD-9-CM3 手术库(名称→编码)
│   └── code2name.json / name2codes.json / metadata.json
│
├── train_data/                     # 推理用映射(由训练集统计得到)
│   ├── name2code_map.json          #   训练集金标准 名称→编码(三级映射第1级)
│   └── dx2op_train.json            #   主诊断→主手术 白名单先验
│
└── model/
    ├── Qwen3.5-4B/                 # base 模型(8.8G)
    └── lora/checkpoint-400/        # LoRA 适配器(adapter_config + adapter_model, 259M)

⚠️ 路径约束:run_baseline.py 在 import 时按"自身上一级目录 / icd_knowledge_base"定位知识库。 因此 baseline/ 与 icd_knowledge_base/ 必须保持上述相对位置,不要单独移动 baseline 目录。

二、快速运行

• 不带参数时:默认输入 ../o_data/B榜.xlsx,输出 ./result.txt。
• 例(对 B 榜):
  bash run_infer.sh /path/to/B榜.xlsx ./result_B.txt
• Docker 场景(读 /input、写 /output):
  bash run_infer.sh /input/B_test.xlsx /output/result.txt

显存需求:4B + LoRA,bf16,gpu_memory_utilization=0.85,max_model_len=8192。 单卡 ≥16GB 即可(实测 NVIDIA vGPU-32GB,峰值约 12GB)。

三、输入 / 输出格式

• 主诊断、主手术有且仅 1 个;其他手术最多 2 个。
• 例:O34.201|O82.000;Z37.000x001;Z35.401|74.1x01|65.2901

四、方案要点(为何这样设计)

1. 名称版微调:模型输出 ICD 中文名称(非编码串),名称有语义、易学;再用映射还原编码 (名称↔编码 100% 可逆)。
2. V2 重排:模型按 主诊|主术|其他诊|其他术 输出(主码相邻,强化 80% 权重主码学习); postprocess_reorder() 解析后还原为提交序 主诊|其他诊|主术|其他术。
3. 名称→编码三级映射:NAME2CODE(训练集金标准,优先)→ 知识库精确 → LCS 模糊兜底。
4. 主码兜底:coerce 把主诊/主术吸附到 18/16 封闭候选集(保证合法); DX2OP 白名单信任模型、仅在明显错时纠正(实测后处理无损 M=1.0)。

五、离线评测(可选)

六、注意事项

• 必须带 --reorder(run_infer.sh 已内置):本 LoRA 是重排版,漏了它会导致主术/其他诊错位,主码全错。
• base+LoRA,勿用 merge 模型:merge 版与 base+LoRA 推理有约 32 行差异,本包用 base+LoRA。
• 环境/依赖见《环境安装说明.md》;若 vLLM 启动报 ninja/nvcc 缺失,见该文档"常见问题"。