HyperAIHyperAI

Command Palette

Search for a command to run...

3년 전

임상 및 생의학 작업에 적용된 지시 파인튜닝된 대규모 언어 모델의 제로샷 및 퓨샷 연구

Yanis Labrak Mickael Rouvier Richard Dufour

대형 언어 모델, 프롬프트 프로그래밍 및 퓨샷 태스크

노트북으로 이동

초록

최근 대규모 언어 모델(LLMs)의 등장은 자연어 처리(NLP) 분야에서 상당한 진전을 가능하게 했습니다. 이러한 새로운 모델들은 다양한 작업에서 우수한 성능을 입증했지만, 그들이 처리할 수 있는 작업의 다양성과 적용 분야 측면에서 그 적용 가능성과 잠재력은 아직 충분히 탐구되지 않았습니다. 이러한 맥락에서, 우리는 13개의 실제 임상 및 생의학 NLP 작업(명사구 인식(NER), 질문-답변(QA), 관계 추출(RE) 등 포함)에 대해 4가지 최첨단 지시 튜닝 LLMs(ChatGPT, Flan-T5 UL2, Tk-Instruct, Alpaca)를 평가했습니다. 우리의 전체 결과는 이러한 평가된 LLMs가 대부분의 작업에서 제로샷 및 퓨샷 시나리오에서 최첨단 모델의 성능에 근접하며, 특히 QA 작업에서 뛰어난 성능을 보인다는 것을 보여주었습니다. 이는 이러한 작업들의 예제를 이전에 접한 적이 없음에도 불구하고 나타나는 현상입니다. 그러나 우리는 또한 분류 및 RE 작업이 PubMedBERT와 같은 의학 분야에 특화된 모델로 훈련된 모델이 달성할 수 있는 성능에는 미치지 못한다는 것도 관찰했습니다. 마지막으로, 우리는 모든 연구된 작업에서 모든 다른 모델보다 더 우수한 단일 LLM이 없으며, 일부 모델은 다른 모델보다 특정 작업에 더 적합하다는 것을 주목했습니다.

One-sentence Summary

Evaluating four instruction-tuned large language models (ChatGPT, Flan-T5 UL2, Tk-Instruct, and Alpaca) across thirteen English clinical and biomedical NLP tasks reveals that while these models approach state-of-the-art zero- and few-shot performance, particularly in question answering, they underperform domain-specific architectures like PubMedBERT in classification and relation extraction and demonstrate that no single model outperforms all others across the evaluated tasks.

Key Contributions

  • This study evaluates four instruction-tuned large language models (ChatGPT, Flan-T5 UL2, Tk-Instruct, and Alpaca) across thirteen clinical and biomedical natural language processing tasks under zero- and few-shot conditions.
  • A novel Recursive Chain-of-Thought (RCoT) prompting strategy is introduced to sequentially enrich input instructions, enabling named-entity recognition across diverse model architectures.
  • Empirical analysis demonstrates that general-purpose models achieve competitive performance on most biomedical tasks, though they underperform domain-specialized architectures like PubMedBERT in classification and relation extraction while exhibiting distinct task-specific capabilities.

Introduction

The medical domain increasingly relies on natural language processing to analyze complex clinical records, yet deploying advanced models in healthcare remains difficult due to scarce, sensitive datasets and the high cost of expert annotation. Traditional masked language models require extensive labeled data and struggle with cross-task generalization, while prior evaluations of large language models in medicine have been limited to narrow task sets and non-standard automatic metrics. The authors address these gaps by benchmarking four instruction-tuned models across thirteen real-world clinical and biomedical tasks. They evaluate zero- and few-shot capabilities against a fine-tuned PubMedBERT baseline using standard accuracy and F1 scores. Furthermore, the authors introduce Recursive Chain-of-Thought prompting, a novel technique that sequentially enriches prompts to mimic human reasoning and enable named entity recognition across diverse large language model architectures.

Dataset

Dataset Composition and Sources

  • The authors use five scientific and medical NLP datasets: MedMCQA, GAD, SciTail, HoC, and DEFT-2020. These sources span multiple-choice question answering, relation extraction, natural language inference, cancer hallmark classification, and semantic textual similarity.

Key Details for Each Subset

  • MedMCQA provides science questions with four lettered options requiring single-character answers.
  • GAD contains statements for gene-disease relation extraction, classified as positive or negative.
  • SciTail offers premise-hypothesis pairs for natural language inference, labeled as entails or neutral.
  • HoC documents are annotated with one or more of ten predefined cancer hallmarks or marked as none.
  • DEFT-2020 Task 1 functions as the semantic textual similarity benchmark.
  • The provided excerpts do not specify exact dataset sizes or training mixture ratios.

Data Usage and Processing

  • The authors convert all subsets into structured instruction prompts for ChatGPT and Flan-T5 UL2.
  • Each task is evaluated using both zero-shot and five-shot configurations.
  • Raw class labels are manually optimized through trial and error to align with model expectations, such as changing entailment to entails for measurable performance gains.
  • Strict output constraints are enforced across all prompts to prohibit justifications and guarantee format compliance.

Metadata and Additional Processing

  • The authors inject standardized entity placeholders like @GENEand@DISEASEand @DISEASEand@DISEASE into the GAD inputs to streamline relation extraction.
  • No cropping strategies or complex metadata construction pipelines are detailed in the provided text.

Method

The authors leverage a few-shot learning framework during inference, where a small number of task examples are provided as conditioning without updating the model's weights. These examples typically consist of an instruction, context, and desired completion, such as a premise, hypothesis, and corresponding label for natural language inference (NLI) tasks. The few-shot technique involves presenting the model with kkk examples of context and completion, followed by a final example of context for which the model must generate the completion. The value of kkk generally ranges from 3 to 100, constrained by the model’s context window size—Flan-UL2, for instance, supports up to 2,048 tokens.

To enhance few-shot performance beyond randomly selected examples, a retrieval-based module is introduced using Sentence-Transformers (Reimers and Gurevych, 2019). This module retrieves the kkk most semantically similar examples from the training set. The process begins by embedding each instruction prompt in the training set into a vector space using a fixed PubMedBERT (Gu et al., 2021) model. For a given test instance, the query is compared to all training examples via cosine distance, and the top kkk closest examples are selected. In the implementation, kkk is set to 5.

The input instruction prompt is constructed by concatenating three components: an instruction that specifies the task, describes the data, and outlines the expected model behavior; the input argument containing relevant information; and constraints on the output space to guide generation. This structured format improves model performance across diverse tasks.

For named entity recognition (NER), two inference methods are evaluated. The first, adapted from Ye et al. (2023), is applicable only to ChatGPT and uses a format where words are separated by double vertical bars and labels by single vertical bars. The second method, introduced as Recursive Chain-of-Thought (RCoT), is more general and works across all tested generative models. RCoT extends the Chain-of-Thought (CoT) framework (Wei et al., 2022b) and builds upon Wang et al. (2022b). It operates by iteratively processing each token in the sequence, using the current prediction state as input to generate the label for the next token. This ensures that every token receives a label and prevents omissions during generation. However, the method incurs a high computational cost due to its ON\mathcal{O}^{N}ON complexity, where NNN is the number of tokens in the sequence, in contrast to the O1\mathcal{O}^{1}O1 complexity of the ChatGPT-specific method.

An example of the RCoT prompt format includes a detailed instruction, constraints, and a set of five few-shot examples, followed by the current sentence and a query about the label of a specific token. This approach ensures precise entity labeling while maintaining consistency with the model’s reasoning process.

Experiment

The experimental setup evaluates four instruction-tuned large language models against a biomedical baseline across thirteen clinical and natural language processing tasks using zero- and few-shot prompting protocols. This evaluation validates the generalization capacity of generic models in specialized medical domains, revealing that while few-shot learning effectively mitigates hallucinations and boosts overall accuracy, the models still lag behind task-specific architectures in classification and relation extraction. Consequently, the findings conclude that question answering is the most reliable application for current LLMs, and since no single model excels across all tasks, practitioners must carefully match model capabilities to specific clinical requirements.

The authors evaluate multiple large language models on a range of clinical and biomedical tasks, including classification, question answering, relation extraction, and named entity recognition. Results indicate that while generative models perform well on question-answering tasks, they fall short on classification and relation extraction compared to domain-specific models, with performance varying significantly across tasks and models. Generative models perform better than domain-specific models on question-answering tasks in zero-shot settings. Classification and relation extraction tasks show inferior performance for generative models compared to specialized models. Performance varies across models, with some excelling in certain tasks while underperforming in others.

The authors evaluate multiple large language models on a range of clinical and biomedical tasks, comparing their performance in zero-shot and few-shot settings against a biomedical-specific model. Results show that generative models perform well on question-answering tasks, particularly in few-shot scenarios, while their performance on classification and relation extraction tasks remains limited compared to specialized models. Generative models achieve strong performance on question-answering tasks, especially in few-shot settings, outperforming specialized models in some cases. Classification and relation extraction tasks show consistently lower performance for generative models compared to the biomedical-specific baseline. Alpaca demonstrates significant improvement in few-shot scenarios across all tasks, indicating high adaptability to new instructions.

The study evaluates multiple large language models across clinical and biomedical tasks, including classification, question answering, and relation extraction, using both zero-shot and few-shot settings against a domain-specific baseline. These experiments validate the comparative effectiveness of generative versus specialized architectures across diverse task types and prompting conditions. Qualitatively, generative models demonstrate strong capabilities in question answering, particularly when leveraging few-shot examples, but consistently lag behind specialized models on classification and relation extraction tasks. Overall, performance proves highly task-dependent, with certain architectures like Alpaca exhibiting notable adaptability and improvement in few-shot scenarios.


AI로 AI 구축

아이디어에서 출시까지 — 무료 AI 코코딩, 즉시 사용 가능한 환경, 최적의 GPU 가격으로 AI 개발을 가속화하세요.

AI 협업 코딩
바로 사용 가능한 GPU
최적의 가격

HyperAI Newsletters

최신 정보 구독하기
한국 시간 매주 월요일 오전 9시 에 이번 주의 최신 업데이트를 메일로 발송합니다
이메일 서비스 제공: MailChimp