Capturing Classic Authorial Style in Long-Form Story Generation with GRPO Fine-Tuning
Summary
A novel two-stage pipeline has been developed to optimize authorial style in long-form story generation, addressing challenges in subjective human evaluation and unstable automatic methods. The first stage involves training a style-similarity judge by fine-tuning a sentence-transformer with authorship-verification supervision, which then calibrates its similarity outputs into a bounded [0,1] reward. In the second stage, this judge serves as the primary reward for Group Relative Policy Optimization (GRPO), used to fine-tune an 8B story generator for style-conditioned writing. This approach bypasses the accept/reject supervision typically required by Direct Preference Optimization (DPO). The GRPO-trained 8B model achieved an average style score of 0.893 across target authors Mark Twain, Jane Austen, Charles Dickens, and Thomas Hardy, surpassing open-weight baselines. This demonstrates that AV-calibrated reward modeling offers a practical mechanism for controllable long-form style transfer, even with moderate model size and training budgets.
Key takeaway
For NLP Engineers developing style-conditioned text generation, you should consider adopting a two-stage reward modeling approach. This method, utilizing an AV-calibrated style-similarity judge with GRPO fine-tuning, allows you to achieve high style transfer scores (e.g., 0.893) on 8B models without the extensive accept/reject supervision required by DPO. This significantly reduces data annotation burdens and training costs, making sophisticated style transfer more accessible for your projects.
Key insights
AV-calibrated reward modeling enables controllable long-form style transfer for story generation without direct preference supervision.
Principles
- Style judgment can be quantified via similarity judges.
- GRPO fine-tuning avoids explicit preference data.
- Moderate models can achieve high style transfer.
Method
A two-stage pipeline: first, train a sentence-transformer as a style-similarity judge with authorship-verification, calibrating rewards [0,1]. Second, use this judge as a reward in GRPO to fine-tune an 8B story generator.
In practice
- Fine-tune 8B models for specific authorial styles.
- Develop style judges using authorship verification.
- Apply GRPO for style transfer without DPO data.
Topics
- Long-Form Story Generation
- Authorial Style Transfer
- GRPO Fine-Tuning
- Reward Modeling
- Sentence Transformers
- Authorship Verification
Best for: Research Scientist, AI Scientist, NLP Engineer, Machine Learning Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Paper Index on ACL Anthology.