Downloads · 30 days
11
3% of all-time downloads
hallisky/cds_style_classifier
cds_style_classifier is a text classification model from hallisky. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as apache-2.0.
If you use/reference this work, please cite us with:
Downloads · 30 days
11
3% of all-time downloads
All-time downloads
334
Public
Repo size
5.7 GB
Likes
1
Public
Click a slice to open those files.
.pt2.8 GB · 67%
From the Hugging Face model README
If you use/reference this work, please cite us with:
@inproceedings{hallinan-etal-2023-steer,
title = "{STEER}: Unified Style Transfer with Expert Reinforcement",
author = "Hallinan, Skyler and
Brahman, Faeze and
Lu, Ximing and
Jung, Jaehun and
Welleck, Sean and
Choi, Yejin",
editor = "Bouamor, Houda and
Pino, Juan and
Bali, Kalika",
booktitle = "Findings of the Association for Computational Linguistics: EMNLP 2023",
month = dec,
year = "2023",
address = "Singapore",
publisher = "Association for Computational Linguistics",
url = "https://aclanthology.org/2023.findings-emnlp.506",
doi = "10.18653/v1/2023.findings-emnlp.506",
pages = "7546--7562",
abstract = "While text style transfer has many applications across natural language processing, the core premise of transferring from a single source style is unrealistic in a real-world setting. In this work, we focus on arbitrary style transfer: rewriting a text from an arbitrary, unknown style to a target style. We propose STEER: Unified Style Transfer with Expert Reinforcement, a unified frame-work developed to overcome the challenge of limited parallel data for style transfer. STEER involves automatically generating a corpus of style-transfer pairs using a product of experts during decoding. The generated offline data is then used to pre-train an initial policy before switching to online, off-policy reinforcement learning for further improvements via fine-grained reward signals. STEER is unified and can transfer to multiple target styles from an arbitrary, unknown source style, making it particularly flexible and efficient. Experimental results on a challenging dataset with text from a diverse set of styles demonstrate state-of-the-art results compared to competitive baselines. Remarkably, STEER outperforms the 175B parameter instruction-tuned GPT-3 on overall style transfer quality, despite being 226 times smaller in size. We also show STEER is robust, maintaining its style transfer capabilities on out-of-domain data, and surpassing nearly all baselines across various styles. The success of our method highlights the potential of RL algorithms when augmented with controllable decoding to overcome the challenge of limited data supervision.",
}