基于 Microsoft Phi-2 微调小模型生成合成临床数据并提升预测医疗分析
Fine-Tuned Small Language Model for Synthetic Clinical Data Generation and Predictive Healthcare Analytics
AI 导读
研究者将 Microsoft Phi-2 模型通过 MedQuad 数据集监督微调、QLoRA 方法及 few-shot 提示词适配医疗领域,用于生成合成临床数据以扩充远程监测生命体征小数据集。微调后模型在 ROUGE 和 BERTScore 评测上均优于预训练模型。生成的增强数据将 LSTM 预测模型的分类准确率从 89.0% 提升至 93.7%,精确率、召回率和 F1-score 也一致改善。
正文 · 原文
Abstract:The reduced availability of medical datasets represents an important challenge in the development of predictive models for remote healthcare solutions, especially when the available data are imbalanced or difficult to extend through traditional methods. Small Language Models (SLMs) have become a promising alternative to Large Language Models (LLMs) for addressing these limitations due to their lower computational requirements and adaptability to specialized tasks. However, their performance remains highly dependent on the quality and representativeness of the available training data. In this paper, we propose a method for clinical data augmentation based on a fine-tuned Microsoft Phi-2 model that is used to generate synthetic instances, with the purpose of improving deep learning-based model for predicting remotely monitored vital signs. The Phi-2 model was adapted to the medical domain through supervised fine-tuning on the MedQuad dataset, through a Quantized Low-Rank Adaptation (QLoRA)-based method and by applying few-shot prompting techniques. The resulting model was used afterwards in the augmentation of a small-sized dataset, composed of vital sign values and labels associated with pathological and physiological states. The experimental results demonstrate that fine-tuned SLMs improved the performance on medical domain, achieving higher scores for both lexical (ROUGE) and semantic (BERTScore) evaluation metrics compared to the pretrained model. Moreover, the fine-tuned SLM is effective for generating structured synthetic clinical data. When used for data augmentation, the generated dataset improved the performance of a LSTM predictive model, increasing the classification accuracy from 89.0% to 93.7% and showing consistent improvements in precision, recall, and F1-score.
| Subjects: | Artificial Intelligence (cs.AI) |
| Cite as: | arXiv:2611.16460 [cs.AI] |
| (or arXiv:2611.16460v1 [cs.AI] for this version) | |
| https://doi.org/10.48550/arXiv.2611.16460 arXiv-issued DOI via DataCite (pending registration) |
Submission history
From: Ionut Anghel [view email]
[v1]
Thu, 12 Nov 2026 08:12:17 UTC (988 KB)
来源:arXiv cs.AI · arxiv.org