arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2310.06239cs.CLcs.AI

模型微调还是提示微调?面向临床概念与关系抽取的大语言模型研究

Model Tuning or Prompt Tuning? A Study of Large Language Models for Clinical Concept and Relation Extraction

University of Florida(佛罗里达大学);University of Florida Health Cancer Center(佛罗里达大学健康癌症中心);NVIDIA(英伟达)

Cheng Peng, Xi Yang, Kaleb E Smith, Zehao Yu, Aokun Chen, Jiang Bian, Yonghui Wu

AI总结:

本研究对比大语言模型的4种训练策略在临床概念与关系抽取任务的表现,验证了软提示的优势及冻结大模型在跨机构迁移、小样本场景的应用潜力。

摘要:

研究目的:开发面向大语言模型(LLM)的基于软提示的学习算法,探究提示的形态、使用冻结/未冻结LLM的提示微调效果、迁移学习以及小样本学习能力。研究方法:我们开发了一个基于软提示的LLM模型,对比了4种训练策略:(1)无提示的微调;(2)未冻结LLM的硬提示;(3)未冻结LLM的软提示;(4)冻结LLM的软提示。我们在两个基准数据集上,采用4种训练策略对7个预训练LLM进行临床概念抽取和关系抽取的评估,同时评估了基于提示的学习算法在跨机构场景下的迁移学习能力,还评估了其小样本学习能力。研究结果与结论:当LLM未冻结时,采用软提示的GatorTron-3.9B在概念抽取任务上取得了最优的严格F1值,分别为0.9118和0.8604,相比传统微调模型和基于硬提示的模型分别高出0.6~3.1%和1.2~2.9%;采用软提示的GatorTron-345M在端到端关系抽取任务上取得了最优F1值,分别为0.8332和0.7488,相比另外两种模型分别高出0.2~2%和0.6~11.7%。当LLM被冻结时,小型(即3.45亿参数)LLM与未冻结模型的性能存在较大差距;将LLM规模扩展到数十亿参数级别后,冻结LLM的性能可与未冻结LLM相当。在跨机构评估中,采用软提示的冻结GatorTron-8.9B模型取得了最优性能。本研究表明:(1)机器学习软提示的效果优于人工设计的提示;(2)冻结LLM具备更优的小样本学习能力和迁移学习能力,可助力多机构应用落地;(3)冻结LLM的应用需要大模型规模的支撑。

英文摘要:

Objective To develop soft prompt-based learning algorithms for large language models (LLMs), examine the shape of prompts, prompt-tuning using frozen/unfrozen LLMs, transfer learning, and few-shot learning abilities. Methods We developed a soft prompt-based LLM model and compared 4 training strategies including (1) fine-tuning without prompts; (2) hard-prompt with unfrozen LLMs; (3) soft-prompt with unfrozen LLMs; and (4) soft-prompt with frozen LLMs. We evaluated 7 pretrained LLMs using the 4 training strategies for clinical concept and relation extraction on two benchmark datasets. We evaluated the transfer learning ability of the prompt-based learning algorithms in a cross-institution setting. We also assessed the few-shot learning ability. Results and Conclusion When LLMs are unfrozen, GatorTron-3.9B with soft prompting achieves the best strict F1-scores of 0.9118 and 0.8604 for concept extraction, outperforming the traditional fine-tuning and hard prompt-based models by 0.6~3.1% and 1.2~2.9%, respectively; GatorTron-345M with soft prompting achieves the best F1-scores of 0.8332 and 0.7488 for end-to-end relation extraction, outperforming the other two models by 0.2~2% and 0.6~11.7%, respectively. When LLMs are frozen, small (i.e., 345 million parameters) LLMs have a big gap to be competitive with unfrozen models; scaling LLMs up to billions of parameters makes frozen LLMs competitive with unfrozen LLMs. For cross-institute evaluation, soft prompting with a frozen GatorTron-8.9B model achieved the best performance. This study demonstrates that (1) machines can learn soft prompts better than humans, (2) frozen LLMs have better few-shot learning ability and transfer learning ability to facilitate muti-institution applications, and (3) frozen LLMs require large models.

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

↑