Cited by 99Machine Learning in HealthcareMedical Coding and Health InformationTopic Modeling

DRG-LLaMA : tuning LLaMA model to predict diagnosis-related group for hospitalized patients

Open access
Wang, Hanyin, Gao, Chufan, Dantona, Christopher, Hull, Bryan, Sun, Jimeng·2024·npj Digital Medicine

Published version of a preprint

Abstract

In the U.S. inpatient payment system, the Diagnosis-Related Group (DRG) is pivotal, but its assignment process is inefficient. The study introduces , an advanced large language model (LLM) fine-tuned on clinical notes to enhance DRGs assignment. Utilizing LLaMA as the foundational model and optimizing it through Low-Rank Adaptation (LoRA) on 236,192 MIMIC-IV discharge summaries, our -7B model exhibited a noteworthy macro-averaged F1 score of 0.327, a top-1 prediction accuracy of 52.0%, and a macro-averaged Area Under the Curve (AUC) of 0.986, with a maximum input token length of 512. This model surpassed the performance of prior leading models in DRG prediction, showing a relative improvement of 40.3% and 35.7% in macro-averaged F1 score compared to ClinicalBERT and CAML, respectively. Applied to base DRG and complication or comorbidity (CC)/major complication or comorbidity (MCC) prediction, achieved a top-1 prediction accuracy of 67.8% and 67.5%, respectively. Additionally, our findings indicate that ’s performance correlates with increased model parameters and input context lengths.

AI snapshot

Written from the paper’s full text.

TL;DR

DRG-LLaMA fine-tunes LLaMA with LoRA on MIMIC-IV discharge summaries to predict diagnosis-related groups (DRGs) from clinical text. It outperforms ClinicalBERT and CAML on DRG prediction and shows improved performance with larger model sizes and longer input contexts.

Key findings

  • With a 7B LLaMA model and max input token size 512, DRG-LLaMA achieved macro-F1 0.327, top-1 accuracy 52.0%, top-5 accuracy 84.8%, and macro-AUC 0.986, with relative macro-F1 improvements of 40.3% vs. ClinicalBERT and 35.7% vs. CAML.
  • Performance improved with larger models and longer contexts: the 13B model with 1024 max input tokens reached top-1 accuracy 54.6%, top-5 accuracy 86.5%, and macro-F1 0.361.
  • In the two-label approach, base DRG and CC/MCC top-1 accuracies were 67.8% and 67.5%, respectively; combining them via mapping rules yielded 51.5% final DRG accuracy, comparable to the 52.0% single-label result.
  • Accuracy was higher for more frequent DRGs: top-1 accuracy was 55.7% for the top 300 DRGs and 69.4% for the top 30 DRGs, while performance declined for less frequent DRGs.

Methods

  • Design: fine-tuned LLaMA using Low-Rank Adaptation (LoRA) for DRG prediction from clinical notes, testing both single-label DRG classification and two-label prediction of base DRG plus CC/MCC status.
  • Data: MIMIC-IV discharge summaries, using the “brief hospital course” section; MS-DRG codes were consolidated to version 34.0, with 738 DRGs represented, a 90%/10% training/testing split stratified by DRG, and 26,244 test cases.
  • Analysis: compared against CAML and ClinicalBERT; reported macro/micro F1, macro/micro AUC, top-1/5/10 accuracy, with standard deviations from 30 bootstrap iterations.

Limitations

  • Stated: the study was limited to MIMIC-IV discharge summaries, which are available only after discharge, so early DRG prediction using admission or ED notes was not tested; computational limits restricted experiments to LLaMA models up to 13B parameters and prevented extensive hyperparameter search.
  • Evident: performance declined for rare DRGs, and the paper notes noisy CC/MCC assignment and cases where correct DRG prediction required information not available in discharge summaries.
  • Evident: manual error analysis was based on only 10 randomly selected erroneous cases, limiting generalizability of the error categories.

Model-written summary. Check it against the paper before relying on it.

Integrity signals

Registry

Crossref · 10.1038/s41746-023-00989-3

Cites

34 of 34 references verified0 citation errors · 0 not checkable

Statistics

Nothing to recompute.

Claims the paper makes

0 of 38 cited sources checked · 0 supported · 0 flagged

  • Findingown data

    DRG-LLaMA-7B achieved a top-5 prediction accuracy of 84.8% on DRG prediction with a maximum input token size of 512.

    “The accuracy of top-1 and top-5 predictions achieved by our fine-tuned DRG-LLaMA -7B model was 52.0% and 84.8%, respectively.”
    Quote in the paperSubstantial claim
  • Findingreasoning

    Out of the DRGs with a top-1 prediction accuracy of 100%, 8 out of 9 are surgical DRGs.

    “out of the DRGs with a top-1 prediction accuracy of 100%, 8 out of 9 are surgical DRGs”
    Quote in the paperNot a substantial claim
  • Findingown data

    A correlation exists between the number of training cases and prediction performance.

    “As noted above, a correlation exists between the number of training cases and prediction performance.”
    Quote in the paperNot a substantial claim
  • Findingown data

    DRG-LLaMA's performance consistently improved with larger models and longer input contexts.

    “observing a consistent improvement in all evaluation metrics with larger models and longer input contexts”
    • Also stated in: § DRG prediction as a single-label classification task · plain
      “We observed that DRG-LLaMA ’s performance consistently improved with larger models and longer input contexts.”
    Quote in the paperSubstantial claim
  • Findingown data

    The optimal configuration of DRG-LLaMA, using a 13B LLaMA model and a maximum input token size of 1024, achieved a top-1 prediction accuracy of 54.6%.

    “The optimal configuration, utilizing a 13B LLaMA model and a maximum input token size of 1024, achieved a top-1 prediction accuracy of 54.6%”
    Quote in the paperSubstantial claim
  • May combine two claims. 22
  • The quote only partly states this claim. 5
  • Classified as probably not a substantial claim. 27
  • May combine two claims.

    Findingown data

    ClinicalBERT achieved higher AUC and top-1 prediction accuracy but lower macro-averaged F1 score than CAML.

    “When compared to CAML, ClinicalBERT achieved higher AUC and top-1 prediction accuracy but lower macro-averaged F1 score.”
    Quote in the paperNot a substantial claim
  • The quote only partly states this claim.

    Findingown data

    DRG-LLaMA-7B achieved a macro-averaged F1 score of 0.327 on DRG prediction with a maximum input token size of 512.

    “our DRG-LLaMA -7B model exhibited a noteworthy macro-averaged F1 score of 0.327”
    Quote in the paperSubstantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Findingown data

    The distribution of cases per DRG is imbalanced, with a median number of 124.5 in the training set.

    “The distribution of cases per DRG is imbalanced, with a median number of 124.5 in the training set”
    Quote in the paperNot a substantial claim
  • May combine two claims. The quote only partly states this claim.

    Findingown data

    DRG-LLaMA-7B achieved a macro-averaged AUC of 0.986 on DRG prediction with a maximum input token size of 512.

    “a macro-averaged Area Under the Curve (AUC) of 0.986”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Findingown data

    In manual error analysis of 10 samples, errors were categorized as erroneous CC/MCC (1/10), correct information needed for DRG prediction unavailable (1/10), difficulty in selecting correct base DRG (3/10), inadequate clinical concept extraction (4/10), and an isolated case of a plausible incorrect DRG label (1/10).

    “Broadly, the identified errors were categorized as follows: erroneous CC/MCC (1/10), correct information needed for DRG prediction unavailable (1/10), difficulty in selecting correct base DRG (3/10), inadequate clinical concept extraction (4/10) and an isolated case of a plausible incorrect DRG label (1/10).”
    Quote in the paperNot a substantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing). May combine two claims.

    Findingown data

    The training and testing set contains 738 and 723 unique DRG labels, respectively.

    “The training and testing set contains 738 and 723 unique DRG labels, respectively.”
    Quote in the paperNot a substantial claim
  • The quote only partly states this claim.

    Findingown data

    The two-label approach's accuracy of 51.5% was comparable with the single-label approach's accuracy of 52.0% using the same base model.

    “Notably, this performance was comparable with the accuracy attained in the single-label approach of 52.0% using the same base model”
    Quote in the paperSubstantial claim
  • May combine two claims. The quote only partly states this claim.

    Findingown data

    DRG-LLaMA-7B achieved a top-1 prediction accuracy of 52.0% on DRG prediction with a maximum input token size of 512.

    “a top-1 prediction accuracy of 52.0%”
    • Also stated in: § DRG prediction as a single-label classification task · plain
      “The accuracy of top-1 and top-5 predictions achieved by our fine-tuned DRG-LLaMA -7B model was 52.0% and 84.8%, respectively.”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Findingreasoning

    The top-1 accuracy for CC/MCC is 67.5%, similar to 67.8% for base DRG, despite the considerably smaller label count (5 labels in CC/MCC vs. 340 labels in base DRG).

    “Surprisingly, the top-1 accuracy for CC/MCC stands at 67.5%, similar to 67.8% of the base DRG despite the considerably smaller label count (5 labels in CC/MCC vs. 340 labels in base DRG).”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Findingown data

    MS-DRG version 34.0 comprises a total of 757 DRG codes, with 738 being represented in our dataset.

    “This version comprises a total of 757 DRG codes, with 738 being represented in our dataset.”
    Quote in the paperNot a substantial claim
  • May combine two claims.

    Findingreasoning

    In Case 2 in Table 4, despite the discharge summary providing a more comprehensive discussion on gastrointestinal bleeding compared to acute renal failure, acute renal failure was deemed the correct base DRG.

    “For instance, in Case 2 in Table 4, despite the discharge summary providing a more comprehensive discussion on gastrointestinal bleeding compared to acute renal failure, the latter was deemed the correct base DRG.”
    Quote in the paperNot a substantial claim
  • May combine two claims.

    Findingnone given

    DRG-LLaMA performance improved with the utilization of larger models and longer input context length.

    “We also observed that the performance of DRG-LLaMA enhanced with the utilization of larger models and longer input context length.”
    Quote in the paperNot a substantial claim
  • May combine two claims. The quote only partly states this claim.

    Findingreasoning

    Rules formulated in the DRG dissection phase and the inference phase cater to various split scenarios and improve accuracy.

    “These rules cater to various split scenarios, thus improving accuracy.”
    Quote in the paperSubstantial claim
  • Classified as probably not a substantial claim. May combine two claims.

    Study design and conductnone given

    90% of the data was allocated as training set while the rest 10% as testing set, and this partitioning was stratified on DRGs.

    “90% of the data was allocated as training set while the rest 10% as testing set, and this partitioning was stratified on DRGs.”
    Quote in the paperSubstantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Premisecitation

    Previous studies have showcased advancements in DRGs classification accuracy through various machine-learning algorithms and deep neural networks.

    “Previous studies have showcased advancements in DRGs classification accuracy through various machine-learning algorithms10 and deep neural networks11.”
    Quote in the paperNot a substantial claim0 of 2 sources checked
    • cites [10] Gartner, D., Kolisch, R., Neill, D. B. & Padman, R. Machine learning approaches for early DRG classification and resource allocation. INFORMS J. Comput. 27, 718–734 (2015). source

      Not checked in the free run.

    • cites [11] Islam, M.M., Li, G.H., Poly, T.N., Li, Y.C. Deepdrg: Performance of artificial intelligence model for real-time prediction of diagnosis-related groups. in Healthcare;9:1632MDPI (2021). source

      Not checked in the free run.

  • Classified as probably not a substantial claim.

    Premisecitation

    Each distinct DRG code is delineated by a particular set of patient attributes, including principal diagnosis, specific secondary diagnoses, procedures, sex and discharge status.

    “Each distinct DRG code is delineated by a particular set of patient attributes, including principal diagnosis, specific secondary diagnoses, procedures, sex and discharge status8.”
    Quote in the paperSubstantial claim0 of 1 sources checked
    • cites [8] CMS. ICD-10-CM/PCS MS-DRG v34. 0 Definitions Manual. https://www.cms.gov/icd10m/version34-fullcode-cms/fullcode_cms/P0001.html (2016).

      Not checked in the free run.

  • May combine two claims.

    Interpretationcitation

    High AUC scores were obtained for all models due to the many infrequent DRG classes, resulting in high true negative predictions for all negative class predictions.

    “High AUC scores were obtained for all models due to the many infrequent DRG classes, resulting in high true negative predictions for all negative class predictions13.”
    Quote in the paperNot a substantial claim0 of 1 sources checked
    • cites [13] Liu, J., Capurro, D., Nguyen, A. & Verspoor, K. Early prediction of diagnostic-related groups and estimation of hospital cost by processing clinical notes. NPJ Digi.l Med. 4, 103 (2021). source

      Not checked in the free run.

  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Premisecitation

    A deep learning-based NLP model leveraging CAML has been applied to predict DRGs based on clinical notes and yielded promising outcomes.

    “More recently, a deep learning-based NLP model leveraging adjusted Convolutional Attention for Multi-Label Classification (CAML) has been applied to predict DRGs based on clinical notes and yielded promising outcomes12,13.”
    Quote in the paperNot a substantial claim0 of 2 sources checked
    • cites [12] Mullenbach, J., Wiegreffe, S., Duke, J., Sun, J., Eisenstein, J. Explainable prediction of medical codes from clinical text. arXiv preprint arXiv:1802.05695 (2018). source

      Not checked in the free run.

    • cites [13] Liu, J., Capurro, D., Nguyen, A. & Verspoor, K. Early prediction of diagnostic-related groups and estimation of hospital cost by processing clinical notes. NPJ Digi.l Med. 4, 103 (2021). source

      Not checked in the free run.

  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Interpretationown data

    Certain errors, like inadequate clinical concept extraction, indicate the model's weaknesses.

    “Certain errors, like inadequate clinical concept extraction, indicate the model’s weaknesses.”
    Quote in the paperNot a substantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Interpretationreasoning

    Other errors, such as the difficulty in selecting the base DRG, likely stem from the intricacies of the DRG assignment rules.

    “Other errors, such as the difficulty in selecting the base DRG, likely stem from the intricacies of the DRG assignment rules.”
    Quote in the paperNot a substantial claim
  • May combine two claims.

    Premisecitation

    DRGs involve multi-class classification, where one DRG code is assigned to each visit, in contrast to the multi-label classification of ICDs, where multiple codes may apply to a single visit.

    “DRGs involve multi-class classification, where one DRG code is assigned to each visit, in contrast to the multi-label classification of ICDs, where multiple codes may apply to a single visit9.”
    Quote in the paperSubstantial claim0 of 1 sources checked
    • cites [9] Kaur, R., Ginige, J.A., Obst, O. AI-based ICD coding and classification approaches using discharge summaries: A systematic literature review. Expert Syst. Appl. 118997 (2022). source

      Not checked in the free run.

  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Study design and conductown data

    There is no significant difference in the average word counts in the training vs. testing set (398 vs. 399; p = 0.51 from two-sided t-test).

    “There is no significant difference in the average word counts in the training vs. testing set (398 vs. 399; p = 0.51 from two-sided t-test).”
    Quote in the paperNot a substantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Premisecitation

    Since its inception by Medicare in 1983, DRG has served as the foundation for the inpatient prospective payment system within the United States.

    “Since its inception by Medicare in 1983, DRG has served as the foundation for the inpatient prospective payment system within the United States7.”
    Quote in the paperNot a substantial claim0 of 1 sources checked
    • cites [7] Quinn, K. After the revolution: DRGs at age 30. Ann. Internal Med. 160, 426–429 (2014). source

      Not checked in the free run.

  • May combine two claims.

    Premisecitation

    In the 34.0 version of the MS-DRG system, there are 154 three-way split DRGs, 44 two-way split DRGs with MCC/CC and no CC, 65 two-way split DRGs with MCC and CC/no CC, and 77 base DRGs with no splits.

    “In the 34.0 version of the MS-DRG system, there are 154 three-way split DRGs, 44 two-way split DRGs with MCC/CC and no CC, 65 two-way split DRGs with MCC and CC/no CC, and 77 base DRGs with no splits (examples in Supplementary Note 1)8.”
    Quote in the paperSubstantial claim0 of 1 sources checked
    • cites [8] CMS. ICD-10-CM/PCS MS-DRG v34. 0 Definitions Manual. https://www.cms.gov/icd10m/version34-fullcode-cms/fullcode_cms/P0001.html (2016).

      Not checked in the free run.

  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Study design and conductown data

    The study was restricted by computational resource limitations, so experiments with the LLaMA model were only up to a parameter size of 13 billion.

    “We were also restricted by computational resource limitations, so we could only experiment with the LLaMA model up to a parameter size of 13 billion.”
    Quote in the paperNot a substantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Premisecitation

    Several groups have directed efforts toward fine-tuning LLaMA in the medical context, including ChatDoctor, HuaTuo, and PMC-LLaMA.

    “Within the medical context, several groups have directed their efforts toward fine-tuning LLaMA. Notable examples among these are ChatDoctor (trained on authentic patient-physician dialogues), HuaTuo (fine-tuned with a Chinese medical knowledge graph), and PMC-LLaMA (fine-tuned on biomedical academic papers)22–24.”
    Quote in the paperNot a substantial claim0 of 3 sources checked
    • cites [22] Wang, H. et al. Huatuo: Tuning llama model with chinese medical knowledge. arXiv preprint arXiv:2304.06975 (2023).

      Not checked in the free run.

    • cites [23] Yunxiang, L., Zihan, L., Kai, Z., Ruilong, D., You, Z. Chatdoctor: A medical chat model fine-tuned on llama model using medical domain knowledge. arXiv preprint arXiv:2303.14070 (2023).

      Not checked in the free run.

    • cites [24] Wu, C., Zhang, X., Zhang, Y., Wang, Y., Xie, W. Pmc-llama: Further finetuning llama on medical papers. arXiv preprint arXiv:2304.14454 (2023).

      Not checked in the free run.

  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Premisecitation

    CAML model, exclusively utilizing clinical notes, surpassed the performance of a Long Short-Term Memory (LSTM) model using structured clinical variables.

    “CAML model, exclusively utilizing clinical notes, surpassed the performance of a Long Short-Term Memory (LSTM) model using structured clinical variables13.”
    Quote in the paperNot a substantial claim0 of 1 sources checked
    • cites [13] Liu, J., Capurro, D., Nguyen, A. & Verspoor, K. Early prediction of diagnostic-related groups and estimation of hospital cost by processing clinical notes. NPJ Digi.l Med. 4, 103 (2021). source

      Not checked in the free run.

  • Probably not a substantial claim: may not pass S3 (load-bearing). May combine two claims.

    Premisecitation

    These LLaMA-based models focused on medical question answering, yielding encouraging outcomes.

    “These LLaMA-based models focused on medical question answering, yielding encouraging outcomes.”
    Quote in the paperNot a substantial claim0 of 3 sources checked
    • cites [22] Wang, H. et al. Huatuo: Tuning llama model with chinese medical knowledge. arXiv preprint arXiv:2304.06975 (2023).

      Not checked in the free run.

    • cites [23] Yunxiang, L., Zihan, L., Kai, Z., Ruilong, D., You, Z. Chatdoctor: A medical chat model fine-tuned on llama model using medical domain knowledge. arXiv preprint arXiv:2303.14070 (2023).

      Not checked in the free run.

    • cites [24] Wu, C., Zhang, X., Zhang, Y., Wang, Y., Xie, W. Pmc-llama: Further finetuning llama on medical papers. arXiv preprint arXiv:2304.14454 (2023).

      Not checked in the free run.

  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Study design and conductreasoning

    An extensive hyperparameter search was not performed.

    “Unfortunately, we couldn’t perform an extensive hyperparameter search.”
    Quote in the paperNot a substantial claim
  • May combine two claims.

    Interpretationreasoning

    Assigning a label of 'not applicable' when a DRG code does not explicitly describe CC/MCC status might bring noisy signals for models to learn, but was found better than assigning to an alternative label such as 'without CC/MCC'.

    “We realize such classification might bring in noisy signals for models to learn (as a patient with “transient ischemia” can indeed has CC/MCC), but found it better than assigning to an alternative label such as “without CC/MCC” which would be more erroneous.”
    Quote in the paperSubstantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Premisecitation

    ClinicalBERT was built upon BioBERT, a domain-specific BERT model pre-trained on PubMed abstracts and full-text articles from PubMed Central.

    “ClinicalBERT was built upon BioBERT, a domain-specific BERT model pre-trained on PubMed abstracts and full-text articles from PubMed Central16.”
    Quote in the paperNot a substantial claim0 of 1 sources checked
    • cites [16] Lee, J. et al. BioBERT: a pre-trained biomedical language representation model for biomedical text mining. Bioinformatics 36, 1234–1240 (2020). source

      Not checked in the free run.

  • Probably not a substantial claim: may not pass S3 (load-bearing). May combine two claims.

    Premisecitation

    When compared with ClinicalBERT, CAML provided improved F1 scores but lower AUC.

    “When compared with ClinicalBERT, CAML provided improved F1 scores but lower AUC13,15.”
    Quote in the paperNot a substantial claim0 of 2 sources checked
    • cites [13] Liu, J., Capurro, D., Nguyen, A. & Verspoor, K. Early prediction of diagnostic-related groups and estimation of hospital cost by processing clinical notes. NPJ Digi.l Med. 4, 103 (2021). source

      Not checked in the free run.

    • cites [15] Alsentzer, E. et al. Publicly available clinical BERT embeddings. arXiv preprint arXiv:1904.03323 (2019). source

      Not checked in the free run.

  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Premisecitation

    CAML model exhibited superior ability to predict DRGs.

    “More recently, CAML model exhibited superior ability to predict DRGs13.”
    Quote in the paperNot a substantial claim0 of 1 sources checked
    • cites [13] Liu, J., Capurro, D., Nguyen, A. & Verspoor, K. Early prediction of diagnostic-related groups and estimation of hospital cost by processing clinical notes. NPJ Digi.l Med. 4, 103 (2021). source

      Not checked in the free run.

  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Premisenone given

    The MIMIC-IV training dataset is substantially larger than the MIMIC-III dataset (236,192 vs. 17,815).

    “Beyond the substantially larger training dataset employed in MIMIC-IV compared to MIMIC-III (236,192 vs. 17,815)”
    Quote in the paperNot a substantial claim
  • May combine two claims.

    Interpretationreasoning

    A hospital stay with the DRG code 'pulmonary edema and respiratory failure' may truly contain MCC, but the MCC would not be labeled as positive in the training set.

    “Therefore, a hospital stay with this DRG code may truly contain MCC, but the MCC would not be labeled as positive in the training set.”
    Quote in the paperSubstantial claim
    • Rests on an assumption: The DRG code 'pulmonary edema and respiratory failure' does not have a CC/MCC split.
  • May combine two claims.

    Premisecitation

    Without domain-specific fine-tuning or specialized prompt crafting, GPT-4 exceeded the passing score on USMLE by over 20 points and set a new state-of-the-art.

    “Without domain-specific fine-tuning or specialized prompt crafting, GPT-4 exceeded the passing score on USMLE by over 20 points and set a new state-of-the-art3.”
    Quote in the paperSubstantial claim0 of 1 sources checked
    • cites [3] Nori, H., King, N., McKinney, S.M., Carignan, D., Horvitz, E. Capabilities of gpt-4 on medical challenge problems. arXiv preprint arXiv:2303.13375 (2023).

      Not checked in the free run.

  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Premisecitation

    Instruction-following models fine-tuned from LLaMA such as Alpaca and Vicuna exhibit performance on par with GPT-3.5.

    “Instruction-following models fine-tuned from LLaMA such as Alpaca20 and Vicuna21, exhibit performance on par with GPT-3.5.”
    Quote in the paperNot a substantial claim0 of 2 sources checked
    • cites [20] Taori, R. et al. Stanford Alpaca: An Instruction-following LLaMA model. https://github.com/tatsu-lab/stanford_alpaca (2023).

      Not checked in the free run.

    • cites [21] Chiang, W.L. et al. Vicuna: An Open-Source Chatbot Impressing GPT-4 with 90% ChatGPT Quality. https://lmsys.org/blog/2023-03-30-vicuna (2023).

      Not checked in the free run.

  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Premisecitation

    ClinicalBERT performed further pre-training of BioBERT using 2 million clinical notes from MIMIC-III.

    “ClinicalBERT performed further pre-training of BioBERT using 2 million clinical notes from MIMIC-III34.”
    Quote in the paperNot a substantial claim0 of 1 sources checked
    • cites [34] Johnson, A. E. et al. MIMIC-III, a freely accessible critical care database. Sci. Data 3, 1–9 (2016). source

      Not checked in the free run.

  • May combine two claims.

    Study design and conductown data

    The study was limited by the constraints of the MIMIC-IV dataset and could only use discharge summaries as input data, which are only available after the patient is discharged from the hospital.

    “We were limited by the constraints of the MIMIC-IV dataset and could only use discharge summaries as input data, which are only available after the patient is discharged from the hospital.”
    Quote in the paperSubstantial claim
  • Classified as probably not a substantial claim. May combine two claims.

    Study design and conductnone given

    An internal medicine physician manually reviewed all DRG codes from step 2 and assigned them to the most appropriate MS-DRG v.34.0 codes; a domain expert specializing in inpatient Clinical Documentation Integrity independently verified the accuracy of the code assignments.

    “An internal medicine physician manually reviewed all DRG codes from step 2, and assigned these codes to the most appropriate MS-DRG v.34.0 codes if applicable. Subsequently, a domain expert specializing in inpatient Clinical Documentation Integrity (CDI) assessed the conversion table and independently verified the accuracy of the code assignments.”
    Quote in the paperSubstantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Premisecitation

    Previous studies have underscored the effectiveness of employing diverse machine learning algorithms and deep neural networks for DRG prediction within healthcare systems outside the United States.

    “Previous studies have underscored the effectiveness of employing diverse machine learning algorithms and deep neural networks for DRG prediction within healthcare systems outside the United States10,11.”
    Quote in the paperNot a substantial claim0 of 2 sources checked
    • cites [10] Gartner, D., Kolisch, R., Neill, D. B. & Padman, R. Machine learning approaches for early DRG classification and resource allocation. INFORMS J. Comput. 27, 718–734 (2015). source

      Not checked in the free run.

    • cites [11] Islam, M.M., Li, G.H., Poly, T.N., Li, Y.C. Deepdrg: Performance of artificial intelligence model for real-time prediction of diagnosis-related groups. in Healthcare;9:1632MDPI (2021). source

      Not checked in the free run.

  • Probably not a substantial claim: may not pass S2 (checkable).

    Interpretationown data

    The results highlight the potential of adapting LLMs for medical purposes, particularly in predicting DRGs.

    “The results presented in this study highlight the potential of adapting LLMs for medical purposes, particularly in predicting DRGs.”
    Quote in the paperNot a substantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing). May combine two claims.

    Premisecitation

    BERT-based models have a maximum input length of 512 tokens, whereas CAML can handle longer context.

    “While BERT-based models have a maximum input length of 512 tokens, CAML has the flexibility to handle longer context13,19.”
    Quote in the paperNot a substantial claim0 of 2 sources checked
    • cites [13] Liu, J., Capurro, D., Nguyen, A. & Verspoor, K. Early prediction of diagnostic-related groups and estimation of hospital cost by processing clinical notes. NPJ Digi.l Med. 4, 103 (2021). source

      Not checked in the free run.

    • cites [19] Devlin, J., Chang, M.W., Lee, K., Toutanova, K. Bert: Pre-training of deep bidirectional transformers for language understanding. arXiv preprint arXiv:1810.04805 (2018).

      Not checked in the free run.

  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Premisecitation

    These studies focused on using structured data as input variables instead of clinical text.

    “These studies focused on using structured data as input variables instead of clinical text.”
    Quote in the paperNot a substantial claim0 of 2 sources checked
    • cites [10] Gartner, D., Kolisch, R., Neill, D. B. & Padman, R. Machine learning approaches for early DRG classification and resource allocation. INFORMS J. Comput. 27, 718–734 (2015). source

      Not checked in the free run.

    • cites [11] Islam, M.M., Li, G.H., Poly, T.N., Li, Y.C. Deepdrg: Performance of artificial intelligence model for real-time prediction of diagnosis-related groups. in Healthcare;9:1632MDPI (2021). source

      Not checked in the free run.

  • Classified as probably not a substantial claim.

    Premisecitation

    Language models based on the transformer architecture, either pretrained or fine-tuned using biomedical corpora, have demonstrated efficacy across a spectrum of NLP benchmarks within the biomedical realm.

    “Language models based on the transformer architecture, either pretrained or fine-tuned using biomedical corpora, have demonstrated efficacy across a spectrum of NLP benchmarks within the biomedical realm16–18.”
    Quote in the paperSubstantial claim0 of 3 sources checked
    • cites [16] Lee, J. et al. BioBERT: a pre-trained biomedical language representation model for biomedical text mining. Bioinformatics 36, 1234–1240 (2020). source

      Not checked in the free run.

    • cites [17] Huang, K., Altosaar, J., Ranganath, R. Clinicalbert: Modeling clinical notes and predicting hospital readmission. arXiv preprint arXiv:1904.05342 (2019).

      Not checked in the free run.

    • cites [18] Gu, Y. et al. Domain-specific language model pretraining for biomedical natural language processing. ACM Transact. Comput. for Healthcare (HEALTH) 3, 1–23 (2021).

      Not checked in the free run.

45 of 45
  • The paper relies on this without support.

    The DRG assignment process is inefficient.

    “but its assignment process is inefficient”
  • The paper relies on this without support.

    Traditionally, the assignment of DRGs constitutes a labor-intensive manual endeavor undertaken by coding specialists, typically subsequent to a patient’s discharge.

    “Traditionally, the assignment of DRGs constitutes a labor-intensive manual endeavor undertaken by coding specialists, typically subsequent to a patient’s discharge.”
  • The paper relies on this without support.

    A pressing interest exists in the accurate early prediction of DRGs during a patient’s hospitalization.

    “a pressing interest exists in the accurate early prediction of DRGs during a patient’s hospitalization”
  • The paper relies on this without support.

    Accurate early DRG prediction is vital for efficacious resource planning and allocation.

    “This prediction is vital for efficacious resource planning and allocation.”
  • The paper relies on this without support.

    In MIMIC-III, a large portion of records during the 48-hour window comprises nursing and radiology notes, potentially lacking pivotal admission History of Present Illness (HPI) notes.

    “In the MIMIC-III database, a large portion of records during this time window comprises nursing and radiology notes, potentially lacking the pivotal admission History of Present Illness (HPI) notes.”
  • The paper relies on this without support.

    The DRG code 'pulmonary edema and respiratory failure' does not have a CC/MCC split.

    “For instance, the DRG code “pulmonary edema and respiratory failure” does not have a CC/MCC split.”
6 of 6

Claims other papers make about this one

Each is checked against this paper’s own text: does it say what the citing paper says it does?

Self-check

Percentages, sums and reported differences re-computed from the paper’s own numbers.

Each contradiction between two passages listed here was confirmed by an independent check.

  • Longer input context does not improve every metric

    Minor

    For the 7B model, increasing the maximum input length from 512 to 1024 tokens leaves both macro-AUC (0.986) and micro-AUC (0.994) unchanged, contrary to the claim of improvement in all evaluation metrics.

1 of 1

Cites

Flagged first, then by citations

Research works printed but not linked: 2

  1. Brown, T. et al. Language models are few-shot learners. Adv. Neural Inform. Process. Syst. 33, 1877–1901 (2020).
  2. Radford, A. et al. Language models are unsupervised multitask learners. OpenAI blog 1, 9 (2019).

Cited by

Sorted by citations

Authors
Wang, H., Gao, C., Dantona, C., Hull, B., & Sun, J.
Affiliations
  • Mayo Clinic (US)
  • University of Illinois Urbana-Champaign (US)
  • Mayo Clinic in Arizona (US)
Corresponding author
Sun, Jimeng
Venue
npj Digital Medicine
Published
2024-01-22
Volume
7
Issue
1
Pages
16-16
ISSN
2398-6352 (ISSN-L)
OpenAlex
W4391098193
Cited by
99
References
23
Citations by year
2026: 22 · 2025: 52 · 2024: 24 · 2023: 1
FWCI
14.37
Type
journal article
Language
en
Open access
Locations
  • npj Digital Medicine (publishedVersion, cc-by)
  • Research Square (acceptedVersion, cc-by)
  • PubMed Central (publishedVersion, cc-by)
  • PubMed (publishedVersion)
  • arXiv (Cornell University) (submittedVersion, cc-by)
  • DOAJ (DOAJ: Directory of Open Access Journals) (submittedVersion, cc-by-sa)
  • arXiv (Cornell University) (cc-by)
Funders
National Science Foundation (IIS-2034479, SCH-2014438)
Topics
  • Physical Sciences › Computer Science › Artificial Intelligence › Machine Learning in Healthcare
  • Health Sciences › Health Professions › Health Information Management › Medical Coding and Health Information
  • Physical Sciences › Computer Science › Artificial Intelligence › Topic Modeling
Keywords
LLaMA fine-tuning, low-rank adaptation, clinical notes, discharge summaries, MIMIC-IV, model parameter size

About these checks

Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.

Claims read by deepseek-v4.1-flash under definition version 3 (extract-v3) on 2026-10-09.

Classified by jev-1.13-20260917 (gold-guideline-v0.2) on 2026-10-09.

Statistics and arithmetic are recomputed, not judged.

Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.

What counts as a claim ›How we measure confidence ›How we check this ›

Are you an author, or do you see a problem with this page? Request a review