Value Residual Learning
While Transformer models have achieved remarkable success in various domains, the effectiveness of information propagation through deep networks remains a critical challenge.Standard hidden state residuals often fail to adequately preserve initial token-level information in deeper layers.This paper introduces ResFormer, a novel architecture that enhances information flow by incorporating value residual connections in addition to hidden state residuals.And a variant is SVFormer, where all layers share the first layer's value embedding.Comprehensive empirical evidence demonstrates ResFormer achieves equivalent validation loss with 16.11% fewer model parameters and 20.3% less training data compared to Transformer, while maintaining similar memory usage and computational cost.Besides, SVFormer reduces KV cache size by nearly half with only a small performance penalty and can be integrated with other KVefficient methods, yielding further reductions in KV cache, with performance influenced by sequence length and cumulative learning rate.
Authors
- Tianyi Wu (ORCID: https://orcid.org/0000-0001-7434-0487)
- Zheyu Zhou (ORCID: https://orcid.org/0000-0001-7166-6799)
- Zhiyun Jiang
- Zhenzhong Lan (ORCID: https://orcid.org/0000-0003-4763-6148)
- Fares Obeid
Publication Details
- Published
- 2025-01-01
- DOI
- https://doi.org/10.18653/v1/2025.acl-long.1375
- Primary Topic
- Bayesian Modeling and Causal Inference
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- Westlake University