sebis at ArchEHR-QA 2026:单笔记本上能做多少?评估单个笔记本上的EHR问答系统 sebis at ArchEHR-QA 2026: How Much Can You Do Locally? Evaluating Grounded EHR QA on a Single Notebook
在单台消费级硬件上实现全本地隐私保护的EHR问答系统,小模型配置得当可接近大模型性能
前置知识
EHR问答系统
电子健康记录问答系统是指能够从患者的临床记录(包括医生笔记、化验结果、用药历史、诊断报告等)中检索并回答自然语言问题的系统。这类系统需要处理高度敏感的个人健康信息,理解医学专业术语,并且在临床环境中提供准确、可靠的答案。EHR问答不同于一般开放域问答,它要求答案必须严格基于患者记录中的事实,不能添加外部医学知识,以保证准确性和可追溯性。
本文研究的核心场景就是如何在单台本地设备上部署这样一个系统,理解EHR问答的特殊性和临床需求对于评估其本地部署的可行性至关重要
证据 grounding
证据grounding是指生成的每个答案句子都必须能够追溯到源文档中的具体句子作为支持证据。这不仅仅是简单的相关性检索,而是需要建立答案句子与证据句子之间的显式映射关系。在EHR问答中,grounding尤其重要,因为临床决策必须基于明确的证据来源,以提高系统的透明度和可信度。grounding任务通常被形式化为一个多对多的映射问题:每个答案句子可能需要多个证据句子支持,一个证据句子也可能支持多个答案句子。
本文研究的四个子任务中有两个直接涉及证据处理(Evidence Identification和Evidence Alignment),这是EHR问答与一般问答系统最核心的区别
模型量化
模型量化是一种减少模型内存占用和加速推理的技术,通过降低模型参数的数值精度来实现。常见的是4-bit或8-bit量化,将原本32-bit的浮点数参数转换为低精度表示。例如,MLX框架可以将大模型转换为4-bit精度的MXFP4格式,在Apple Silicon设备上高效运行。量化会带来一定的精度损失,但通过合适的校准和后处理,这种损失通常可以控制在可接受范围内。对于大模型来说,量化是实现在消费级硬件上运行的关键技术。
本文能够在Apple Silicon上运行120B参数的大模型,正是通过MLX框架的4-bit量化技术实现的,这是本地部署大模型的核心技术
Cross-encoder
Cross-encoder是一种用于计算两个文本之间相关性的模型架构。与先分别编码两个文本再计算相似度的bi-encoder不同,cross-encoder将两个文本拼接后一起输入到Transformer模型中,通过自注意力机制直接学习它们之间的交互关系。这种方式能够捕获更精细的语义匹配,但计算复杂度更高,特别是在需要计算一个查询与大量候选文档相关性时,必须为每个候选都运行一次完整的前向传播。MedCPT就是一个专门为生物医学文本设计的cross-encoder模型。
本文在Evidence Identification子任务中对比了cross-encoder与embedding-based方法,发现简单的embedding相似度反而优于微调的cross-encoder,这是一个反直觉的重要发现
研究动机
现有的EHR问答系统主要依赖于云端部署的大型语言模型,这在实际临床环境中面临严重障碍。首先,临床记录包含高度敏感的个人健康信息,受HIPAA和GDPR等严格隐私法规约束,许多医疗机构无法将EHR数据发送到外部云服务进行处理。其次,现代基于LLM的问答系统通常需要专门的硬件加速器或云端推理,而许多医疗机构缺乏这样的基础设施,医生通常只能使用标准工作站或笔记本硬件。这造成了一个研究与实践的严重脱节:虽然在研究环境中使用大模型取得了显著进展,但这些方法难以部署到真实的临床场景中。
本文的目标是本文的核心目标是探索在严格的本地部署约束下,EHR问答系统能够达到怎样的性能水平。具体来说,作者参与ArchEHR-QA 2026共享任务的所有四个子任务,评估多种能够在商品硬件上运行的策略。所有实验(包括训练、推理和评估)都在本地完成,不依赖外部API或云基础设施。通过这个目标,作者希望回答一个关键问题:隐私保护、完全本地的EHR问答系统是否可以用现有模型和可访问的硬件实现。
与已有工作不同的是,本文的独特切入点在于它是在全本地部署约束下对grounded EHR问答系统的全面评估。虽然有研究开始关注临床NLP的隐私保护本地部署(如Griot等人2025年在医院EHR系统中实现了符合GDPR的LLM助手),但很少有工作系统性地比较各种grounded EHR问答方法在所有训练和推理都必须在单个本地设备上时的性能。大多数现有研究假设充足的计算资源,留下本地、隐私保护部署的问题未被充分探索。本文填补了这个研究空白,通过在Apple Silicon设备上的系统性实验,揭示了哪些架构和策略在严格本地部署约束下仍然有效。
核心方法
本文的方法整体思路是针对EHR问答流水线的不同阶段(问题理解、证据识别、答案生成、证据对齐),分别探索适合本地部署的模型和策略。作者将四个子任务视为一个完整的问答流水线,对每个任务尝试多种方法:包括微调BERT风格的分类器、基于嵌入的检索方法、以及小型或量化的语言模型用于答案生成。整个实验设计强调能在商品硬件上运行这一约束,所有模型都经过精心选择,参数量从4B到120B不等,并且使用Apple Silicon的MLX框架进行量化以支持本地执行。
本文的核心创新点在于证明了简单的out-of-the-box嵌入模型在证据提取和对齐任务上竟然能够超越精心微调的cross-encoder,这是一个反直觉的发现。通常认为fine-tuned cross-encoder应该更擅长细粒度的相关性判断,但本文的实验表明,在低资源临床场景下,简单的余弦相似度计算可能更鲁棒。此外,作者还提出了使用本地大模型生成合成数据来稳定BERT风格分类器训练的策略,这是一个缓解低资源场景下训练不稳定问题的创新方法。另一个关键创新是展示了量化和小型语言模型在本地处理生成任务上的高度能力,证明了模型规模不是性能的唯一驱动因素。
方法步骤详情
对于Question Interpretation子任务,作者使用三个模型:Qwen3-4B、Qwen2.5-14B和gpt-oss-120b。他们评估了不同的few-shot prompting设置,使用开发集的前3或5个案例作为示例。具体测试了Qwen3-4B的2种3-shot prompts、Qwen2.5-14B的3种3-shot prompts,以及gpt-oss-120b的3种3-shot和4种5-shot prompts。此外,作者还实验了两步方法:先用Qwen2.5-14B提供初始答案,再用Qwen3-4B修订得到最终答案。对于Evidence Identification子任务,作者探索了基于检索的方法和监督分类方法。为缓解开发集规模有限的问题,作者使用本地部署的Llama3.1-70B生成了合成数据:为每个原始开发案例生成10个新合成案例,共200个新案例,包含1,818个句子并标注了三类相关性标签。使用这个合成数据集训练cross-encoder证据分类器,采用BERT风格架构,特征是共享表示层同时输入到两个不同的任务头:3-way细粒度分类头和2-way二分类头。为防止合成案例主导训练信号,将真实案例上采样和合成数据下采样,实现严格的1:1比例。对于检索方法,使用Qwen3-Embedding-8B和MedCPT-Cross-Encoder两个模型计算q_{clin}与每个s_i的相似度分数,然后基于开发集上严格micro F1分数确定决策阈值t,将E = { s_i ∈ D | score(q_{clin}, s_i) > t }定义为相关句子集合。对于Answer Generation子任务,作者实验了两种设置:包含患者问题q、医生解释问题q_{clin}和临床记录片段D的0-shot prompt(使用gpt-oss-120b);以及使用q和D的两步0-shot方法(使用Qwen3-4B)。对于Evidence Alignment子任务,作者探索了基于嵌入和生成模型的out-of-the-box方法,以及监督分类方法。fine-tuning alignment cross-encoder时,使用BERT风格架构建模查询(q和q_{clin})、生成答案A和源证据D之间的成对关系。具体在每个a_i ∈ A和s_j ∈ D之间进行分类,输出相关决策。不同于Subtask 2的多头设置,这里采用标准二分类头。作者还通过重用Subtask 2生成的合成数据作为pseudo-alignments来扩展训练语料,保持真实和合成数据1:1比例。对于out-of-the-box模型,考虑四种方法:使用Qwen3-Embedding-8B的阈值方法(类似于Subtask 2);使用Qwen3-4B的两步prompting list-wise对齐;使用Qwen3.5-35B-A3B的0-shot prompt作为二分类器的pair-wise对齐;使用Qwen3.5-35B-A3B的1-shot list-wise对齐(类似于第二种方法但没有reformatting步骤)。对于生成推理实验,作者使用默认解码参数,温度为0.7,top-p值为0.9,没有进行额外的调优。对于Subtask 2和Subtask 4的分类器fine-tuning,作者进行网格搜索识别最优超参数,搜索空间包括学习率{1 × 10^{-5}, 2 × 10^{-5}, 3 × 10^{-5}, 5 × 10^{-5}}、训练轮次{1, 2}和dropout概率{0.10, 0.20}。最优参数通过在开发集上进行5-fold交叉验证的性能评估来确定。
技术新颖性
本文的技术新颖性体现在三个方面。首先,系统性评估了grounded EHR问答在全本地部署约束下的性能,填补了临床NLP领域的一个重要研究空白。大多数现有工作关注模型本身的性能提升,而本文关注的是部署约束下的性能权衡,这在真实临床场景中同样重要。其次,发现了out-of-the-box dense embedding模型在证据提取和对齐任务上超越fine-tuned cross-encoder的反直觉现象。这个发现表明在某些低资源临床场景下,简单的方法可能比复杂的fine-tuning更鲁棒,这对未来的研究方向有重要启示。第三,提出了使用本地大模型生成合成数据来稳定BERT风格分类器训练的pipeline,并且通过平衡真实和合成数据比例来防止合成数据主导训练信号,这是一个实用的数据增强策略。此外,本文还展示了量化和小型语言模型在本地执行生成任务上的高度能力,证明了适当的配置和训练策略可以让小模型达到接近大模型的性能,这对资源受限环境具有重要的实践意义。
实验结果
核心发现可以总结为四点。第一,在Question Interpretation子任务中,虽然大模型通常达到更强的性能,但与较小模型的性能差距仍然适中。在开发集上,单查询方法BERTScore为36.93,双查询为37.29,两步方法为33.10。在测试集上,整体得分分别为25.63、25.61和24.59,表明prompt重复(根据Leviathan等人2025年的提议)与单查询方法相比没有显著优势,虽然在四个评估指标中有三个指标测试结果稍高。第二,Evidence Identification子任务揭示了许多方法的强召回偏差。值得注意的是,将所有候选分类为相关的基线实现了完美的召回和惊人的竞争性F1性能。开发集上基线F1为44.08,而Qwen3-Embedding-8B F1为51.74。这表明数据集隐式地奖励高召回,而可靠识别真正不相关的上下文仍然具有挑战性。更高级的方法如embedding相似度或cross-encoder排序通常会增加精确度但往往以降低召回为代价。有趣的是,使用HYDRA和合成数据训练的模型在某些情况下表现出更平衡的精确度-召回权衡。第三,在Answer Generation子任务中,开发集上Qwen3-4B整体得分27.76,gpt-oss-120b为32.38;测试集上gpt-oss-120b为31.51。BERTScore在测试集上分别为30.95和39.30,表明大模型仍有优势但差距可控。第四,Evidence Alignment子任务显示出开发集和测试集性能之间的巨大差异。开发集上Qwen3-Embedding-8B的Micro-F1为22.01,测试集上达到59.45;Qwen3.5-35B list-wise在开发集上为25.74,测试集上达到74.84。作者假设这可能是由两个分割之间的分布偏移引起的,这也解释了fine-tuned模型在测试集上与out-of-the-box模型相比观察到的严重退化。比较DeBERTa和Bio_ClinicalBERT进一步突出了领域特定建模的重要性。在Subtask 2中,biomedical模型始终比通用模型表现更好,表明该任务包含相当多的领域特定术语和上下文线索,专用预训练更好地捕获了这些线索。与其它子任务不同,Subtask 4显示出开发集和测试集性能之间的巨大差异,测试集上的Bio_ClinicalBERT F1仅为8.33,远低于开发集的21.47。同样在Subtask 4中,合成数据改善了通用领域模型(DeBERTa)的性能,但没有改善biomedical模型(Bio_ClinicalBERT)。考虑到Subtask 4中观察到的更平衡的精确度-召回行为,这表明当模型预测高度偏斜时(如Subtask 2的情况),合成数据可能特别有益。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| Question Interpretation | BERTScore | 36.93 (测试集,单查询方法) | 将所有候选分类为相关 | 此任务无传统基线,本文建立了本地部署基线 |
| Question Interpretation | Overall Score | 25.63 (测试集,单查询方法) | 无直接可比基线 | 在竞争排行榜上表现高于平均水平 |
| Evidence Identification | Micro-F1 | 51.61 (测试集,Qwen3-Embedding-8B) | 44.08 (开发集,所有句子相关基线) | 测试集性能显著优于朴素基线,embedding方法稳健 |
| Evidence Identification | Micro-F1 | 44.43 (测试集,Bio_ClinicalBERT-synth-HYDRA) | 50.61 (开发集,同一方法) | 测试集性能下降表明分布偏移问题,合成数据帮助稳定训练 |
| Answer Generation | BERTScore | 39.30 (测试集,gpt-oss-120b) | 30.95 (测试集,Qwen3-4B) | 大模型性能领先23.6%,但小模型差距可控 |
| Answer Generation | Overall Score | 31.51 (测试集,gpt-oss-120b) | 无直接基线 | 在竞争排行榜上表现低于平均水平但与其他系统相当 |
| Evidence Alignment | Micro-F1 | 74.84 (测试集,Qwen3.5-35B list-wise) | 14.73 (开发集,所有句子相关基线) | 显著优于朴素基线,大型生成模型在list-wise对齐上表现优异 |
| Evidence Alignment | Micro-F1 | 59.45 (测试集,Qwen3-Embedding-8B) | 22.01 (开发集,同一方法) | 测试集性能大幅提升,embedding方法在测试集上非常鲁棒 |
| Evidence Alignment | Micro-F1 | 8.33 (测试集,Bio_ClinicalBERT) | 21.47 (开发集,同一方法) | 严重退化表明fine-tuned模型过拟合开发分布,out-of-the-box方法更鲁棒 |
局限与改进
本文的局限性主要体现在三个方面。首先,虽然本文的设置避免了外部云基础设施,但它仍然依赖高端商品硬件(例如配备M4 Pro芯片和48GB内存的Apple MacBook),这在所有资源不足的临床环境中可能不易获得。这限制了在更受限环境(如普通笔记本或移动设备)中的直接应用。其次,本文的评估受限于ArchEHR-QA数据集,这限制了发现跨不同临床专科或多语言患者群体的直接可泛化性。该数据集虽然精心标注,但只代表了一种特定的问答场景(患者询问住院期间的临床过程),不能涵盖EHR问答的所有应用场景。第三,作者承认生成AI在医疗保健中固有的患者安全风险。所评估的模型容易出现hallucinations和错误结论。在临床环境中,如果盲目接受这些错误,可能会对患者护理产生负面影响。虽然本地部署避免了数据传输到云端的风险,但模型本身的错误风险仍然存在。作者还提到了数据集本身的局限性:低资源数据可能导致了fine-tuned cross-encoders的一般化挑战,开发集和测试集之间可能的分布偏移导致了Subtask 4中观察到的性能差异。
独立分析的弱点
本文独立分析的第一个弱点是合成数据生成策略虽然创新,但完全依赖LLaMA-3-70B模型的质量,且生成过程的质量阈值(句子长度10-500字符、相关性标签比例约束等)是人工设定的,可能无法完全捕捉真实临床数据的细微差别。合成数据在Subtask 2中帮助了通用模型但未帮助biomedical模型,在Subtask 4中甚至出现了相反的效果,这表明合成数据的策略需要针对不同模型和任务进行更精细的调整。改进方向可以是探索更先进的合成数据生成方法,如结合大型代理数据集或使用更复杂的质量控制机制。第二个弱点是对于Subtask 2,所有方法都难以显著优于所有句子相关的基线,这表明该任务的定义可能存在问题。基线实现完美召回和竞争性F1性能,说明数据集可能隐式地奖励高召回,而真正不相关上下文的可靠识别仍然具有挑战性。改进方向可以重新思考任务定义,例如增加负样本的权重或设计更精确的评估指标。第三个弱点是fine-tuned模型在测试集上出现严重退化(如Subtask 4中Bio_ClinicalBERT从开发集的21.47降至测试集的8.33),这表明模型过拟合了开发分布,而开发分布可能不能完全反映测试集的特征。改进方向可以包括更robust的正则化技术、domain adaptation方法,或者使用更多样化的训练数据。第四个弱点是实验硬件仍然偏向高端(Apple M4 Pro with 48GB),这限制了在更受限环境中的应用。改进方向可以探索更小、高度优化的语言模型,能够在高度受限的边缘设备(如手机或临床平板)上执行,进一步降低入门门槛。
未来方向
作者提出的未来工作方向包括探索更小、高度优化的语言模型,能够在高度受限的边缘设备(如移动设备或临床平板)上执行,进一步降低入门门槛。此外,由于这个共享任务在低资源数据下运行,这可能导致了fine-tuned cross-encoders的一般化挑战,未来研究应该探索集成来自相关临床领域的大规模代理数据集以及更先进的合成数据生成方法,以更好地稳定模型fine-tuning并提高对分布偏移的鲁棒性。基于本文成果可以延伸的方向包括:研究在更多样化的临床场景和多语言环境中的本地EHR问答性能;探索更高效的模型架构和量化技术,使大模型能够在更受限的硬件上运行;研究human-in-the-loop的机制,结合临床医生的专业知识来提高答案的准确性和可靠性;探索更robust的grounding机制,能够处理更复杂的推理和跨多个临床记录的信息综合;研究模型的持续学习和适应能力,使其能够随着新数据的积累而不断改进,而不需要重新训练整个模型。
复现评估
本文的复现评估情况如下。代码已开源在https://github.com/ibrahimey/ArchEHR-QA-2026.Mar,这为复现提供了基础。使用的ArchEHR-QA 2026数据集是公开的,包含开发集20个案例、测试集100个案例和test-2026集47个案例,每个案例包括患者问题、医生解释问题、临床记录片段、医生答案、证据链接和临床专科等多层专家标注。硬件要求相对较高:大多数实验在Apple MacBook M4 Pro with 48GB unified memory上进行,这足以训练分类器模型并运行大多数推理实验;对于gpt-oss-120B模型的实验,使用了Apple Mac Studio M3 Max with 96GB内存。所有模型都从Hugging Face获取,包括量化的MLX版本和标准版本。复现难度中等偏上,主要挑战在于硬件要求和本地部署大模型的技术细节。不过,由于所有实验都在本地进行且没有使用外部API,一旦硬件准备就绪,复现过程应该是确定性的。作者严格遵循了共享任务的官方评估脚本,确保了结果的可靠性。
论文图表