Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Boao Kong, Junzhu Liang, Yuxi Liu, Renjia Deng, Kun Yuan• 2025

Related benchmarks

TaskDatasetResultRank
Language ModelingWikiText-2 (val)
Perplexity (BVS)20.66
70
Language ModelingQwen3 (val)--
49
Language ModelingC4 LLaMA-130M (val)
Perplexity23.74
40
Language ModelingArXiv (val)
Perplexity24.48
34
Language ModelingC4 LLaMA-60M (val)
Perplexity32.76
25
Language ModelingC4 LLaMA-350M (val)
Perplexity17.08
23
Language ModelingC4-en LLaMA-1B, 13.1B tokens
Perplexity (PPL)14.05
11
Language ModelingC4 en (val)
Perplexity14.79
6
Language ModelingLLaMA-2 7B pre-training (val)
Validation Perplexity (40K steps)16.01
5
Language ModelingLLaMA 1B pre-training 2 (val)
Perplexity15.22
5
Showing 10 of 11 rows

Other info

Follow for update