AnesSuite: 一个用于LLM麻醉推理的全面基准和数据集套件
AnesSuite: A Comprehensive Benchmark and Dataset Suite for Anesthesiology Reasoning in LLMs
AI总结:
AnesSuite为LLM麻醉推理提供首个全面基准和数据集,开发Morpheus模型在麻醉领域实现显著性能提升。
摘要:
大语言模型(LLMs)在医疗领域的应用已经引起了广泛关注,然而它们在更专业的领域如麻醉学中的推理能力仍缺乏深入研究。为填补这一空白,我们引入了AnesSuite,这是首个专门为LLM中的麻醉推理设计的全面数据集套件。该套件包含AnesBench,一个专门评估麻醉相关推理的评估基准,涵盖三个层次:事实检索(System 1)、混合推理(System 1.x)和复杂决策制定(System 2)。除了这个基准,该套件还包括三个训练数据集,为持续预训练(CPT)、监督微调(SFT)和带有可验证奖励的强化学习(RLVR)提供基础设施。利用该套件,我们开发了Morpheus,首个用于麻醉推理的基线模型集合。尽管经过有限的SFT和群体相对策略优化(GRPO)训练,Morpheus不仅在麻醉领域实现了与大规模模型相当的显著改进,还展示了在通用医疗和广泛领域基准上的增强推理能力。此外,通过全面的评估和实验,我们分析了影响麻醉推理性能的关键因素,包括模型特性、训练策略和训练数据。AnesSuite和Morpheus将在https://github.com/MiliLab/AnesSuite上开源。
英文摘要:
The application of large language models (LLMs) in the medical field has garnered significant attention, yet their reasoning capabilities in more specialized domains like anesthesiology remain underexplored. To bridge this gap, we introduce AnesSuite, the first comprehensive dataset suite specifically designed for anesthesiology reasoning in LLMs. The suite features AnesBench, an evaluation benchmark tailored to assess anesthesiology-related reasoning across three levels: factual retrieval (System 1), hybrid reasoning (System 1.x), and complex decision-making (System 2). Alongside this benchmark, the suite includes three training datasets that provide an infrastructure for continued pre-training (CPT), supervised fine-tuning (SFT), and reinforcement learning with verifiable rewards (RLVR). Leveraging this suite, we develop Morpheus, the first baseline model collection for anesthesiology reasoning. Despite undergoing limited training with SFT and group relative policy optimization (GRPO), Morpheus not only achieves substantial improvements in anesthesiology that rival larger-scale models, but also demonstrates enhanced reasoning capabilities across general medical and broad-domain benchmarks. Furthermore, through comprehensive evaluations and experiments, we analyze the key factors influencing anesthesiology reasoning performance, including model characteristics, training strategies and training data. Both AnesSuite and Morpheus will be open-sourced at https://github.com/MiliLab/AnesSuite.
机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。