HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

ViT5: Pretrained Text-to-Text Transformer for Vietnamese Language Generation

Long Phan; Hieu Tran; Hieu Nguyen; Trieu H. Trinh

ViT5: Pretrained Text-to-Text Transformer for Vietnamese Language Generation

Abstract

We present ViT5, a pretrained Transformer-based encoder-decoder model for the Vietnamese language. With T5-style self-supervised pretraining, ViT5 is trained on a large corpus of high-quality and diverse Vietnamese texts. We benchmark ViT5 on two downstream text generation tasks, Abstractive Text Summarization and Named Entity Recognition. Although Abstractive Text Summarization has been widely studied for the English language thanks to its rich and large source of data, there has been minimal research into the same task in Vietnamese, a much lower resource language. In this work, we perform exhaustive experiments on both Vietnamese Abstractive Summarization and Named Entity Recognition, validating the performance of ViT5 against many other pretrained Transformer-based encoder-decoder models. Our experiments show that ViT5 significantly outperforms existing models and achieves state-of-the-art results on Vietnamese Text Summarization. On the task of Named Entity Recognition, ViT5 is competitive against previous best results from pretrained encoder-based Transformer models. Further analysis shows the importance of context length during the self-supervised pretraining on downstream performance across different settings.

Code Repositories

vietai/vit5
Official
jax
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
abstractive-text-summarization-on-vietnewsViT5 large
Rouge-1: 63.37
Rouge-2: 34.24
Rouge-L: 43.55
abstractive-text-summarization-on-vietnewsViT5 base
Rouge-1: 62.77
Rouge-2: 33.16
Rouge-L: 42.75
named-entity-recognition-in-vietnamese-on-1ViT5 base
F1 (%): 94.5

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
ViT5: Pretrained Text-to-Text Transformer for Vietnamese Language Generation | Papers | HyperAI