Sign-SGD via Parameter-Free Optimization: sign-ΠΎΠΏΡΠΈΠΌΠΈΠ·Π°ΡΠΈΡ Π±Π΅Π· ΡΡΡΠ½ΠΎΠ³ΠΎ ΠΏΠΎΠ΄Π±ΠΎΡΠ° ΡΠ°Π·ΠΌΠ΅ΡΠ° ΡΠ°Π³Π°
Π‘Π΅Π³ΠΎΠ΄Π½Ρ ΡΠ°Π·Π±ΠΈΡΠ°Π΅ΠΌ ΡΡΠ°ΡΡΡ, ΠΏΡΠΎΡΠ΅Π΄ΡΡΡ Π½Π° ICLR-26 ΠΈ ΠΏΠΎΠ΄Π³ΠΎΡΠΎΠ²Π»Π΅Π½Π½ΡΡ ΡΠΎΠ²ΠΌΠ΅ΡΡΠ½ΠΎ Ρ ΠΊΠΎΠΌΠ°Π΄Π½ΠΎΠΉ
https://research.yandex.com/ (ΠΈΡΡΠ»Π΅Π΄ΠΎΠ²Π°ΡΠ΅Π»ΠΈ ΠΏΠΈΡΡΡ ΠΏΡΠΎ ΠΏΡΠΈΠ½ΡΡΡΠ΅ ΡΡΠ°ΡΡΠΈ Π² ΠΊΠ°Π½Π°Π»Π΅
https://t.me/MLunderhood).π‘ Π ΡΡΠΌ ΡΠ°Π±ΠΎΡΠ°?
Sign-SGD Π»ΡΠ±ΡΡ Π·Π° ΠΏΡΠΎΡΡΠΎΡΡ ΠΈ Β«Π΄Π΅ΡΠ΅Π²ΡΡΒ» ΠΏΠ°ΠΌΡΡΡ: Π²ΠΌΠ΅ΡΡΠΎ ΠΏΠΎΠ»Π½ΠΎΡΠ΅Π½Π½ΡΡ
Π³ΡΠ°Π΄ΠΈΠ΅Π½ΡΠΎΠ² ΠΌΠΎΠΆΠ½ΠΎ ΠΏΠ΅ΡΠ΅Π΄Π°Π²Π°ΡΡ/Ρ
ΡΠ°Π½ΠΈΡΡ ΡΠΎΠ»ΡΠΊΠΎ ΠΈΡ
Π·Π½Π°ΠΊΠΈ β ΡΠ΄ΠΎΠ±Π½ΠΎ ΠΈ Π΄Π»Ρ ΡΠ°ΡΠΏΡΠ΅Π΄Π΅Π»Π΅Π½Π½ΠΎΠ³ΠΎ ΠΎΠ±ΡΡΠ΅Π½ΠΈΡ, ΠΈ Π½Π° ΠΎΠ΄Π½ΠΎΠΌ ΡΡΡΡΠΎΠΉΡΡΠ²Π΅.
ΠΠΎ Ρ Π½Π΅Π³ΠΎ, ΠΊΠ°ΠΊ ΠΈ Ρ Π΄ΡΡΠ³ΠΈΡ
ΠΊΠ»Π°ΡΡΠΈΡΠ΅ΡΠΊΠΈΡ
ΠΎΠΏΡΠΈΠΌΠΈΠ·Π°ΡΠΎΡΠΎΠ², Π΅ΡΡΡ Π±ΠΎΠ»ΡΡΠ°Ρ ΠΏΡΠ°ΠΊΡΠΈΡΠ΅ΡΠΊΠ°Ρ Π±ΠΎΠ»Ρ: ΡΡΡΠ΅ΠΊΡΠΈΠ²Π½ΡΠΉ ΡΠ°Π³ (stepsize) Π·Π°ΡΠ°Π½Π΅Π΅ Π½Π΅ Π²ΡΠ±ΡΠ°ΡΡ, ΠΏΠΎΡΠΎΠΌΡ ΡΡΠΎ ΠΎΠ½ Π·Π°Π²ΠΈΡΠΈΡ ΠΎΡ Π½Π΅ΠΈΠ·Π²Π΅ΡΡΠ½ΡΡ
ΡΠ²ΠΎΠΉΡΡΠ² ΠΊΠ°ΠΆΠ΄ΠΎΠΉ ΠΎΡΠ΄Π΅Π»ΡΠ½ΠΎΠΉ Π·Π°Π΄Π°ΡΠΈ β ΠΈ Π² ΠΈΡΠΎΠ³Π΅ Π²ΡΡ ΡΠΏΠΈΡΠ°Π΅ΡΡΡ Π² ΡΡΠ½ΠΈΠ½Π³.
β οΈ ΠΠΎΡΠ΅ΠΌΡ ΡΡΠΎ ΠΏΡΠΎΠ±Π»Π΅ΠΌΠ°?
Π Π±ΠΎΠ»ΡΡΠΈΡ
ΠΌΠΎΠ΄Π΅Π»ΡΡ
, Π½Π°ΠΏΡΠΈΠΌΠ΅Ρ LLM, ΠΏΠΎΠ΄Π±ΠΎΡ learning rate β ΡΡΠΎ Π½Π΅ Β«ΠΏΠ°ΡΠ° Π·Π°ΠΏΡΡΠΊΠΎΠ²Β», Π° ΠΏΠΎΠ»Π½ΠΎΡΠ΅Π½Π½ΡΠΉ grid search + ΠΏΠΎΠΈΡΠΊ ΡΠ°ΡΠΏΠΈΡΠ°Π½ΠΈΡ (schedule), ΡΡΠΎ:
β
ΡΡΠ°ΡΠΈΡ GPU-ΡΠ°ΡΡ ΠΈ Π΄Π΅Π½ΡΠ³ΠΈ;
β
Π΄Π΅Π»Π°Π΅Ρ ΠΌΠ΅ΡΠΎΠ΄ ΡΡΠΆΠ΅Π»ΠΎ ΠΏΠ΅ΡΠ΅Π½ΠΎΡΠΈΠΌΡΠΌ ΠΌΠ΅ΠΆΠ΄Ρ Π·Π°Π΄Π°ΡΠ°ΠΌΠΈ/Π΄Π°ΡΠ°ΡΠ΅ΡΠ°ΠΌΠΈ.
π― Π§ΡΠΎ ΠΌΡ ΠΏΡΠ΅Π΄Π»Π°Π³Π°Π΅ΠΌ?
ΠΡ ΡΡΡΠΎΠΈΠΌ parameter-free Sign-SGD: Π°Π»Π³ΠΎΡΠΈΡΠΌ ΡΠ°ΠΌ ΠΏΠΎΠ΄ΡΡΡΠ°ΠΈΠ²Π°Π΅Ρ ΡΠ°Π³ ΠΎΠ±ΡΡΠ΅Π½ΠΈΡ Π½Π° ΠΊΠ°ΠΆΠ΄ΠΎΠΉ ΠΈΡΠ΅ΡΠ°ΡΠΈΠΈ, Π±Π΅Π· ΡΠ΅ΡΡΠ°ΡΡΠΎΠ² ΠΈ Π±Π΅Π· Π΄ΠΎΠΏΠΎΠ»Π½ΠΈΡΠ΅Π»ΡΠ½ΠΎΠ³ΠΎ ΠΏΠΎΠΈΡΠΊΠ° Π³ΠΈΠΏΠ΅ΡΠΏΠ°ΡΠ°ΠΌΠ΅ΡΡΠΎΠ².
ΠΠ»ΡΡΠ΅Π²Π°Ρ ΠΈΠ΄Π΅Ρ β ΠΎΡΠ΅Π½ΠΈΠ²Π°ΡΡ Π³Π»ΠΎΠ±Π°Π»ΡΠ½ΡΠ΅ ΠΊΠΎΠ½ΡΡΠ°Π½ΡΡ Π·Π°Π΄Π°ΡΠΈ, ΠΎΡ ΠΊΠΎΡΠΎΡΡΡ
Π·Π°Π²ΠΈΡΠΈΡ ΠΎΠΏΡΠΈΠΌΠ°Π»ΡΠ½ΡΠΉ ΡΠ°Π³ (Π² ΡΠ°ΡΡΠ½ΠΎΡΡΠΈ, Π³Π»Π°Π΄ΠΊΠΎΡΡΡ ΠΈ Π½Π°ΡΠ°Π»ΡΠ½ΠΎΠ΅ ΠΏΡΠΈΠ±Π»ΠΈΠΆΠ΅Π½ΠΈΠ΅ Ρ ΡΠΎΡΠΊΠΈ Π·ΡΠ΅Π½ΠΈΡ ΡΡΠ½ΠΊΡΠΈΠΈ), ΠΏΠΎ ΡΠ΅ΠΊΡΡΠ΅ΠΉ ΠΈΠ½ΡΠΎΡΠΌΠ°ΡΠΈΠΈ ΠΎ Π³ΡΠ°Π΄ΠΈΠ΅Π½ΡΠ°Ρ
.
π§© Π§ΡΠΎ Π²Π½ΡΡΡΠΈ (Π΅ΡΠ»ΠΈ ΠΊΠΎΡΠΎΡΠΊΠΎ, Π½ΠΎ ΠΏΠΎ Π΄Π΅Π»Ρ):
β
ALIAS: Π°Π»Π³ΠΎΡΠΈΡΠΌ Π½Π° ΠΎΡΠ½ΠΎΠ²Π΅ Sign-SGD Ρ Π°Π²ΡΠΎΠΌΠ°ΡΠΈΡΠ΅ΡΠΊΠΈΠΌ Π²ΡΠ±ΠΎΡΠΎΠΌ ΡΠ°Π³Π°.
β
Π‘ΡΠ΅Π½Π°ΡΠΈΠΈ: ΡΠ°Π·ΠΎΠ±ΡΠ°Π½Ρ Π΄Π΅ΡΠ΅ΡΠΌΠΈΠ½ΠΈΡΠΎΠ²Π°Π½Π½ΡΠΉ, ΡΡΠΎΡ
Π°ΡΡΠΈΡΠ΅ΡΠΊΠΈΠΉ ΠΈ ΡΠ°ΡΠΏΡΠ΅Π΄Π΅Π»ΡΠ½Π½ΡΠΉ Π°Π»Π³ΠΎΡΠΈΡΠΌΡ Π΄Π»Ρ ΠΎΠ±ΡΡΠ΅Π½ΠΈΡ.
β
Momentum-Π²Π°ΡΠΈΠ°Π½Ρ (Π² Π΄ΡΡ
Π΅ Adam): ΡΡΠΎΠ±Ρ ΠΏΠΎΠ΄ΡΡΠ½ΡΡΡ ΠΏΡΠ°ΠΊΡΠΈΡΠ΅ΡΠΊΡΡ ΡΡΡΠ΅ΠΊΡΠΈΠ²Π½ΠΎΡΡΡ, ΡΠΎΡ
ΡΠ°Π½ΠΈΠ² parameter-free ΠΈΠ΄Π΅Ρ.
β
Memory-efficient Π²Π΅ΡΡΠΈΡ: Π²Π°ΡΠΈΠ°Π½Ρ, ΠΊΠΎΡΠΎΡΡΠΉ Ρ
ΡΠ°Π½ΠΈΡ ΡΠΎΠ»ΡΠΊΠΎ Π·Π½Π°ΠΊΠΈ ΠΏΡΠΎΡΠ»ΡΡ
Π³ΡΠ°Π΄ΠΈΠ΅Π½ΡΠΎΠ² Π΄Π»Ρ ΠΏΠΎΠ΄Π±ΠΎΡΠ° ΡΠ°Π³Π° ΠΈ ΡΠΎΡ
ΡΠ°Π½ΡΠ΅Ρ Π°Π΄Π°ΠΏΡΠΈΠ²Π½ΠΎΡΡΡ (Π²Π°ΠΆΠ½ΠΎ Π΄Π»Ρ ΠΏΠ°ΠΌΡΡΠΈ).
β
ΠΠΎΠ½ΡΡ: Π² ΡΠΊΡΠΏΠ΅ΡΠΈΠΌΠ΅Π½ΡΠ°Ρ
Π²ΠΈΠ΄Π½ΠΎ, ΡΡΠΎ Ρ Π½Π°ΡΠΈΠΌ ΠΏΠΎΠ΄Π±ΠΎΡΠΎΠΌ ΡΠ°Π³Π° Π°Π²ΡΠΎΠΌΠ°ΡΠΈΡΠ΅ΡΠΊΠΈ Π²ΠΎΠ·Π½ΠΈΠΊΠ°Π΅Ρ ΠΊΠΎΡΠΈΠ½ΡΡΠ½ΠΎΠ΅ ΡΠ°ΡΠΏΠΈΡΠ°Π½ΠΈΠ΅, Π±Π΅Π· ΡΠ²Π½ΠΎΠ³ΠΎ cosine scheduler.
π ΠΠΊΡΠΏΠ΅ΡΠΈΠΌΠ΅Π½ΡΡ
ΠΡΠΎΠ²Π΅ΡΠΈΠ»ΠΈ Π½Π° Π·Π°Π΄Π°ΡΠ°Ρ
, Π³Π΄Π΅ ΡΡΠ½ΠΈΠ½Π³ ΠΎΡΠΎΠ±Π΅Π½Π½ΠΎ Π΄ΠΎΡΠΎΠ³:
β
pre-training LLaMA (130M ΠΈ 350M ΠΏΠ°ΡΠ°ΠΌΠ΅ΡΡΠΎΠ²);
β
fine-tuning Swin Transformer (28M ΠΏΠ°ΡΠ°ΠΌΠ΅ΡΡΠΎΠ²).Β
β
Π΄ΡΡΠ³ΠΈΠ΅ Π·Π°Π΄Π°ΡΠΈ ΠΈΠ· Π±Π΅Π½ΡΠΌΠ°ΡΠΊΠ° AlgoPerf.
π₯ ΠΡΠΎΠ³
ΠΠ°ΡΠΈ parameter-free Π²Π°ΡΠΈΠ°Π½ΡΡ ΠΏΠΎ ΠΊΠ°ΡΠ΅ΡΡΠ²Ρ ΠΎΠ±ΡΡΠ΅Π½ΠΈΡ ΡΠΎΠΏΠΎΡΡΠ°Π²ΠΈΠΌΡ Ρ βΠ·Π°ΡΡΠ½Π΅Π½ΡΠΌΠΈβ Sign-SGD ΠΈ AdamW (Π³Π΄Π΅ lr ΠΏΠΎΠ΄Π±ΠΈΡΠ°Π»ΠΈ ΡΠ΅ΡΠ΅Π· grid search + cosine schedule), ΠΈ ΠΏΡΠΈ ΡΡΠΎΠΌ ΡΠ±ΠΈΡΠ°ΡΡ Π΄ΠΎΠΏΠΎΠ»Π½ΠΈΡΠ΅Π»ΡΠ½ΡΠ΅ ΡΠ°ΡΡ
ΠΎΠ΄Ρ Π½Π° ΠΏΠΎΠ΄Π±ΠΎΡ.
ΠΡΠΎ Π΄Π°ΡΡ ΠΏΡΠΈΠΌΠ΅ΡΠ½ΠΎ ~1.5Γ ΡΡΠΊΠΎΡΠ΅Π½ΠΈΠ΅ end-to-end ΠΏΠΎ ΡΡΠ°Π²Π½Π΅Π½ΠΈΡ Ρ Π·Π°ΠΏΡΡΠΊΠ°ΠΌΠΈ, Π³Π΄Π΅ lr ΠΏΠΎΠ΄Π±ΠΈΡΠ°ΡΡ ΡΠ΅ΡΠΊΠΎΠΉ.
β¨ ΠΠΎΡΠ΅ΠΌΡ ΡΡΠΎ Π²Π°ΠΆΠ½ΠΎ?
β
ΠΠ΅Π½ΡΡΠ΅ ΡΡΡΠ½ΠΎΠ³ΠΎ ΡΡΠ½ΠΈΠ½Π³Π° β Π±ΡΡΡΡΠ΅Π΅ ΠΎΠ±ΡΡΠ΅Π½ΠΈΠ΅ Π½Π° Π»ΡΠ±ΡΡ
Π·Π°Π΄Π°ΡΠ°Ρ
β
Sign-ΠΎΠΏΡΠΈΠΌΠΈΠ·Π°ΡΠΈΡ ΠΎΡΡΠ°Π΅ΡΡΡ βΠ΄Π΅ΡΠ΅Π²ΠΎΠΉβ ΠΏΠΎ ΠΏΠ°ΠΌΡΡΠΈ β ΠΌΠ΅Π½ΡΡΠ΅ Π·Π°ΡΡΠ°Ρ Π½Π° GPU-ΡΠ΅ΡΡΡΡΡ
β
Π Π°Π±ΠΎΡΠ°Π΅Ρ Π² ΡΠ΅Π°Π»ΠΈΡΡΠΈΡΠ½ΡΡ
ΡΠ΅ΠΆΠΈΠΌΠ°Ρ
β ΡΡΠΎΡ
Π°ΡΡΠΈΠΊΠ°, ΡΠ°ΡΠΏΡΠ΅Π΄Π΅Π»Π΅Π½ΠΊΠ°
π
https://openreview.net/pdf?id=yDLD3D95w3π»
https://github.com/brain-lab-research/ALIAS