论文

先推理后诊断:面向ECG分类的医生启发式结构化思考

Reasoning Before Diagnosis: Physician-Inspired Structured Thinking for ECG Classification

模型训练强化学习

摘要

临床实践中的心电图(ECG)诊断依赖于对多个层级方面的结构化推理,包括心律、传导特性、波形形态和总体诊断印象。然而,大多数现有方法直接从ECG信号预测标签,没有显式的临床推理,导致决策不透明且缺乏临床一致性。为弥合这一差距,我们提出CardioThink,一个受医生启发的多模态大语言模型(MLLM)框架,它通过人类可解释的中间阶段(心律、传导、形态和印象)显式建模诊断推理过程,从而得出最终分类结果。此外,我们提出结构化集合策略优化(SSPO),在不需要人工标注推理轨迹的情况下,联合优化对结构化推理格式的遵循度和可变尺寸诊断集合的准确性。在多样化ECG基准上的大量实验表明,我们的方法在诊断准确率上显著占优,同时提供可解释的临床推理。值得注意的是,推理质量评估证实SSPO显著增强了所生成理由的临床有效性。这些发现揭示,超越直接标签预测、转向结构化推理,为未来ECG建模提供了更符合临床的方向。

先推理后诊断:面向ECG分类的医生启发式结构化思考:论文配图
图 1:CardioThink 的整体流程。我们的模型分两个阶段进行训练:监督指令调整(冷启动)和基于心电图数据的强化学习(SSPO)。在冷启动阶段,模型被初始化以生成结构化心电图推理轨迹以及诊断预测。在强化学习阶段,训练由两个奖励信号引导:强制有效推理格式的结构奖励和评估诊断正确性的诊断奖励。