XMerge: Cross-Axis Selection and Reconstructive Layer Merging for LLM Depth Compression

📅 2026-09-02
📈 Citations: 0
Influential: 0
📄 PDF
🤖 AI Summary
为了解决LLM深度压缩中质量损失问题,提出XMerge方法,通过跨轴选择和局部边界重建来移除transformer层,无需标签或微调。
📝 Abstract
Removing complete transformer layers preserves a standard serving architecture, but existing depth-compression methods can lose substantial quality, and the loss varies unpredictably across models. We introduce XMerge, a post-training method with two components. Cross-axis selection identifies a block with low relative-magnitude and angular hidden-state change, and local boundary reconstruction re-fits the adjacent surviving block to match the original two-block output. XMerge uses no task labels or end-to-end fine-tuning, and it introduces neither architectural changes nor additional inference-time parameters. Across seven Llama and Qwen backbones (0.5B-8B), five published baselines, and three layer-reduction levels, its advantage over baselines is largest at the most aggressive removal: at k=4 it ranks first on six of seven backbones on CORE (a 22-task aggregate) and, separately, on six of seven on MMLU (five of seven on both at once), while avoiding the large perplexity increases of several competing operators. In a task-level bootstrap, the 95% confidence intervals for the three largest CORE margins exclude zero; the remaining margins are consistent with ties. Across the 14 (model, regime) cells it is also the only evaluated operator that never collapses, ranking top-2 in both zero-shot and in-context regimes; on a first calibration probe (one backbone) it is the best-calibrated operator. Ablations show that local reconstruction provides most of the gain, while cross-axis fusion helps when the two selection axes disagree. The additional construction cost is recovered through per-token decode savings after roughly tens of thousands of requests.
Problem

Research questions and friction points this paper is trying to address.

depth compression
transformer layers
quality loss
model performance
Innovation

Methods, ideas, or system contributions that make the work stand out.

Cross-axis selection
Local boundary reconstruction
Post-training method
Depth compression
No architectural changes
🔎 Similar Papers
2024-06-24Conference on Empirical Methods in Natural Language ProcessingCitations: 2
💼 Related Jobs
No related jobs found.
J
Jundong Hu
PayPal AI
S
Shekar Ramachandran
PayPal AI