DARTS: Decoder-Aware Representation Tuning via Surgery for Model Merging
DARTS targets representation drift in merged decoder LLMs with entropy-weighted, position-aware correction, adding 0.1% parameters in the reported Llama-2-7B tests.
DARTS addresses hidden-state drift after merging task-tuned decoder models. It combines an **entropy-weighted L1 loss** for decision-critical tokens with a **per-position additive bias** to handle drift that accumulates under causal attention.
For builders experimenting with merged local models, the work suggests evaluating token-position effects rather than applying encoder-style correction uniformly. The reported module adds **0.1% of total parameters**.
DARTS addresses hidden-state drift after merging task-tuned decoder models. It combines an **entropy-weighted L1 loss** for decision-critical tokens with a **per-position additive bias** to handle drift that accumulates under causal attention. For builders experimenting with merged local models, the work suggests evaluating token-position effects rather than applying encoder-style correction uniformly. The reported module adds **0.1% of total parameters**. Evidence is limited to **Llama-2-7B** across HumanEval, GSM8K, and AlpacaEval. The abstract reports gains over standard surgery but gives no effect sizes, so generalization to newer model families and agent workloads remains open.
DARTS adds post-merge representation repair as a distinct customization lever beside training-data, mid-training, and self-distillation interventions. It narrows the repair problem to decoder-specific, position-dependent drift and offers a very small module, but its Llama-2-only evidence and missing effect sizes leave its value for current coding agents and newer open models unresolved.