Skip to content

Finetuning on InfographicVQA #125

Description

I was unable to achieve the result shown in the UDOP paper.

I used the udop-unimodel-large-224 checkpoint.

My ANLS score is 0.407903.
This is nowhere near 0.461 as shown in the table below taken from the paper.

image

Since I noticed that the batch size, warmup steps and weight decay given in https://github.com/microsoft/i-Code/blob/main/i-Code-Doc/scripts/finetune_duebenchmark.sh is different from reported in the paper, I also tried changing the finetuning script to use the paper's settings.

drawing

Lastly, I also tried adding the task prompt prefix since it is not done so in the existing code. I followed approach from #71 (comment)

Results of the 3 different finetuning configurations:

Task prefixHyperparameter settingsANLS Score
NoUnchanged finetuning script0.407903
NoPaper's settings0.40174
YesUnchanged finetuning script0.408355

Other changes I made:

Please assist

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions