Fig. 3: An illustration of both approaches we tested.
From: DRG-LLaMA : tuning LLaMA model to predict diagnosis-related group for hospitalized patients
Single Label Prediction–which directly predicts the DRG code from the text–as well as Two Label Prediction–which breaks down the classification task into 2 tasks. The two predictions are then combined using filtering rules (discovered from data for each DRG) at inference time for the final DRG prediction. LoRA training is used to train the LLM due to computational constraints.
