English

Deterministic Continuous Replacement: Fast and Stable Module Replacement in Pretrained Transformers

Machine Learning 2025-11-25 v1 Artificial Intelligence Computer Vision and Pattern Recognition

Abstract

Replacing modules in pretrained models, especially swapping quadratic self-attention for efficient attention alternatives, poses a hard optimization problem: cold-start reinitialization destabilizes frozen backbones. We isolate this core stability challenge in a controlled study. Deterministic Continuous Replacement (DCR) blends teacher and student outputs with a deterministic, annealed weight. Theoretically, DCR eliminates gate-induced gradient variance inherent to stochastic replacement. In a single-seed study, DCR attains faster convergence and stronger alignment than stochastic gating and distillation baselines on controlled attention replacement, establishing a foundation for heterogeneous operator swaps.

Keywords

Cite

@article{arxiv.2511.18670,
  title  = {Deterministic Continuous Replacement: Fast and Stable Module Replacement in Pretrained Transformers},
  author = {Rowan Bradbury and Aniket Srinivasan Ashok and Sai Ram Kasanagottu and Gunmay Jhingran and Shuai Meng},
  journal= {arXiv preprint arXiv:2511.18670},
  year   = {2025}
}

Comments

Accepted to NeurIPS 2025 ScaleOPT Workshop; 8 pages; includes figures

R2 v1 2026-07-01T07:51:20.534Z