LGAIApr 22

IRIS: Interpolative Rényi Iterative Self-play for Large Language Model Fine-Tuning

arXiv:2604.2093356.4h-index: 3
Predicted impact top 41% in LG · last 90 daysOriginality Highly original
AI Analysis

Provides a unified and adaptive self-play fine-tuning method for LLMs that outperforms fixed-divergence approaches, addressing the challenge of varying distributional gaps during training.

IRIS introduces a Rényi-based self-play fine-tuning framework with an adjustable objective that interpolates between existing divergence-based methods, achieving 44.57% average score across ten benchmarks and surpassing supervised fine-tuning on 200k samples using only 26k annotated samples.

Self-play fine-tuning enables large language models to improve beyond supervised fine-tuning without additional human annotations by contrasting annotated responses with self-generated ones. Many existing methods rely on a fixed divergence regime. SPIN is closely related to a KL-based regime, SPACE to a Jensen-Shannon-style objective via noise contrastive estimation, and SPIF to $χ^2$-regularized self-play. Since these divergences exhibit different strengths depending on the distributional gap between model and target, no single choice appears to provide favorable learning dynamics across training stages. We propose IRIS (Interpolative Rényi Iterative Self-play), a Rényi-based self-play fine-tuning framework with a continuously adjustable objective. IRIS decomposes into two independent tilted risk terms over annotated and synthetic data, with exponential importance weights controlled by the order parameter $α$. We show that several self-play objectives can be interpreted as limiting or representative regimes at particular values of $α$, providing a unified theoretical perspective on these methods. An adaptive order schedule further adjusts $α$ to the distributional gap, shifting from sharper importance weighting early in training to smoother refinement near convergence. Theoretically, we establish the fixed-point property of IRIS and analyze how $α$ controls gradient concentration. Experiments on Zephyr-7B and Qwen2.5-3B across ten benchmarks show that IRIS improves upon baselines, reaching 44.57\% average score with gains across iterations. In our setting, IRIS with only 26$k$ annotated samples surpasses standard supervised fine-tuning trained on the full 200$k$ dataset.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes