arXiv:2509.23368cs.CLcs.AI2025-09被引 1

小模型通过自我纠错机制,低成本实现医疗推理性能突破

MedCritical: Enhancing Medical Reasoning in Small Language Models via Self-Collaborative Correction

  • 让小模型与自己迭代对抗,用自身错误改进推理能力
  • 在CMExam上超越同类模型3.04%至10.12%,达7B级新纪录
  • 无需昂贵大模型评判,适合资源有限的医疗AI研发

在医学领域,临床诊断、治疗方案制定和医学知识整合等复杂推理任务对小语言模型构成严峻挑战,其表现常落后于GPT-4、Deepseek等大模型。现有基于知识蒸馏的方法依赖大模型作为裁判进行错误修正,但存在成本高、耗时长等问题。为此,我们提出新型两阶段框架MedCritical,利用大模型微调的小模型与自身进行对抗式自迭代。第一阶段从大模型提取高层与详细长链思维模板,引导学生模型生成更复杂的推理过程;第二阶段通过直接偏好优化(DPO)实现模型自学习,让学生模型在训练中以自身修正轨迹为参照,利用错误驱动洞察强化技能。该自学习方法在更低成本下达到与传统知识蒸馏相当的效果。值得注意的是,我们的MedCritical 7B模型在CMExam基准上分别优于Taiyi和Huatuo-o1-7B模型3.04%与10.12%,创下7B级小模型新SOTA性能。

原文摘要 · Abstract (English)

In the field of medicine, complex reasoning tasks such as clinical diagnosis, treatment planning, and medical knowledge integration pose significant challenges, where small language models often underperform compared to large language models like GPT-4 and Deepseek. Recent knowledge distillation-based methods aim to address these issues through teacher-guided error correction, but this LLM as judge approach remains challenging in terms of cost, time, and efficiency. To circumvent this issue, we propose a novel two-stage framework, MedCritical, which uses a small language model fine-tuned by a large teacher model to play against itself. In the first stage, we extract high-level and detailed long-chain thought templates from the teacher model to guide the student model to generate more complex reasoning thoughts. In the second stage, we introduce direct preference optimization (DPO) through model self-iteration collaboration to enhance the reasoning ability of the student model by playing against the correction trajectory of the fine-tuned model during training. This model self-learning DPO approach teaches the student model to use its own error-driven insights to consolidate its skills and knowledge to solve complex problems, and achieves comparable results to traditional knowledge distillation methods using teacher models at a lower cost. Notably, our MedCritical 7B model outperforms the Taiyi and Huatuo-o1-7B models by 3.04\% and 10.12\% respectively on the CMExam benchmark, achieving new SOTA performance among 7B-class small models.

医疗AI小模型自我纠错推理增强

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