arXiv AI By Yanggan Gu, Shuo Cai, Zihao Wang, Wenjun Wang, Yuanyi Wang, Pengkai Wang, Sirui Huang, Su Lu, Jianmin Wu, Hongxia Yang

FeatCal: Feature Calibration for Post-Merging Models

Read the original on arXiv AI →

FeatCal is a post‑merging calibration method that reduces feature drift between a merged model and its task experts by adjusting weights layer by layer in forward order using a small calibration set. It employs a closed‑form update that avoids gradient descent or iterative optimization. On CLIP and GLUE benchmarks, FeatCal outperforms existing calibration baselines, achieving higher accuracy with fewer calibration examples and lower computational cost.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv AI.

arXiv Machine Learning
5d ago

SMAT: Simple and Efficient Merge-Aware Training

SMAT (Simple and Efficient Merge-Aware Training) is a method that jointly optimizes expert loss and expected loss at simulated merged parameters, addressing the shortcomings of standard expert training for model merging. It models common merging operations—Scale, Mask, and Perturb—by sampling scaling coefficients, masks, and additive noise, and introduces efficient techniques such as periodic scheduling, kernel fusion, and parameter storage switching to keep training overhead minimal. Across multiple language and vision-language backbones, SMAT improves mean scores by 1.07–2.16 points over the strongest baseline while adding less than 2% training-time overhead.

By Yanggan Gu, Yuanyi Wang, Zhen Li, Shuo Cai, Yuhang Liu, Junzhuo Li, Zihao Wang, Hongxia Yang
arXiv AI
Oct 2

ReForge: Refining Merged Models with Anchor-Regularized Regression

ReForge is a bilevel optimization framework that refines merged models by treating module-wise refinement as Bayesian linear regression with an anchor-centered prior. The inner level produces a closed‑form MAP estimate from unlabeled calibration activations, while the outer level employs Bayesian optimization to jointly select regularization strengths and assembly scales using validation data. A data‑free variant replaces activation statistics with task‑vector Grams, enabling refinement without calibration examples, and across extensive vision and language benchmarks ReForge consistently outperforms existing plug‑and‑play anchor baselines, achieving significant accuracy gains on large‑scale tasks such as 20‑task ViT‑B/32 and eight‑task ViT‑L/14.

By Kaiyang Li, Shaobo Han, Qing Su, Shihao Ji
arXiv AI
Jun 4

Revisiting Model Stitching In the Foundation Model Era

arXiv:2603. 12433v3 Announce Type: replace-cross Abstract: Model stitching, connecting early layers of one model (source) to later layers of another (target) via a light stitch layer, has served as a probe of representational compatibility.

By Zheda Mai, Ke Zhang, Fu-En Wang, Zixiao Ken Wang, Albert Y. C. Chen, Lu Xia, Min Sun, Wei-Lun Chao, Cheng-Hao Kuo