The objective of this project is to design a model to generate 3 different kinds of headlines(Chinese) of news.
The style here are offcial style, normal style, and clickbait style.
The model is based on the BART from huggingface. The style guided mechanisms are applied in Decoder.
The cleaned version dataset can be collected from https://drive.google.com/drive/folders/1aCBBZAiDwC_Duumr3vRJGF30vtGuAgrB?usp=sharing
The two tasks share the encoder.
As shown in above figure, The style guided mechanisms are implemented by using style-guided normalization layer and style guided attention layer.