Parameter and Memory Efficient Pretraining via Low-rank Riemannian Optimization
Pretraining large language models often requires significant computational resources and memory due to their vast parameter amount. An effective approach to enhance parameter efficiency in both training and inference is to parameterize each full-size weight as the product of two trainable low-rank f…