Neural Style Transfer for Text and Chats


  • Machine Learning, NLP


Mentors :

  • Abhinav Raghuvanshi & Siddhesh Pawar

Mentees :

  • 6-8


Text Style Transfer (TST) is an important task in natural language generation, which aims to control certain attributes in the generated text, such as politeness, emotion, humor, and many others while preserving the context. As a use case for the TST algorithms, we would be making a simple Shakespearean paraphraser (See here, it's fine if you don't get the technical terms at the start, that’s what the mentors are here for !)

Stage 1
We aim to build a Machine Learning Model (a model is nothing more than a program that is backed by complex mathematical algos) that could manipulate the way you feel after reading a chunk of text. In simpler words you would be working on and learning how to make a program that could change the sentiment carried by that text without changing the content carried by the text.

Stage 2
In this project, we would be working on neural networks as well as the recently introduced transformer models for text style transfer (starting with classical methods). Towards the end, we would also be looking for application of style transfer algorithms to chatbots (this depends on individual progress at the end).

Soft Prerequisites (Desired but not necessary):
This project can be a great fit for you if you enjoy reading various genres of literature along with creative writing (Of Course enthusiasm for programming is a must ). We would also be considering a use case of author style transfer using neural networks. You should have working knowledge or some basic experience in python/C++ before applying (this needs to be mentioned in the SoP). It is advised to send your SoP ASAP with commenting access so that we can add feedback and also get an idea if you're fit for the project.

Tentative Timeline :

Week Work
Week 1-2 Revisiting Python and Introduction to the transformer library and pytorch
Week 3 Implementing RNNs and LSTMs in pytorch
Week 4 Basics of Neural Style transfer and implementing a RNN based TST model
Week 5-6 Implementing transformer based NST models
Week 7 Use case: Style transfer on Chats