CR-Net: Scaling Parameter-Efficient Training with Cross-Layer Low-Rank Structure
About
Low-rank architectures have become increasingly important for efficient large language model (LLM) pre-training, providing substantial reductions in both parameter complexity and memory/computational demands. Despite these advantages, current low-rank methods face three critical shortcomings: (1) compromised model performance, (2) considerable computational overhead, and (3) limited activation memory savings. To address these limitations, we propose Cross-layer Low-Rank residual Network (CR-Net), an innovative parameter-efficient framework inspired by our discovery that inter-layer activation residuals possess low-rank properties. CR-Net implements this insight through a dual-path architecture that efficiently reconstructs layer activations by combining previous-layer outputs with their low-rank differences, thereby maintaining high-rank information with minimal parameters. We further develop a specialized activation recomputation strategy tailored for CR-Net that dramatically reduces memory requirements. Extensive pre-training experiments across model scales from 60M to 7B parameters demonstrate that CR-Net consistently outperforms state-of-the-art low-rank frameworks while requiring fewer computational resources and less memory.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Language Modeling | WikiText-2 (val) | Perplexity (BVS)20.66 | 70 | |
| Language Modeling | Qwen3 (val) | -- | 49 | |
| Language Modeling | C4 LLaMA-130M (val) | Perplexity23.74 | 40 | |
| Language Modeling | ArXiv (val) | Perplexity24.48 | 34 | |
| Language Modeling | C4 LLaMA-60M (val) | Perplexity32.76 | 25 | |
| Language Modeling | C4 LLaMA-350M (val) | Perplexity17.08 | 23 | |
| Language Modeling | C4-en LLaMA-1B, 13.1B tokens | Perplexity (PPL)14.05 | 11 | |
| Language Modeling | C4 en (val) | Perplexity14.79 | 6 | |
| Language Modeling | LLaMA-2 7B pre-training (val) | Validation Perplexity (40K steps)16.01 | 5 | |
| Language Modeling | LLaMA 1B pre-training 2 (val) | Perplexity15.22 | 5 |