arXiv:2609.02898cs.CL2026-09

小模型也能懂医学:用知识迁移让轻量模型媲美大模型

Distilled Rapid Embedding Transfer (DRET): Parameter-Efficient Biomedical Domain Adaptation via Priority-Based Embedding Transfer

论文配图:Distilled Rapid Embedding Transfer (DRET): Parameter-Efficient Biomedical Domain Adaptation via Priority-Based Embedding Transfer
图 1 · 摘自论文原文
  • 从大医学模型提取关键嵌入,注入小通用模型无需重新训练
  • 在医学文本分类任务中达到大模型水平,参数仅66M
  • 适合资源有限但需高精度医学文本分析的场景

BioBERT和ClinicalBERT等专用大模型在生物医学自然语言处理任务中表现优异,但计算开销大,难以部署。通用轻量模型如DistilBERT虽高效,却缺乏领域知识。本文提出一种名为DRET的知识迁移方法,将大模型中的生物医学知识注入小模型,无需在原始数据上重新训练。DRET采用多阶段策略:统一分词器合并(DRET 1.x)、混合嵌入平均(DRET 2.0)、基于优先级的嵌入选择机制(DRET 3.x),并结合嵌入层冻结、差异化学习率、标签传播和不平衡损失函数(DRET 4.x)。在EBM-NLP语料库上评估时,增强后的DistilBERT(66M参数)在十二项指标中,平衡准确率、召回率与ROC-AUC均达到甚至超过参数量大一倍的模型,同时保持原有轻量化优势。通过余弦相似度、语义偏移和t-SNE分析证实知识确实在嵌入层面迁移。DRET为生物医学文本挖掘提供了一条可扩展、低资源的近专家级性能路径,可直接应用于自动化系统性文献回顾与临床决策支持。

原文摘要 · Abstract (English)

Large domain-specific language models such as BioBERT and ClinicalBERT achieve strong performance on biomedical NLP tasks, but their computational demands make them impractical for many real-world deployments. General-purpose, parameter-efficient models such as DistilBERT are lightweight yet lack the domain knowledge required for specialized tasks such as PICO (Population, Intervention, Comparison, Outcome) classification. We introduce Distilled Rapid Embedding Transfer (DRET), a knowledge-transfer paradigm that injects biomedical domain knowledge from large specialized models into a smaller general-purpose model without retraining on the original specialized corpora. DRET is developed as an iterative family of strategies: a unified tokenizer-merge strategy (DRET 1.x), hybrid embedding averaging (DRET 2.0), and a priority-based embedding-transfer mechanism (DRET 3.x) that hierarchically selects embeddings from the most authoritative source models, further combined with embedding-layer freezing, differential learning rates, label propagation, and imbalance-aware loss functions (DRET 4.x). We evaluate DRET on token-level PICO classification using the EBM-NLP corpus under severe class imbalance, across a twelve-metric battery. DRET-enhanced DistilBERT (66M parameters) attains balanced accuracy, recall, and ROC-AUC competitive with, and on several class-wise metrics exceeding, models an order of magnitude larger, while retaining DistilBERT's efficiency. We further show that transfer occurs at the embedding level through cosine-similarity, semantic-shift, and t-SNE analyses. DRET offers a scalable, resource-efficient route to near-domain-expert performance for biomedical text mining, with direct application to automated systematic literature reviews and clinical decision support.

知识迁移轻量模型医学NLP参数高效

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