A Local Perturbation Theory for Cross-Domain Interference and Recovery in Multi-Domain RL

TL;DR

Proposes a local perturbation theory revealing low-dimensional conflict subspaces in multi-domain RL, with short refreshes improving performance by geometric contraction.

cs.LG 🔴 Advanced 2026-06-01 19 views
Lei Yang Siyu Ding Deyi Xiong
Reinforcement Learning Multi-Domain Adaptation Parameter Perturbation Interference & Recovery Model Interpretability

Key Findings

Methodology

This paper develops a local perturbation framework analyzing how sparse, low-magnitude parameter updates interact along shared active computation routes. By measuring activation overlaps, update directions, and parameter changes, combined with theoretical derivations, it models interference as second-order damage concentrated in a low-dimensional shared conflict subspace. The approach integrates gradient analysis, path overlap metrics, and geometric contraction theorems to explain how short task-specific refreshes geometrically shrink harmful components, enabling targeted recovery.

Key Results

  • In Math tasks, performance improved from 57.66 to 66.04 after a brief refresh, with an overall average score of 66.39, outperforming baselines. Parameter updates were extremely sparse (77%-89% < 1e-7), yet shared active routes and directional analysis explained cross-task effects. Proxy conflict coordinate rollback further validated localized damage. The theory accurately predicts that short refreshes geometrically contract harmful components in shared subspaces, enabling effective, selective recovery.

Significance

This work challenges the traditional view that global gradient conflicts cause interference, highlighting instead the role of localized, path-based second-order damage in multi-task RL. It provides a mechanistic understanding of how sparse, shared computation routes lead to interference and how targeted, geometric contraction can restore task performance. The findings have broad implications for designing more robust, adaptable models capable of rapid, localized repair, advancing the fields of continual learning and multi-task AI.

Technical Contribution

The paper introduces a novel theoretical model linking sparse parameter updates, shared active computation routes, and second-order curvature-sensitive damage within a low-dimensional conflict subspace. It formalizes the geometric contraction of harmful components via short refreshes, supported by rigorous propositions and theorems. This framework extends current understanding beyond gradient conflict, offering a new paradigm for localized interference analysis and repair strategies in large models, with potential for practical deployment.

Novelty

This is the first comprehensive model attributing multi-task interference to localized, low-dimensional second-order damage along shared active routes, rather than global gradient conflicts. It combines path-level activation analysis, sparse parameter perturbation, and geometric contraction to explain and mitigate interference, representing a significant conceptual advance over prior work focused on parameter overlap or gradient orthogonality.

Limitations

  • The model assumes high sparsity and low-dimensional shared conflict subspaces, which may not hold in more complex, real-world scenarios with dense interactions.
  • Short refresh strategies are effective in controlled settings but may face stability issues or limited scalability in very large models or highly adversarial tasks.
  • The theoretical assumptions rely on local smoothness and curvature bounds, requiring validation in diverse, real-world multi-task environments.

Future Work

Future research will explore dynamic path identification, adaptive intervention strategies, and extension to larger models and more complex tasks. Developing automated methods for conflict subspace detection and real-time correction could further enhance model robustness. Integrating these insights into continual learning frameworks promises more resilient AI systems capable of rapid, localized adaptation.

AI Executive Summary

This study offers a groundbreaking perspective on the mechanisms underlying cross-domain interference in reinforcement learning models. Traditional explanations centered on global gradient conflicts are insufficient, as experiments reveal near-orthogonal gradients despite significant performance degradation. The authors propose a local perturbation model, emphasizing that sparse, low-magnitude parameter updates interact along shared active computation routes. These routes form a low-dimensional conflict subspace where second-order curvature-sensitive damage accumulates, causing localized interference. The theory predicts that short, targeted refreshes can geometrically contract these harmful components, effectively restoring performance with minimal collateral damage. Empirical validation on a multi-task benchmark demonstrates that a brief refresh after sequential training on Code, Math, QA, and CW tasks recovers Math performance from 57.66 to 66.04, while largely preserving other domains, resulting in an average score of 66.39. Additional experiments with proxy conflict coordinate rollback further confirm the localized nature of damage and repair. These findings challenge the conventional gradient conflict paradigm, highlighting the importance of shared active routes and low-dimensional geometry in understanding and mitigating interference. The work opens new avenues for designing robust, self-healing multi-task models capable of rapid, localized adaptation, with broad implications for continual learning, AI robustness, and efficient model maintenance.

Deep Analysis

Background

Recent advances in large-scale pretraining have significantly improved language understanding and generation. However, multi-task fine-tuning often introduces interference, degrading performance on some tasks while improving others. Traditional explanations attribute this to global gradient conflicts or parameter overlap, but these do not fully account for observed phenomena such as near-orthogonal gradients with substantial task interference. Mechanistic studies suggest that knowledge and skills are localized within specific neuron subsets and pathways, emphasizing the importance of understanding internal computation routes. Despite progress, the precise mechanisms of cross-task interference and recovery remain elusive, especially in the context of sparse, low-dimensional parameter changes and shared active routes.

Core Problem

The core challenge lies in explaining why models trained sequentially on multiple domains exhibit selective, asymmetric degradation, despite near-orthogonal full-model gradients. Existing theories fail to account for how sparse, localized parameter updates along shared computation paths can cause significant interference. Furthermore, understanding how short, targeted interventions can effectively restore performance without full retraining is critical. Addressing these issues requires a fine-grained mechanistic framework that captures the interplay between sparse parameter changes, shared active routes, and the model's curvature-sensitive directions. This understanding is vital for developing robust multi-task learning systems capable of rapid, localized repair.

Innovation

The paper introduces a local perturbation framework that models interference as second-order damage concentrated in a low-dimensional shared conflict subspace. It emphasizes the role of shared active computation routes, where sparse parameter updates can cause localized harm depending on their directional alignment with curvature-sensitive directions. The key innovation is the geometric contraction theorem, which shows that short, targeted refreshes can rapidly diminish harmful components in this subspace. This approach departs from traditional gradient conflict explanations, offering a mechanistic, path-level understanding of interference and repair, with potential for practical deployment in multi-task models.

Methodology

  • �� Constructed a multi-task RL setting with datasets from OpenR1-math, KlearReasoner-Code, SuperGPQA, and Creative-Writing. • Trained single-task experts and performed sequential multi-task training following the Omni-Thinker curriculum. • Analyzed full-model gradients, parameter changes, and neuron activation overlaps to identify localized conflict sources. • Measured shared active routes during inference, quantifying overlap and directional alignment. • Developed a theoretical model assuming local smoothness, sparse updates, and low-dimensional conflict subspaces. • Derived propositions linking second-order damage to curvature and shared route projections. • Validated predictions through short refresh experiments and proxy coordinate rollback, demonstrating geometric contraction of harmful components.

Experiments

Using datasets like AIME24/25/26, OlympiadBench, LiveCodeBench, SuperGPQA-test, and WritingBench, the experiments involved training single-task experts, sequential multi-task models, and baseline joint training. Performance metrics included accuracy, F1, and task-specific scores. The key hyperparameters involved learning rates, refresh steps, and update sparsity thresholds. Ablation studies examined the effects of refresh duration, proxy conflict set size, and path overlap. The experimental setup validated the theory that localized, low-dimensional second-order damage causes interference, and short refreshes effectively mitigate it.

Results

Sequential training caused significant Math performance drops from 66.49 to 57.66, recoverable to 66.04 after a brief refresh. The average multi-task score reached 66.39, surpassing baselines. Parameter changes were extremely sparse, yet shared active routes and directional misalignments explained cross-task effects. Proxy rollback experiments confirmed localized damage. The geometric contraction theorem accurately predicted rapid decay of harmful components, demonstrating the practical effectiveness of short refreshes in multi-task settings.

Applications

The findings enable the design of models that can rapidly repair local interference without full retraining, suitable for continual learning, AI assistants, and adaptive systems. The short refresh approach reduces computational costs and improves robustness in multi-task environments. Future integration with dynamic path detection and real-time correction could further enhance model resilience, making AI systems more autonomous and self-healing.

Limitations & Outlook

The model assumes high sparsity and low-dimensional shared conflict subspaces, which may not generalize to dense, complex interactions in real-world tasks. Short refresh strategies might face stability issues in very large models or highly adversarial scenarios. The theoretical assumptions rely on local smoothness and curvature bounds, requiring further validation across diverse, real-world datasets and tasks.

Plain Language Accessible to non-experts

想象一个工厂里有许多不同的生产线,每条生产线负责不同的产品,比如衣服、家具和电子产品。每条生产线都用一些共同的机器和路径,但操作方式不同。当工厂为了提高某条生产线的效率,调整了某些机器的设置,这些调整可能会影响到其他生产线的工作。特别是,如果这些调整只涉及少量机器,但这些机器都在多个生产线上共享,调整可能会引起其他生产线的效率下降。工厂的管理者发现,通过短时间重新调整某些关键机器,可以快速修正问题,避免大规模停工。这就像论文中提出的“局部扰动”和“短刷新”策略,帮助模型在多任务环境中快速修复局部干扰,保持整体运行的稳定。

ELI14 Explained like you're 14

想象你在学校参加不同的兴趣班,比如画画、音乐和运动。有时候,你在一个兴趣班学到的东西会影响到其他兴趣班,比如学会了新的画画技巧,可能会让你在音乐课上表现得更棒,或者反过来。有时候,学习新技能会让你之前的技能变得不那么好,就像你用一种新方法画画,但在音乐中却不适用。科学家们发现,当模型学习不同任务时,也会出现类似的情况。有些调整只影响很少的部分,但这些部分在不同任务中都被用到,所以会引起干扰。为了修复这个问题,他们提出了“短时间调整”的方法,就像你在学校里短暂地回顾某个兴趣班的内容,快速修正影响。这样,模型就能更好地在多个任务间切换,表现得更稳定、更聪明。

Abstract

Reinforcement learning (RL) post-training improves large language models (LLMs) on individual domains such as mathematical reasoning, code generation, question answering, and creative writing (CW), but training on one domain often degrades performance on others. Existing explanations based on catastrophic forgetting or global gradient conflict are incomplete: substantial interference can occur even when full-model gradients are nearly orthogonal. We show that single-domain RL produces sparse, small-magnitude parameter edits with weak overlap among top-changed neurons, while different domains still share substantial active computation routes on which update directions determine whether they act synergistically or conflict. Guided by this observation, we prove under a local perturbation model of multi-domain RL that later-domain training harms an earlier domain mainly through a second-order damage term, which under the observed sparse route structure concentrates in a low-dimensional shared conflict subspace. Moreover, a short domain refresh contracts the harmful component on this subspace, enabling selective recovery with limited collateral damage. Consistent with the theory, a brief Re-Math refresh after Code $\rightarrow$ Math $\rightarrow$ QA $\rightarrow$ CW recovers Math from 57.66 to 66.04 while largely preserving performance on the other domains, yielding the best average score of 66.39. Beyond refresh, a training-free rollback on a sparse proxy conflict coordinate set for the Math-QA pair partially restores Math, providing direct proxy-level evidence for localized damage. These results provide a localized mechanistic account of interference and recovery in multi-domain RL.

cs.LG cs.CL