Добавьте top-k + обратную связь об ошибках и уберите внешний Nesterov из DiloCo (так что это ~внешний простой SGD). Я все еще довольно удивлен, что это работает / может заменить Nesterov :0 Классная работа!
Amir Sarfi
Amir Sarfi22 авг., 10:13
Представляем SparseLoCo: метод эффективной коммуникации для предобучения LLM. Кратко: Мы используем разрежение Top-k + обратную связь об ошибках с редкими внешними шагами DiLoCo — передавая только 1–3% градиентов с 2-битной квантизацией — превосходя DiLoCo и DeMo. 1/N, ArXiv: Github:
5K