Advancing Korean Medical Large Language Models: Automated Pipeline for Korean Medical Preference Dataset Construction
2
조회
0
다운로드
0
인용
0
참고문헌
키워드
초록
Objectives: Developing large language models (LLMs) in biomedicine requires access to high-quality training and alignmenttuning datasets. However, publicly available Korean medical preference datasets are scarce, hindering the advancement ofKorean medical LLMs. This study constructs and evaluates the efficacy of the Korean Medical Preference Dataset (KoMeP),an alignment tuning dataset constructed with an automated pipeline, minimizing the high costs of human annotation. Methods:KoMeP was generated using the DAHL score, an automated hallucination evaluation metric. Five LLMs (Dolly-v2-3B,MPT-7B, GPT-4o, Qwen-2-7B, Llama-3-8B) produced responses to 8,573 biomedical examination questions, from which5,551 preference pairs were extracted. Each pair consisted of a “chosen” response and a “rejected” response, as determined bytheir DAHL scores. The dataset was evaluated when trained through two different alignment tuning methods, direct preferenceoptimization (DPO) and odds ratio preference optimization (ORPO) respectively across five different models. TheKorMedMCQA benchmark was employed to assess the effectiveness of alignment tuning. Results: Models trained with DPOconsistently improved KorMedMCQA performance; notably, Llama-3.1-8B showed a 43.96% increase. In contrast, ORPOtraining produced inconsistent results. Additionally, English-to-Korean transfer learning proved effective, particularly forEnglish-centric models like Gemma-2, whereas Korean-to-English transfer learning achieved limited success. Instructiontuning with KoMeP yielded mixed outcomes, which suggests challenges in dataset formatting. Conclusions: KoMeP is thefirst publicly available Korean medical preference dataset and significantly improves alignment tuning performance in LLMs. The DPO method outperforms ORPO in alignment tuning. Future work should focus on expanding KoMeP, developing aKorean-native dataset, and refining alignment tuning methods to produce safer and more reliable Korean medical LLMs.
Abstract
Objectives: Developing large language models (LLMs) in biomedicine requires access to high-quality training and alignmenttuning datasets. However, publicly available Korean medical preference datasets are scarce, hindering the advancement ofKorean medical LLMs. This study constructs and evaluates the efficacy of the Korean Medical Preference Dataset (KoMeP),an alignment tuning dataset constructed with an automated pipeline, minimizing the high costs of human annotation. Methods:KoMeP was generated using the DAHL score, an automated hallucination evaluation metric. Five LLMs (Dolly-v2-3B,MPT-7B, GPT-4o, Qwen-2-7B, Llama-3-8B) produced responses to 8,573 biomedical examination questions, from which5,551 preference pairs were extracted. Each pair consisted of a “chosen” response and a “rejected” response, as determined bytheir DAHL scores. The dataset was evaluated when trained through two different alignment tuning methods, direct preferenceoptimization (DPO) and odds ratio preference optimization (ORPO) respectively across five different models. TheKorMedMCQA benchmark was employed to assess the effectiveness of alignment tuning. Results: Models trained with DPOconsistently improved KorMedMCQA performance; notably, Llama-3.1-8B showed a 43.96% increase. In contrast, ORPOtraining produced inconsistent results. Additionally, English-to-Korean transfer learning proved effective, particularly forEnglish-centric models like Gemma-2, whereas Korean-to-English transfer learning achieved limited success. Instructiontuning with KoMeP yielded mixed outcomes, which suggests challenges in dataset formatting. Conclusions: KoMeP is thefirst publicly available Korean medical preference dataset and significantly improves alignment tuning performance in LLMs. The DPO method outperforms ORPO in alignment tuning. Future work should focus on expanding KoMeP, developing aKorean-native dataset, and refining alignment tuning methods to produce safer and more reliable Korean medical LLMs.
