arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2603.08166cs.CL

RexDrug:通过增强推理的大型语言模型实现可靠的多药物组合提取

RexDrug: Reliable Multi-Drug Combination Extraction through Reasoning-Enhanced LLMs

Zhijun Wang, Ling Luo, Dinghao Pan, Huan Zhuang, Lejing Yu, Yuanyuan Sun, Hongfei Lin

AI总结:

RexDrug通过增强推理的大型语言模型,实现了可靠的多药物组合提取,优于现有方法并在二元和n元任务中均表现优异。

摘要:

从大规模生物医学文献中自动提取药物组合(DCE)对于推进精准医学和药理学研究至关重要。然而,现有关系提取方法主要关注二元相互作用,难以建模变量长度的n元药物组合,其中复杂的兼容性逻辑和分布证据需要考虑。为了解决这些限制,我们提出了RexDrug,一种基于大型语言模型的端到端推理增强的关系提取框架,用于n元药物组合提取。RexDrug采用两阶段训练策略。首先,利用多代理协作机制自动生成高质量的专家级推理轨迹用于监督微调。第二,应用强化学习,使用专门针对DCE设计的多维奖励函数进一步优化推理质量和提取准确性。在DrugComb数据集上的广泛实验表明,RexDrug在n元提取方面始终优于最先进的基线。对DDI13语料库的额外评估证实了其在二元药物-药物相互作用任务中的泛化能力。人类专家评估和自动推理指标进一步表明,RexDrug能够生成连贯的医学推理并准确识别复杂的治疗方案。这些结果确立了RexDrug作为从非结构化文本中提取复杂生物医学关系的可扩展且可靠的解决方案。源代码和数据可在https://github.com/DUTIR-BioNLP/RexDrug获取。

英文摘要:

Automated Drug Combination Extraction (DCE) from large-scale biomedical literature is crucial for advancing precision medicine and pharmacological research. However, existing relation extraction methods primarily focus on binary interactions and struggle to model variable-length n-ary drug combinations, where complex compatibility logic and distributed evidence need to be considered. To address these limitations, we propose RexDrug, an end-to-end reasoning-enhanced relation extraction framework for n-ary drug combination extraction based on large language models. RexDrug adopts a two-stage training strategy. First, a multi-agent collaborative mechanism is utilized to automatically generate high-quality expert-like reasoning traces for supervised fine-tuning. Second, reinforcement learning with a multi-dimensional reward function specifically tailored for DCE is applied to further refine reasoning quality and extraction accuracy. Extensive experiments on the DrugComb dataset show that RexDrug consistently outperforms state-of-the-art baselines for n-ary extraction. Additional evaluation on the DDI13 corpus confirms its generalizability to binary drugdrug interaction tasks. Human expert assessment and automatic reasoning metrics further indicates that RexDrug produces coherent medical reasoning while accurately identifying complex therapeutic regimens. These results establish RexDrug as a scalable and reliable solution for complex biomedical relation extraction from unstructured text. The source code and data are available at https://github.com/DUTIR-BioNLP/RexDrug

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑