Research Article

Enhancing Text Generation via Parse Tree Embedding

Table 1

Performance of the Tree-VAE and the four baselines on the Yelp review generation, e-mail generation, and COCO image captions generation tasks. Higher is better. “Tree-VAE (WA)” and “Tree-VAE (WOA)” represent Tree-VAE with attention mechanism and Tree-VAE without attention mechanism, respectively. BLEU-2, BLEU-3, BLEU-4, and BLEU-5 are the BLEU scores.

Yelp reviewBLEU-2BLEU-3BLEU-4BLEU-5

MLE0.6090.4820.4770.409
SVAE0.6630.6540.5870.571
SeqGAN0.7680.5980.5640.507
RankGAN0.7950.6830.6100.554
Tree-VAE (WA)0.8170.6540.6370.624
Tree-VAE (WOA)0.8060.6970.6640.631

E-mailBLEU-2BLEU-3BLEU-4BLEU-5

MLE0.4730.4620.3170.306
SVAE0.5480.5170.4410.353
SeqGAN0.6840.5960.5780.511
RankGAN0.6310.5600.4930.445
Tree-VAE (WA)0.7030.6670.5930.512
Tree-VAE (WOA)0.7110.6700.6000.527

COCO image captionsBLEU-2BLEU-3BLEU-4BLEU-5

MLE0.7390.6420.5310.499
SVAE0.8010.7840.6680.633
SeqGAN0.8340.8050.7080.693
RankGAN0.8170.8260.7650.614
Tree-VAE (WA)0.8580.8230.7730.707
Tree-VAE (WOA)0.8530.8190.7940.730