LoRA-Fine-Tuned LLMs for Discharge Medication Recommendation on Chinese EHRs
摘要
This study proposes a robust framework for automated discharge medication recommendation by leveraging LoRA-based fine-tuning of large language models (LLMs) on Chinese electronic health records (EHRs), addressing the CHIP 2025 shared task. We introduce a multi-stage framework that integrates several key techniques: (1) parameter-efficient instruction fine-tuning of four base LLMs (Qwen3-8B, Qwen3-4B-Instruct-2507, Llama-3.1-8B-Instruct, and Mistral-7B-Instruct-v0.3) using Low-Rank Adaptation (LoRA); (2) stabilization of predictions via repeated sampling and ensemble voting to mitigate generative variability and model-specific biases; and (3) systematic output refinement through pseudo-labeling and rule-based post-processing for drug name normalization. Experiments on 3,602 training and 1,722 test samples demonstrate substantial improvements over baseline performance. The final ensemble model achieved a Jaccard score of 0.4870, an F1 score of 0.6014, and a final evaluation score of 0.5442, validating the effectiveness of the proposed framework in enhancing the accuracy and practical reliability of medication recommendation systems.