arXiv:2607.11954cs.LGcs.AI2026-07

用图约束策略逐步下钻编码树,解决罕见病历代码预测难题。

Graph-Constrained Policy Learning for Extreme Clinical Code Prediction

  • 把病历编码看作逐层选择有效子节点的决策过程。
  • 在50个代码集上达0.709微F1,全量15761个代码时仍优于基线0.044微F1。
  • 无需强化学习,简单监督策略即可超越复杂级联模型。

临床代码预测将非结构化出院记录映射到大型、稀疏且深层嵌套的ICD-10-CM叶子代码。现有系统多将其视为扁平多标签分类,独立评分代码,对罕见标签训练信号不足。本文提出图约束遍历策略,将ICD预测建模为在修剪后的代码层级图上的有限时域决策过程。单一语言模型逐层下降,选择合法子节点直至到达可计费叶子代码。该方法将极端多标签预测转化为稀疏、层级感知的子集决策,同时保证输出结构有效性。在MIMIC-IV出院记录上,最优监督策略SFT-1+在精选50代码子集上取得0.709微F1,全量15,761代码空间中达0.527微F1,显著优于包括CAML、LAAT和PLM-ICD在内的扁平基线。在全量设置中,相比最强基线提升0.044微F1与0.157宏F1,表明图约束分解有效缓解罕见代码瓶颈。控制因子实验评估架构、训练算法与数据量。两种尺度下,单一策略表现堪比三阶段专家级联,且避免其在28-32%测试样本中的上下文窗口溢出问题。增加监督轨迹数据是唯一持续提升性能的干预措施,而GRPO强化学习在匹配数据下未带来收益。结果表明,简单图约束策略学习可超越更复杂的扁平、级联及强化学习方案。

原文摘要 · Abstract (English)

Clinical code prediction maps unstructured discharge summaries to ICD-10-CM leaf codes in a large, sparse, and deeply hierarchical label space. Most systems treat the task as flat multi-label classification, scoring codes independently and providing limited training signal for rare labels. We propose a graph-constrained traversal policy that formulates ICD prediction as a finite-horizon decision process over a pruned code hierarchy. A single language model descends the graph level by level, selecting valid child nodes until billable leaf codes are reached. This converts extreme multi-label prediction into sparse, hierarchy-aware subset decisions while guaranteeing structurally valid outputs. On MIMIC-IV discharge summaries, our best supervised policy, SFT-1+, achieves 0.709 micro-F1 on a curated 50-code subset and 0.527 micro-F1 on the full 15,761-code space, outperforming flat baselines including CAML, LAAT, and PLM-ICD. In the full setting, SFT-1+ improves over the strongest flat baseline by 0.044 micro-F1 and 0.157 macro-F1, suggesting that graph-constrained decomposition mitigates the rare-code bottleneck. A controlled factorial study evaluates architecture, training algorithm, and data budget. Across both scales, one shared policy matches a three-specialist cascade while avoiding its context-window overflow on 28-32% of full-space test notes. Increasing supervised trajectory data is the only intervention that consistently improves performance, while GRPO reinforcement learning provides no benefit over supervised continuation with matched data. These results show that simple graph-constrained policy learning can outperform more complex flat, cascaded, and reinforcement-learning alternatives for extreme clinical code prediction.

医疗编码图神经网络稀有标签策略学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。