Как распределить вычисления: большая модель и остановка раньше
при фиксированном бюджете вычислений выгоднее обучать большую модель на умеренном объёме данных и останавливаться до сходимости.
Зачем это нужно
Есть бюджет вычислений, и его нужно разделить между размером модели, числом шагов и объёмом данных. Статья отвечает, как это делать оптимально.
Как это работает
Большие модели эффективнее по примерам: им нужно меньше данных и шагов, чтобы достичь того же loss. Поэтому оптимальное обучение состоит в том, чтобы обучать очень большие модели на относительно небольшом объёме данных и останавливаться заметно раньше сходимости.
Пример
Из статьи: при фиксированном бюджете вычислений оптимально обучать модель крупнее, а не обучать меньшую модель дольше на большем числе данных. Конкретные таблицы распределения в этом посте не приводятся.
Что запомнить
- Большие модели требуют меньше примеров для того же loss.
- Оптимум: крупная модель, умеренные данные, остановка до сходимости.
- Раздел 6 статьи посвящён оптимальному распределению бюджета.
Где читать дальше
- Kaplan et al., arXiv:2001.08361, разделы 1.1 и 6