Consolidation or Adaptation? PRISM: Disentangling SFT and RL Data via Gradient Concentration

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhao, Yang, Ouyang, Yangou, Ding, Xiao, Wang, Hepeng, Cai, Bibo, Xiong, Kai, Gao, Jinglong, Sun, Zhouhao, Du, Li, Qin, Bing, Liu, Ting
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910121729196032
author Zhao, Yang
Ouyang, Yangou
Ding, Xiao
Wang, Hepeng
Cai, Bibo
Xiong, Kai
Gao, Jinglong
Sun, Zhouhao
Du, Li
Qin, Bing
Liu, Ting
author_facet Zhao, Yang
Ouyang, Yangou
Ding, Xiao
Wang, Hepeng
Cai, Bibo
Xiong, Kai
Gao, Jinglong
Sun, Zhouhao
Du, Li
Qin, Bing
Liu, Ting
contents While Hybrid Supervised Fine-Tuning (SFT) followed by Reinforcement Learning (RL) has become the standard paradigm for training LLM agents, effective mechanisms for data allocation between these stages remain largely underexplored. Current data arbitration strategies often rely on surface-level heuristics that fail to diagnose intrinsic learning needs. Since SFT targets pattern consolidation through imitation while RL drives structural adaptation via exploration, misaligning data with these functional roles causes severe optimization interference. We propose PRISM, a dynamics-aware framework grounded in Schema Theory that arbitrates data based on its degree of cognitive conflict with the model's existing knowledge. By analyzing the spatial geometric structure of gradients, PRISM identifies data triggering high spatial concentration as high-conflict signals that require RL for structural restructuring. In contrast, data yielding diffuse updates is routed to SFT for efficient consolidation. Extensive experiments on WebShop and ALFWorld demonstrate that PRISM achieves a Pareto improvement, outperforming state-of-the-art hybrid methods while reducing computational costs by up to 3.22$\times$. Our findings suggest that disentangling data based on internal optimization regimes is crucial for scalable and robust agent alignment.
format Preprint
id arxiv_https___arxiv_org_abs_2601_07224
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Consolidation or Adaptation? PRISM: Disentangling SFT and RL Data via Gradient Concentration
Zhao, Yang
Ouyang, Yangou
Ding, Xiao
Wang, Hepeng
Cai, Bibo
Xiong, Kai
Gao, Jinglong
Sun, Zhouhao
Du, Li
Qin, Bing
Liu, Ting
Artificial Intelligence
Machine Learning
While Hybrid Supervised Fine-Tuning (SFT) followed by Reinforcement Learning (RL) has become the standard paradigm for training LLM agents, effective mechanisms for data allocation between these stages remain largely underexplored. Current data arbitration strategies often rely on surface-level heuristics that fail to diagnose intrinsic learning needs. Since SFT targets pattern consolidation through imitation while RL drives structural adaptation via exploration, misaligning data with these functional roles causes severe optimization interference. We propose PRISM, a dynamics-aware framework grounded in Schema Theory that arbitrates data based on its degree of cognitive conflict with the model's existing knowledge. By analyzing the spatial geometric structure of gradients, PRISM identifies data triggering high spatial concentration as high-conflict signals that require RL for structural restructuring. In contrast, data yielding diffuse updates is routed to SFT for efficient consolidation. Extensive experiments on WebShop and ALFWorld demonstrate that PRISM achieves a Pareto improvement, outperforming state-of-the-art hybrid methods while reducing computational costs by up to 3.22$\times$. Our findings suggest that disentangling data based on internal optimization regimes is crucial for scalable and robust agent alignment.
title Consolidation or Adaptation? PRISM: Disentangling SFT and RL Data via Gradient Concentration
topic Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2601.07224