POLARIS: Guiding Small Models to Write Long Stories

Rishanth Rajendhran, Jenna Russell, Mohit Iyyer, John Frederick Wieting


Abstract
Small open-weight models struggle at long-form creative writing: their generated stories either fall far short of the requested length, or their quality significantly degrades as length increases, especially when compared to frontier models. We present **POLARIS** (**P**olicy **O**ptimization with **L**LM-as-a-judge rewards and **A**nchored-**R**eference **I**njection for **S**torywriting), **a lower-compute GRPO recipe** with two key ingredients: a frontier LLM judge with a **structured *Story Quality* rubric** as the online reward, and **human-reference injection (HRI)**, where a teacher-forced human-written story serves as a high-reward anchor within each GRPO group. By applying our training recipe to Qwen3.5-9B, using a dataset of ∼1.4K prompt–story pairs derived from 100 short-story anthologies and 4 A100 GPUs, we obtain **POLARIS-9B**. Across five benchmarks spanning in-distribution and out-of-distribution prompts and rubrics, POLARIS-9B is **competitive with much larger open-weight models** while following length instructions more closely. A blinded human evaluation confirms that POLARIS-9B is preferred to the base Qwen3.5-9B and on par with Qwen3.5-27B. Despite training only on stories up to 4k words, POLARIS-9B preserves quality on prompts requesting stories up to 3× the training length, a regime where most open-weight models degrade substantially in quality, length adherence, or both. More broadly, our results suggest that **length generalization is a meaningful stress test for creative-writing models** and a useful lens for distinguishing otherwise close models.
Anthology ID:
2026.emnlp-main.627
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
14392–14428
Language:
URL:
https://aclanthology.org/2026.emnlp-main.627/
DOI:
Bibkey:
Cite (ACL):
Rishanth Rajendhran, Jenna Russell, Mohit Iyyer, and John Frederick Wieting. 2026. POLARIS: Guiding Small Models to Write Long Stories. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 14392–14428, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
POLARIS: Guiding Small Models to Write Long Stories (Rajendhran et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.627.pdf