Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Conferences, Workshops, Symposiums, and Seminars
  4. Local to global: Learning dynamics and effect of initialization for transformers
 
conference poster

Local to global: Learning dynamics and effect of initialization for transformers

Makkuva, Ashok Vardhan
•
Bondaschi, Marco  
•
Ekbote, Chanakya
Show more
September 25, 2024
Advances in Neural Information Processing Systems 37 (NeurIPS 2024)
38th Annual Conference on Neural Information Processing Systems

In recent years, transformer-based models have revolutionized deep learning, particularly in sequence modeling. To better understand this phenomenon, there is a growing interest in using Markov input processes to study transformers. However, our current understanding in this regard remains limited with many fundamental questions about how transformers learn Markov chains still unanswered. In this paper, we address this by focusing on first-order Markov chains and single-layer transformers, providing a comprehensive characterization of the learning dynamics in this context. Specifically, we prove that transformer parameters trained on next-token prediction loss can either converge to global or local minima, contingent on the initialization and the Markovian data properties, and we characterize the precise conditions under which this occurs. To the best of our knowledge, this is the first result of its kind highlighting the role of initialization. We further demonstrate that our theoretical findings are corroborated by empirical evidence. Based on these insights, we provide guidelines for the initialization of single-layer transformers and demonstrate their effectiveness. Finally, we outline several open problems in this arena. Code is available at:\url {https://github. com/Bond1995/Markov}.

  • Files
  • Details
  • Metrics
Loading...
Thumbnail Image
Name

2420_Local_to_Global_Learning_ (1).pdf

Type

Main Document

Version

http://purl.org/coar/version/c_970fb48d4fbd8a85

Access type

openaccess

License Condition

CC BY

Size

5.57 MB

Format

Adobe PDF

Checksum (MD5)

33a0f3481eb97c3381ec1274d7a97ca7

Loading...
Thumbnail Image
Name

2420_Local_to_Global_Learning__Supplementary Material.zip

Type

Supplementary Material/information

Version

http://purl.org/coar/version/c_be7fb7dd8ff6fe43

Access type

openaccess

License Condition

CC BY

Size

11.85 MB

Format

ZIP

Checksum (MD5)

cefe7e87c896284bfb1d07bb83d26baf

Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés