Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Warmup Generations: A Task-Agnostic Approach for Guiding Sequence-to-Sequence Learning with Unsupervised Initial State Generation

About

Traditional supervised fine-tuning (SFT) strategies for sequence-to-sequence tasks often train models to directly generate the target output. Recent work has shown that guiding models with intermediate steps, such as keywords, outlines, or reasoning chains, can significantly improve performance, coherence, and interpretability. However, these methods often depend on predefined intermediate formats and annotated data, limiting their scalability and generalizability. In this work, we introduce a task-agnostic framework that enables models to generate intermediate "warmup" sequences. These warmup sequences, serving as an initial state for subsequent generation, are optimized to enhance the probability of generating the target sequence without relying on external supervision or human-designed structures. Drawing inspiration from reinforcement learning principles, our method iteratively refines these intermediate steps to maximize their contribution to the final output, similar to reward-driven optimization in reinforcement learning with human feedback. Experimental results across tasks such as translation, summarization, and multi-choice question answering for logical reasoning show that our approach outperforms traditional SFT methods, and offers a scalable and flexible solution for sequence-to-sequence tasks.

Senyu Li, Zipeng Sun, Jiayi Wang, Xue Liu, Pontus Stenetorp, Siva Reddy, David Ifeoluwa Adelani• 2025

Related benchmarks

TaskDatasetResultRank
Abstractive SummarizationXsum
ROUGE-140.67
18
Logical reasoning multi-choice QALogiQA v2 (test)
Macro F1 Score55.5
6
Machine TranslationWMT Flores de-en (test)
BLEU34.71
4
Machine TranslationWMT Flores ru-en (test)
BLEU25.76
4
Machine TranslationWMT Flores zh-en (test)
BLEU18.66
4
Machine TranslationWMT Flores fr-en (test)
BLEU35.63
4
Machine TranslationWMT Flores en-de (test)
BLEU26.84
4
Machine TranslationWMT Flores en-ru (test)
BLEU16.3
4
Machine TranslationWMT Flores en-zh (test)
BLEU27.89
4
Machine TranslationWMT Flores en-fr (test)
BLEU36.29
4
Showing 10 of 11 rows

Other info

Follow for update