paper-with-me

홈 › Papers

Spectral Rewiring for Exploration, Purification, and Model Merging

2026-07-03 · Zhilong Zhang, Hongli Yu, Huan-ang Gao, Hanlin Wu, Yuxuan Song, Wei-Ying Ma, Ya-Qin Zhang, Hao Zhou arxiv

Reinforcement learning has become a standard post-training recipe for large language models, but dense full-parameter updates create two deployment-relevant bottlenecks: suppressed reasoning performance, often reflected by premature saturation of test-time scaling, and interference when consolidating multiple capabilities through multi-domain training or model merging. We show that the reasoning-effective component of these updates is largely concentrated in the base model's spectral space, motivating Subspace-Aligned Rewiring (SAR), a post-hoc editing method that retains this spectral core while removing orthogonal components. SAR therefore preserves reasoning gains and filters residual update directions that suppress performance or amplify cross-domain interference. Across several model families and scales, SAR extracts compact reasoning cores using as little as approximately 0.58% of total parameters: it preserves over 99% of post-training performance and improves high-k exploration in mathematical reasoning, and generalizes to agentic coding by improving six of seven open benchmarks on an in-house model. SAR also purifies mixed-domain training updates by releasing suppressed coding capability while maintaining math reasoning and instruction following. It further enables model merging across experts, yielding cross-domain generalization that surpasses previous merging baselines and even the best single-domain experts. Overall, SAR shows that extracting reasoning-effective updates from parameter geometry can serve as a training-free mechanism to improve reasoning and multi-domain performance.

📄 PDF Abstract BibTeX arXiv:2607.03065

Code (0)

등록된 구현이 없습니다.

Tasks

Reinforcement LearningMathematical ReasoningDomain GeneralizationInstruction Following

Similar Papers 제목 키워드 기반

GNNs Getting ComFy: Community and Feature Similarity Guided Rewiring

2025-02-07 · Celia Rubio-Madrigal, Adarsh Jamadandi, Rebekka Burkholz

Maximizing the spectral gap through graph rewiring has been proposed to enhance the performance of message-passing graph neural networks (GNNs) by addressing over-squashing. However, as we show, minimizing the spectral g…

Node Classification

Locality-Aware Graph-Rewiring in GNNs

2023-10-02 · Federico Barbero, Ameya Velingker, Amin Saberi, Michael Bronstein 외

Graph Neural Networks (GNNs) are popular models for machine learning on graphs that typically follow the message-passing paradigm, whereby the feature of a node is updated recursively upon aggregating information over it…

Inductive Bias

Purify++: Improving Diffusion-Purification with Advanced Diffusion Models and Control of Randomness

2023-10-28 · Boya Zhang, Weijian Luo, Zhihua Zhang

Adversarial attacks can mislead neural network classifiers. The defense against adversarial attacks is important for AI safety. Adversarial purification is a family of approaches that defend adversarial attacks with suit…

Adversarial Purification

Joint Graph Rewiring and Feature Denoising via Spectral Resonance

2024-08-13 · Jonas Linkerhägner, Cheng Shi, Ivan Dokmanić

When learning from graph data, the graph and the node features both give noisy information about the node labels. In this paper we propose an algorithm to jointly denoise the features and rewire the graph (JDR), which im…

DenoisingGraph LearningNode Classification

PANDA: Expanded Width-Aware Message Passing Beyond Rewiring

2024-06-06 · Jeongwhan Choi, Sumin Park, Hyowon Wi, Sung-Bae Cho 외

Recent research in the field of graph neural network (GNN) has identified a critical issue known as "over-squashing," resulting from the bottleneck phenomenon in graph structures, which impedes the propagation of long-ra…

Graph ClassificationGraph Neural NetworkGraph RegressionNode Classification