TransWikia.com

Question of pretraining text-generation task, it seems that pretraining is not work for a small model?

Data Science Asked on September 30, 2021

My task is to generate keywords from sentences.

I pretrain a text-generation model. I mask the sentences’ tokens and predict the whole sentences’ tokens.

Pretraining batch_size = 8 and step = 1000000

I haven’t observed improvement from pretraining. BLEU score is 10.5 for not pretraining, BLEU score is 9.5 for pretraining.

Code

I take the python code from

https://github.com/google-research/pegasus/blob/master/pegasus/models/transformer.py#L38

hidden_size = 512
num_encoder_layers = 3
num_decoder_layers = 3

Discussion

The task is to generate keyword from sentences.
The keyword may not appear in the sentences.
So input masked sentences to predict whole sentences, it is not benefit the keywords generation task.
Input masked sentences to predict whole sentences, it do not have relation to the keywords generation task.
Am I right? Is it the reason that pretraining do not improve the BLEU score?

Thank you very much.

One Answer

1, I pad some zeros in the input tokens for multi sentences. The output positions of output tokens should be exactly same to the input tokens, which means I should keep the padding zeros in the output tokens.

2, The pretraining time should be longer.

Correct answer by DunkOnly on September 30, 2021

Add your own answers!

Ask a Question

Get help from others!

© 2024 TransWikia.com. All rights reserved. Sites we Love: PCI Database, UKBizDB, Menu Kuliner, Sharing RPP