PulseAugur
EN
LIVE 13:53:35

New method boosts efficiency of neural network training algorithms

Researchers have developed a new method to reparametrize Shampoo and SOAP algorithms, improving their efficiency for training neural networks. This technique supports BFloat16 storage, which reduces memory usage, and mitigates performance degradation often associated with this storage format. By updating only a subspace of the basis vectors, the approach significantly cuts down computational overhead, making Shampoo-based methods more time- and memory-efficient, particularly for large preconditioning matrices. AI

IMPACT Enhances efficiency for neural network training, potentially enabling larger models or faster iteration cycles.

RANK_REASON The cluster contains a research paper detailing novel algorithmic improvements for neural network training. [lever_c_demoted from research: ic=1 ai=1.0]

Read on arXiv cs.LG →

AI-generated summary · Google Gemini · from 1 sources. How we write summaries →

New method boosts efficiency of neural network training algorithms

COVERAGE [1]

  1. arXiv cs.LG TIER_1 English(EN) · Alan Milligan, Zikun Xu, Simon Lacoste-Julien, Felix Dangel, Wu Lin ·

    Reparametrizing Shampoo and SOAP for Subspace Basis Updates and BFloat16 Storage

    arXiv:2605.26327v1 Announce Type: new Abstract: Shampoo-based methods, such as KL-Shampoo and SOAP, have demonstrated strong performance in training neural networks and rely on QR decomposition. Because existing QR implementations require single-precision (FP32) arithmetic and re…