Codes and Datasets of paper "Investigating and Mitigating Catastrophic Forgetting in Medical Knowledge Injection through Internal Knowledge Augmentation Learning"

October 15, 2025 ยท View on GitHub

Requirement

vllm==0.8.5
torch==2.6.0
transformers==4.51.3
pandas==2.2.3

Run

1. Generate datasets for knowledge injection

bash generate_datasets.sh

2. Training

bash run_RefInject.sh
bash run_InternAL.sh

3. Evaluation

python analysis.py

Contents

  1. 1Requirement
  2. 2Run
  3. 2.11. Generate datasets for knowledge injection
  4. 2.22. Training
  5. 2.33. Evaluation