rubentito/t5-base-mpdocvqa

Model

T5 base fine-tuned on MP-DocVQA

1

16 commits

1 linked in READMEs

updated Nov 20, 2023

See the code

README

T5 base fine-tuned on MP-DocVQA

This is pretrained T5 base fine-tuned on Multipage DocVQA (MP-DocVQA) dataset.

This model was used as a baseline in Hierarchical multimodal transformers for Multi-Page DocVQA.

  • Results on the MP-DocVQA dataset are reported in Table 2.
  • Training hyperparameters can be found in Table 8 of Appendix D.

How to use

Here is how to use this model to get the features of a given text in PyTorch:

import torch
from transformers import T5Tokenizer, T5ForConditionalGeneration

tokenizer = LongformerTokenizerFast.from_pretrained("rubentito/t5-base-mpdocvqa")
model = LongformerForQuestionAnswering.from_pretrained("rubentito/t5-base-mpdocvqa")

context = "Huggingface has democratized NLP. Huge thanks to Huggingface for this."
question = "What has Huggingface done?"
input_text = "question: {:s}  context: {:s}".format(question, context)

encoding = tokenizer(input_text, return_tensors="pt")
output = self.model.generate(**encoding)
answer = tokenizer.decode(output['sequences'], skip_special_tokens=True)

Metrics

Average Normalized Levenshtein Similarity (ANLS)

The standard metric for text-based VQA tasks (ST-VQA and DocVQA). It evaluates the method's reasoning capabilities while smoothly penalizes OCR recognition errors. Check Scene Text Visual Question Answering for detailed information.

Answer Page Prediction Accuracy (APPA)

In the MP-DocVQA task, the models can provide the index of the page where the information required to answer the question is located. For this subtask accuracy is used to evaluate the predictions: i.e. if the predicted page is correct or not. Check Hierarchical multimodal transformers for Multi-Page DocVQA for detailed information.

Model results

Extended experimentation can be found in Table 2 of Hierarchical multimodal transformers for Multi-Page DocVQA. You can also check the live leaderboard at the RRC Portal.

ModelHF nameParametersANLSAPPA
Bert largerubentito/bert-large-mpdocvqa334M0.418351.6177
Longformer baserubentito/longformer-base-mpdocvqa148M0.528771.1696
BigBird ITC baserubentito/bigbird-base-itc-mpdocvqa131M0.492967.5433
LayoutLMv3 baserubentito/layoutlmv3-base-mpdocvqa125M0.453851.9426
T5 baserubentito/t5-base-mpdocvqa223M0.50500.0000
Hi-VT5rubentito/hivt5-base-mpdocvqa316M0.620179.23

Citation Information

@article{tito2022hierarchical,
  title={Hierarchical multimodal transformers for Multi-Page DocVQA},
  author={Tito, Rub{\`e}n and Karatzas, Dimosthenis and Valveny, Ernest},
  journal={arXiv preprint arXiv:2212.05935},
  year={2022}
}
Document Question Answering
Document Visual Question Answering
DocVQA
endpoints_compatible
pytorch
t5
text2text-generation
text-generation-inference
transformers

Contributors

rubentito

15 commits

RT
Rubèn Tito

1 commits

rubentito/t5-base-mpdocvqa

Model

T5 base fine-tuned on MP-DocVQA

1

16 commits

1 linked in READMEs

updated Nov 20, 2023

See the code

README

T5 base fine-tuned on MP-DocVQA

This is pretrained T5 base fine-tuned on Multipage DocVQA (MP-DocVQA) dataset.

This model was used as a baseline in Hierarchical multimodal transformers for Multi-Page DocVQA.

  • Results on the MP-DocVQA dataset are reported in Table 2.
  • Training hyperparameters can be found in Table 8 of Appendix D.

How to use

Here is how to use this model to get the features of a given text in PyTorch:

import torch
from transformers import T5Tokenizer, T5ForConditionalGeneration

tokenizer = LongformerTokenizerFast.from_pretrained("rubentito/t5-base-mpdocvqa")
model = LongformerForQuestionAnswering.from_pretrained("rubentito/t5-base-mpdocvqa")

context = "Huggingface has democratized NLP. Huge thanks to Huggingface for this."
question = "What has Huggingface done?"
input_text = "question: {:s}  context: {:s}".format(question, context)

encoding = tokenizer(input_text, return_tensors="pt")
output = self.model.generate(**encoding)
answer = tokenizer.decode(output['sequences'], skip_special_tokens=True)

Metrics

Average Normalized Levenshtein Similarity (ANLS)

The standard metric for text-based VQA tasks (ST-VQA and DocVQA). It evaluates the method's reasoning capabilities while smoothly penalizes OCR recognition errors. Check Scene Text Visual Question Answering for detailed information.

Answer Page Prediction Accuracy (APPA)

In the MP-DocVQA task, the models can provide the index of the page where the information required to answer the question is located. For this subtask accuracy is used to evaluate the predictions: i.e. if the predicted page is correct or not. Check Hierarchical multimodal transformers for Multi-Page DocVQA for detailed information.

Model results

Extended experimentation can be found in Table 2 of Hierarchical multimodal transformers for Multi-Page DocVQA. You can also check the live leaderboard at the RRC Portal.

ModelHF nameParametersANLSAPPA
Bert largerubentito/bert-large-mpdocvqa334M0.418351.6177
Longformer baserubentito/longformer-base-mpdocvqa148M0.528771.1696
BigBird ITC baserubentito/bigbird-base-itc-mpdocvqa131M0.492967.5433
LayoutLMv3 baserubentito/layoutlmv3-base-mpdocvqa125M0.453851.9426
T5 baserubentito/t5-base-mpdocvqa223M0.50500.0000
Hi-VT5rubentito/hivt5-base-mpdocvqa316M0.620179.23

Citation Information

@article{tito2022hierarchical,
  title={Hierarchical multimodal transformers for Multi-Page DocVQA},
  author={Tito, Rub{\`e}n and Karatzas, Dimosthenis and Valveny, Ernest},
  journal={arXiv preprint arXiv:2212.05935},
  year={2022}
}
Document Question Answering
Document Visual Question Answering
DocVQA
endpoints_compatible
pytorch
t5
text2text-generation
text-generation-inference
transformers

Contributors

rubentito

15 commits

RT
Rubèn Tito

1 commits