microsoft / microsoft/LLMLingua
[Question]: Reproducing the score of official microsoft/llmlingua-2-xlm-roberta-large-meetingbank
@pzs19 is already working on this.
Since May 24, 2024.
- Dominant language
- Python
- Stars
- 6.7k
- Forks
- 428
- Avg merge
- 2d 4h
- Merged PRs (30d)
- 1
Description
Describe the issue
Following the issue 155, I'm trying to reproduce the results of the official llmlingua-2-xlm-roberta-large-meetingbank model using Mistral-7B as black-box llm.
In specific, I tried to fine-tune the XLM-RoBERTa model with the officially provided dataset, using this train.sh.
Here is my detailed process:
- Format, label, and filter the official dataset with reference to the collect_data.sh.
- Fine-tune the XLM-RoBERTa model using train.sh, with hyperparameters from the LLMLingua-2 paper.
Here are the current issues:
- It's hard to reproduce the Table-4 results of LLMLingua-2 paper, or even scores in issue 155. Here are my reproduced results:
| MeetingBank | MeetingBank | LongBench | |||||
|---|---|---|---|---|---|---|---|
| QA | summary | 2000 token avg. | 2000 token narrativeqa | multifieldqa_en | multifieldqa_zh | qasper | |
| LLMLingua-2 scores reproduced with official model weights | 73.59 | 29.95 | 25.65 | 10.07 | 36.61 | 26.47 | 29.46 |
| LLMLingua-2 reproduced with fine-tuning | 68.95 | 30.05 | 24.67 | 9.14 | 33.91 | 26.49 | 29.12 |
- I found that the official llmlingua-2-xlm-roberta-large-meetingbank model weight has the word_embedding size of [250102, 1024]. This is larger than the original [250002, 1024] size of XLM-RoBERTa.
I guess this is relevant to the added special tokens in prompt_compressor.py, but train_roberta.py example does nothing about this, so my fine-tuned model has the same word_embedding weight size with the original RoBERTa ([250002, 1024]).
- I tried to resize token embedding size first then fine-tune, but the results were almost the same.
- I guess the example in train.sh doesn't use filtered results, which is named annotation_kept_cs512_meetingbank_train_formated.pt in collect_data.sh. This seems like a minor issue :)
If the process of training the official model is the same as the process provided as an example here, can you please let me know what needs to be changed in the above process?
Thank you for reading.
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.