arXiv:2410.15702cs.CL2024-10EMNLP被引 18

用对比解码减少大模型在医疗信息抽取中的幻觉问题

Mitigating Hallucinations of Large Language Models in Medical Information Extraction via Contrastive Decoding

  • 将医疗信息抽取拆分为识别与分类,分步优化
  • 交替对比子任务模型输出,提升准确率12.3%以上
  • 适合医疗文本提取场景,尤其对幻觉敏感应用

大语言模型在医疗领域的应用前景广阔,但临床环境复杂,易产生幻觉,限制其落地。本文针对医疗信息抽取(MIE)任务,提出交替对比解码(ALCD)方法。将MIE重新定义为识别-分类流程,通过微调时有选择地掩码优化令牌,分离模型的识别与分类能力。推理阶段,交替对比子任务模型生成的输出分布,以增强特定能力并抑制无关能力。同时引入交替自适应约束策略,动态调节对比范围。在两种不同骨干模型和六个多样化医疗信息抽取任务上验证,相较传统解码方法显著降低幻觉,性能提升超12.3%。

原文摘要 · Abstract (English)

The impressive capabilities of large language models (LLMs) have attracted extensive interests of applying LLMs to medical field. However, the complex nature of clinical environments presents significant hallucination challenges for LLMs, hindering their widespread adoption. In this paper, we address these hallucination issues in the context of Medical Information Extraction (MIE) tasks by introducing ALternate Contrastive Decoding (ALCD). We begin by redefining MIE tasks as an identify-and-classify process. We then separate the identification and classification functions of LLMs by selectively masking the optimization of tokens during fine-tuning. During the inference stage, we alternately contrast output distributions derived from sub-task models. This approach aims to selectively enhance the identification and classification capabilities while minimizing the influence of other inherent abilities in LLMs. Additionally, we propose an alternate adaptive constraint strategy to more effectively adjust the scale and scope of contrastive tokens. Through comprehensive experiments on two different backbones and six diverse medical information extraction tasks, ALCD demonstrates significant improvements in resolving hallucination issues compared to conventional decoding methods.

大模型幻觉医疗信息抽取对比解码

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