Skip to content

Deep Generative Model Class Project, use cVAE + RNN(GRU/LSTM) + GAN to produce adversarial sentences

Notifications You must be signed in to change notification settings

Randy-whiteSugar/Adversarial-Sample-Generation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Natural Language Adversarial Sample Generation

Model Architecture

  • Overall Model Architecture : cVAE + GAN 模型架构

  • Generator : we implemented two generators respectively with LSTM and GRU (folders: LSTM_VAE and GRU_VAE). 生成器

Model Results

current results are not ideal enough but you can distinguish positive results from negative ones as below.

  • positive:

    • i movie is a of the best films ever ever seen it i the was it to lot of watch a the i was and seen it lot of the the the movie was the movie of not a lot of the and own is i to be a and and the it thing movie is a the the i was enjoyed to see it and have not i to the the movie the i and lot times ago to i a and movie is the i a that the the
  • negative:

    • i this is a of the worst movies ever ever seen the i was it seen favorite was have a disappointed and of the and the minutes of i be the the and was a funny and and existent have the and the not are the and are know the the as the and the of the and and the and the the and are have the of and of the of the the is the that and and br

Data and Pre-training

  • We use the dataset IMDB(25,000 movies reviews, positive comments and negative comments account for half of each).
  • We have trained a textCNN model as the target model, the accuracy of which reaches 98.4%.

Training and evaluation

  • Train
    • Run the file train_with_LSTM.py(recommended because we have trained once) or train_with_GRU.py to train the model with a LSTM/GRU generator.
  • Evaluation
    • Run the file evaluation.py to evaluate the model from three aspects: generation speed, text quality and attack success rate.

Reference

Authors

  • Fang Chen
  • Sang Yuchen

License

  • For academic and non-commercial use only.

About

Deep Generative Model Class Project, use cVAE + RNN(GRU/LSTM) + GAN to produce adversarial sentences

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published