arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2302.04725cs.CLcs.AIcs.LG
收藏

用于临床自然语言处理的轻量级Transformer

Lightweight Transformers for Clinical Natural Language Processing

University of Oxford(牛津大学);NLPie Research(NLPie研究院);University of Surrey(萨里大学);Oxford-Suzhou Centre for Advanced Research(牛津苏州高等研究院);Nuffield Department of Population Health, University of Oxford(牛津大学纳菲尔德人口健康系);Pandemic Sciences Institute, University of Oxford(牛津大学流行病科学研究所);Sharif University of Technology(谢里夫理工大学);Infectious Diseases Data Observatory (IDDO), University of Oxford(牛津大学传染病数据观测站(IDDO))

Omid Rohanian, Mohammadmahdi Nouriborji, Hannah Jauncey, Samaneh Kouchaki, ISARIC Clinical Characterisation Group, Lei Clifton, Laura Merson, David A. Clifton

AI总结:

本文通过知识蒸馏和持续学习开发了参数为1500万至6500万的轻量级临床Transformer,在多个临床文本挖掘任务上性能媲美BioBERT等大模型,并显著优于其他紧凑模型。

摘要:

专门的预训练语言模型在自然语言处理(NLP)中正变得越来越普遍,因为它们有可能优于在通用文本上训练的模型。BioBERT和BioClinicalBERT是这类模型的两个例子,它们在医学NLP任务中已展现出潜力。这些模型中有许多存在参数过多和资源密集的问题,但借助知识蒸馏(KD)等技术,可以创建更小的版本,其性能几乎与较大的对应模型相当。在这项工作中,我们特别关注开发用于处理临床文本(即病程记录、出院小结等)的紧凑语言模型。我们利用知识蒸馏和持续学习开发了多个高效的轻量级临床Transformer,参数数量从1500万到6500万不等。这些模型的性能与BioBERT和ClinicalBioBERT等较大模型相当,并且显著优于其他在通用数据或生物医学数据上训练的紧凑模型。我们在多个标准数据集上进行了广泛评估,涵盖了一系列临床文本挖掘任务,包括自然语言推理、关系抽取、命名实体识别和序列分类。据我们所知,这是第一项专门针对为临床NLP任务创建高效紧凑Transformer的全面研究。本研究中使用的模型和代码可分别在我们的Huggingface主页https://huggingface.co/nlpie和Github页面https://github.com/nlpie-research/Lightweight-Clinical-Transformers上找到,以促进我们结果的可复现性。

英文摘要:

Specialised pre-trained language models are becoming more frequent in NLP since they can potentially outperform models trained on generic texts. BioBERT and BioClinicalBERT are two examples of such models that have shown promise in medical NLP tasks. Many of these models are overparametrised and resource-intensive, but thanks to techniques like Knowledge Distillation (KD), it is possible to create smaller versions that perform almost as well as their larger counterparts. In this work, we specifically focus on development of compact language models for processing clinical texts (i.e. progress notes, discharge summaries etc). We developed a number of efficient lightweight clinical transformers using knowledge distillation and continual learning, with the number of parameters ranging from 15 million to 65 million. These models performed comparably to larger models such as BioBERT and ClinicalBioBERT and significantly outperformed other compact models trained on general or biomedical data. Our extensive evaluation was done across several standard datasets and covered a wide range of clinical text-mining tasks, including Natural Language Inference, Relation Extraction, Named Entity Recognition, and Sequence Classification. To our knowledge, this is the first comprehensive study specifically focused on creating efficient and compact transformers for clinical NLP tasks. The models and code used in this study can be found on our Huggingface profile at https://huggingface.co/nlpie and Github page at https://github.com/nlpie-research/Lightweight-Clinical-Transformers, respectively, promoting reproducibility of our results.

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

↑