The application of Large Language Models in the medical domain in Vietnam presents significant challenges due to the scarcity of domain-specific training data. Traditional Supervised Fine-Tuning methods are often ineffective on small datasets, leading to overfitting and a failure to capture the complex nuances of medical language. To address this, our study proposes Token Value-Aware Fine-Tuning, an advanced fine-tuning method, as an alternative to SFT on low-resource datasets. The TVAFT process uses the Gemini API to generate judgments, followed by computing prominent weights to guide the pre-trained model’s focus toward critical tokens during training. To evaluate TVAFT, we fine-tune the PhoGPT-4B-Chat model for a medical question-answering task using a dataset comprising 250 Vietnamese medical question–answer pairs. Experimental results demonstrate that TVAFT significantly improves evaluation metrics compared to SFT, including BLEU, ROUGE, BERTScore, and token-level F1. These findings establish TVAFT as an effective approach for tasks requiring deep understanding under limited data conditions, highlighting its potential for developing specialized Vietnamese health consultation systems.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

A Novel Fine-Tuning Method for Enhancing Large Language Model Performance on Low-Resource Vietnamese Medical Question-Answering Datasets

  • Thang Nguyen Xuan,
  • Bao Nguyen Gia,
  • Nhi Vu Hoai,
  • Van Tran Cam

摘要

The application of Large Language Models in the medical domain in Vietnam presents significant challenges due to the scarcity of domain-specific training data. Traditional Supervised Fine-Tuning methods are often ineffective on small datasets, leading to overfitting and a failure to capture the complex nuances of medical language. To address this, our study proposes Token Value-Aware Fine-Tuning, an advanced fine-tuning method, as an alternative to SFT on low-resource datasets. The TVAFT process uses the Gemini API to generate judgments, followed by computing prominent weights to guide the pre-trained model’s focus toward critical tokens during training. To evaluate TVAFT, we fine-tune the PhoGPT-4B-Chat model for a medical question-answering task using a dataset comprising 250 Vietnamese medical question–answer pairs. Experimental results demonstrate that TVAFT significantly improves evaluation metrics compared to SFT, including BLEU, ROUGE, BERTScore, and token-level F1. These findings establish TVAFT as an effective approach for tasks requiring deep understanding under limited data conditions, highlighting its potential for developing specialized Vietnamese health consultation systems.