A practical review of mechanistic interpretability for transformer-based language models

Motivation

• Known: 트랜스포머 기반 언어모델이 NLP 분야에서 뛰어난 성능을 보이고 있으나, 안전성과 신뢰성에 대한 우려가 증대되고 있다. 기존 해석가능성 연구는 모델의 행동을 설명하는 데 제한적이었다.
• Gap: MI 분야의 빠른 성장에도 불구하고 초보자를 위한 포괄적 가이드가 부족하며, 다양한 기법과 평가 방법, 그리고 실제 응용 사례를 체계적으로 정리한 자료가 없었다. MI의 확장성과 일반화 가능성에 대한 우려도 존재한다.
• Why: 언어모델의 안전성, 신뢰성, 강건성을 보장하기 위해서는 내부 계산 메커니즘을 정확히 이해해야 하며, 이는 AI 안전과 모델 개선에 직결되는 중요한 문제이다.
• Approach: 작업 중심 분류체계(task-centric taxonomy)를 통해 기능(Feature) 연구, 회로(Circuit) 연구, 범용성(Universality) 연구로 구분하고, 각 카테고리별 기법, 평가 방법, 주요 발견을 체계적으로 정리한다.

Achievement

• MI 기본 개념 정립: 기능, 회로, 범용성이라는 MI의 세 가지 기본 연구 대상을 명확히 정의
• 초보자 로드맵 제시: 특정 기능 연구부터 개방형 기능 연구, 회로 연구, 범용성 연구로 이어지는 단계별 워크플로우 제공
• 기법 비교 분석: 어휘 투영법(Vocabulary Projection), 개입 기반 방법(Intervention-based Methods), 희소 자동인코더(Sparse Autoencoder, SAE), 탐사(Probing), 시각화 등 5가지 주요 기법의 장단점 상세 분석
• 평가 방법론 체계화: 기능 연구와 회로 연구를 위한 평가 벤치마크 정리 및 제시
• 실제 응용 사례: 모델 강화, AI 안전, 인문맥 학습(In-Context Learning), 추론, 지식 메커니즘 등 다양한 응용 분야의 구체적 사례 제시
• 학습 동역학 분석: 모델 학습 중 위상 변화와 사후 학습(post-training) 효과에 대한 MI 관점의 분석

How

Figure 4: Logit lens implementation at (1) RS, (2) attention head, and (3) FF sublayer.

• 트랜스포머 아키텍처의 배경 지식 제공 (뉘런, 주의 헤드, 피드포워드 층)
• 어휘 투영법: 뉘런/헤드 활성화를 어휘 공간에 투영하여 해석 (로짓 렌즈, 주의 패턴 분석)
• 개입 기반 방법: 노이징(Noising), 제거(Ablation), 패치(Patching) 등을 통한 인과관계 추론
• 희소 자동인코더: 활성화를 희소한 기저(sparse basis)로 분해하여 해석 가능한 기능 추출
• 탐사(Probing): 선형/비선형 분류기를 이용한 숨겨진 표현의 정보 검출
• 시각화: 주의 패턴, 활성화 분포 등을 시각적으로 표현
• 사례 연구: 프로빙을 이용한 목표 기능 연구와 SAE를 이용한 개방형 기능 연구 사례 제시

Originality

• 작업 중심 분류체계는 기존의 기법 중심 관점에서 벗어나 문제 정의 단계부터 실제 구현까지의 전체 프로세스를 다룸
• 초보자 로드맵은 이론과 실제 연구의 간극을 메우기 위해 체계적인 워크플로우와 구체적 사례를 결합
• 다양한 MI 기법 간 비교 분석을 통해 각 접근법의 적용 상황과 한계를 명확히 함
• MI의 실제 응용을 단순한 나열이 아닌 구체적 성과와 영향으로 정리

Limitation & Further Study

• 확장성 제약: 더 큰 모델과 복잡한 작업으로의 MI 기법 확장 가능성에 대한 검증 부족
• 일반화 문제: 특정 모델이나 작업에서 발견된 기능과 회로의 다른 모델/작업으로의 전이 가능성 미검증
• 평가 표준화 부재: MI 기법의 평가를 위한 통일된 벤치마크와 메트릭이 아직 확립되지 않음
• 자동화 부재: 가설 생성과 해석 프로세스의 자동화 부족으로 인한 수동 작업 의존
• 후속 연구 방향: (1) 더 직관적이고 실용적인 응용으로의 확대, (2) 인간-AI 협업을 중심으로 한 새로운 패러다임 개발, (3) MI 기법의 계산 효율성 개선

Evaluation

총평: 이 논문은 빠르게 성장하는 MI 분야에서 초보자부터 경험자까지 모두를 위한 실용적이고 포괄적인 가이드를 제공하며, 작업 중심의 분류체계와 구체적 워크플로우를 통해 해석가능성 연구의 새로운 표준을 제시한다. 현장 적용을 위한 실제 고려사항과 미래 방향을 함께 제시한 점에서 높은 가치를 지닌다.