Transformer 해석 용이성을 위한 SoLU 활성화 함수
요약
MLP 뉴런의 해석 난이도를 낮추기 위해 Softmax Linear Units (SoLU) 를 제안합니다. 성능 저하 없이 이해하기 쉬운 기능에 반응하는 뉴런 비율을 높였습니다. 1 레이어 모델에서는 end-to-end 규칙과 명확히 대응하며, 대규모 모델에서는 토큰 분해 및 언어별 단어 인식 구조를 드러냅니다.
핵심 포인트
- SoLU 활성화 함수는 성능 저하 없이 해석 가능한 기능에 반응하는 뉴런 비율을 증가시킵니다.
- 1 레이어 Transformer 에서 SoLU 는 MLP 뉴런이 end-to-end 규칙 (예: base64 텍스트 인식) 과 명확히 대응하게 합니다.
- 대규모 모델에서 SoLU 는 초기 레이어가 'de-tokenize'하여 다중 토큰 단어와 특정 언어 단어를 인식하는 구조를 드러냅니다.
Transformer MLP neurons are challenging to understand.
We find that using a different activation function (Softmax Linear Units or SoLU) increases the fraction of neurons that appear to respond to understandable features without any performance penalty.
https://transformer-circuits.pub/2022/solu/index.html
SoLU is a preliminary experiment in whether architectural changes can make mechanistic interpretability easier. Our hope is that the right architecture might make it easier for us (and others) to reverse engineer neural networks, without loss of performance.
In 1-layer transformers, SoLU causes many MLP neurons to cleanly correspond to end-to-end rules. For example, this base64 neuron which fires on base64 text and increases the probability the next token is also in base64.
In larger models, it reveals an interesting structure across layers:
Early layers seem to "de-tokenize", recognizing multi-token words (eg. "bank|ing" and "Not|withstandng") or words in the context of a specific language (eg. neurons for "die" in Afrikaans/German/Dutch).
AI 자동 생성 콘텐츠
본 콘텐츠는 X @danielaamodei (Anthropic 사장)의 원문을 AI가 자동으로 요약·번역·분석한 것입니다. 원 저작권은 원저작자에게 있으며, 정확한 내용은 반드시 원문을 확인해 주세요.
원문 바로가기