연구교류 세미나Understanding and Acceleration of Grokking Phenomena in Learning Arithmetic Operations via Kolmogorov-Arnold Representation
작성일2024-08-21작성자 박세영
접수기간2024-08-09 09:24 ~ 2024-08-21 10:30
행사기간2024-08-21 00:00
발표자박예찬 박사(고등과학원)
## 통신상의 문제로 유튜브 스트리밍은 진행되지 않습니다. 양해 부탁드립니다.
1. 일시: 2024년 8월 21일(수), 10:30-12:30
2. 장소: 판교 테크노밸리 산업수학혁신센터 세미나실
- 경기 성남시 수정구 대왕판교로 815, 기업지원허브 231호 국가수리과학연구소
- 무료주차는 2시간 지원됩니다.
3. 발표자: 박예찬 박사(고등과학원)
4. 주요내용: Understanding and Acceleration of Grokking Phenomena in Learning Arithmetic Operations via Kolmogorov-Arnold Representation
We propose novel methodologies aimed at accelerating the grokking phenomenon, which refers to the rapid increment of test accuracy after a long period of overfitting as reported in~\cite{power2022grokking}. Focusing on the grokking phenomenon that arises in learning arithmetic binary operations via the transformer model, we begin with a discussion on data augmentation in the case of commutative binary operations. To further accelerate, we elucidate arithmetic operations through the lens of the Kolmogorov-Arnold (KA) representation theorem, revealing its correspondence to the transformer architecture: embedding, decoder block, and classifier. Observing the shared structure between KA representations associated with binary operations, we suggest various transfer learning mechanisms that expedite grokking. This interpretation is substantiated through a series of rigorous experiments. In addition, our approach is successful in learning two nonstandard arithmetic tasks: composition of operations and a system of equations. Furthermore, we reveal that the model is capable of learning arithmetic operations using a limited number of tokens under embedding transfer, which is supported by a set of experiments as well.