뉴런 약화(Weakening Neurons): 트랜스포머의 입력-출력 기능성 분석
요약
본 논문은 LLM에서 GLU 기반 뉴런의 입력 및 출력 동작을 분석하고, 코사인 유사도를 이용해 '약화 뉴런(weakening neuron)'이라는 개념을 제안합니다. 이 연구를 통해 약화 뉴런이 주로 후기 레이어에 나타나며 모델 동작에 큰 영향을 미치고, 게이트 값이 음수일 때도 강력한 기능성을 가짐을 밝혀냈습니다.
핵심 포인트
- 약화 뉴런은 코사인 유사도를 이용해 정의되며, LLM의 입력-출력 기능을 분석합니다.
- 약화 뉴런은 주로 후기 레이어에 나타나고, 강화 뉴런은 초기-중기 레이어에서 발견됩니다.
- 비록 수가 적지만 약화 뉴런은 모델 동작에 큰 영향을 미치는 중요한 역할을 합니다.
우리는 대규모 언어 모델(LLMs)에서 GLU 기반 뉴런의 학습된 입력-출력 동작을 분석합니다. 우리는 간단한 분석 방법을 제안합니다: 각 뉴런에 대해, 그 입력(읽기) 가중치 벡터와 출력(쓰기) 가중치 벡터 간의 코사인 유사도를 계산합니다. 이 방식에서, 강한 음의 코사인 유사도는 해당 뉴런이 잔여 스트림(residual stream)에서 감지하는 방향을 약화시킨다는 것을 의미하므로, 이를 약화 뉴런(weakening neuron)이라고 부릅니다. 이는 우리가 여러 가지 새로운 통찰력을 얻게 해줍니다. 첫째, 우리는 아홉 개의 다른 LLM들이 유사한 패턴을 보인다는 것을 보여줍니다: 약화 뉴런은 주로 후기 레이어(late layers)에서 나타나는 반면, 그 대응물인 (조건부) 강화 뉴런(conditional strengthening neurons)은 초기-중기 레이어(early-middle layers)에서 빈번하게 나타납니다. 둘째, 우리는 약화 뉴런이 놀라운 동작을 보인다는 것을 발견합니다: 비록 수가 적지만, 자주 활성화되며 모델 동작에 큰 영향을 미칩니다. 셋째, 게이트 값이 음수일 때 약화 뉴런은 모델 출력에 강한 영향을 미치는데, 이는 음의 게이트 값이 기능성을 인코딩할 것으로 예상되지 않기 때문에 놀라운 결과입니다.
AI 자동 생성 콘텐츠
본 콘텐츠는 arXiv cs.CL의 원문을 AI가 자동으로 요약·번역·분석한 것입니다. 원 저작권은 원저작자에게 있으며, 정확한 내용은 반드시 원문을 확인해 주세요.
원문 바로가기