pc-layer-polynomial-weight-preconditioning-for-improving-llm-pre-training-26751bd8·1 events·first seen Aliases: PC Layer: Polynomial Weight Preconditioning for Improving LLM Pre-Training
Researchers propose a PC (preconditioning) layer that applies polynomial preconditioning to reshape the singular-value spectrum of weight matrices during LLM training, improving conditioning stability. The preconditioned weights merge back into the original architecture at inference time with no overhead. Experiments on Llama-1B pre-training show advantages over standard transformers for both AdamW and Muon optimizers, with theoretical convergence guarantees for deep linear networks.