M+Adam

M+Adam 优化器改进低精度大语言模型训练

研究人员推出 M+Adam，这是一种新颖的优化方法，旨在提高低精度权重下大语言模型训练的准确性。标准优化器在低精度下可能会遇到困难，导致进展停滞，尤其是在权重幅度较大时。M+Adam 通过结合加法和乘法更新类型来解决这个问题，利用它们互补的优势来确保在各种权重幅度和符号变化下都能取得持续的进展。使用 LLaMA 类型模型和不同精度级别（包括 BF16、FP8 和 FP4）进行的实验表明，M+Adam 持续提高了低精度训练的成果。