TL;DR
이 게시물은 GPT-2-small의 WTE에서 추출한 32,070개 알파벳 토큰 벡터를 대상으로 한 대화형 임베딩 시각화에 관한 내용이다. 임베딩 테이블을 압축한 표현을 t-SNE로 2차원에 투영하고 투영 공간에서 토큰 간 거리를 기준으로 최소 신장 트리를 구성하여 모든 간선이 실제 최근접 관계를 반영하도록 레이아웃을 만들었다. 모바일에서 핀치 투 줌과 토큰 탭으로 최근접 연결을 확인하고 이웃 토큰을 따라 그래프를 걷거나 검색 상자로 특정 토큰 위치로 바로 이동할 수 있는 상호작용을 제공한다. 이러한 구성은 임베딩의 국소적 유사 관계와 연결성 패턴을 직관적으로 탐색할 수 있게 설계되어 있다.
섹션별 상세
용어 해설
- WTE
- — WTE는 모델이 각 토큰을 수치 벡터로 변환하기 위해 유지하는 임베딩 행렬로서 입력 토큰 인덱스를 행렬의 행으로 매핑해 고정 차원 임베딩을 생성한다. 이 게시물 맥락에서는 GPT-2-small의 WTE에서 추출한 토큰 벡터들이 시각화 대상이며 임베딩을 그대로 사용해 유사도 계산과 차원 축소 전처리의 입력으로 활용된다. WTE는 모델 추론을 위한 초기 표현을 제공하므로 임베딩의 분포가 토큰 간 의미적 관계를 반영한다.
- t-SNE
- — t-SNE는 고차원 벡터의 국소적 유사 관계를 보존해 저차원에 투영하는 비선형 차원 축소 알고리즘으로, 고차원에서 가까운 샘플들은 저차원에서도 가깝게 배치되도록 확률적 이웃 구조를 보존한다. 이 게시물에서는 임베딩 테이블을 압축한 표현에 t-SNE를 적용해 2차원 좌표를 얻고 시각적 레이아웃의 기초로 사용했다. t-SNE는 국소 패턴 탐색에는 유리하지만 결과가 하이퍼파라미터에 민감하고 전역 구조 해석에는 주의가 필요하다.
- Minimum Spanning Tree
- — 최소 신장 트리는 가중치가 부여된 그래프에서 모든 정점을 연결하면서 간선 가중치의 합을 최소화하는 부분 그래프로, 유사도 기반 그래프에서는 거리 또는 역유사도를 가중치로 삼아 인접 관계를 연결하는 데 사용된다. 본 시각화에서는 t-SNE 투영된 좌표 공간에서 토큰들 간의 근접성에 기반해 최소 신장 트리를 구성하여 각 간선이 실제 근접 관계를 반영하도록 레이아웃의 연결성을 보장했다. MST는 고차원 유사도 구조를 단순화해 그래프 기반 탐색에서 경로와 연결성 정보를 제공한다.
- Embedding Table
- — 임베딩 테이블은 모델 내부에서 토큰, 서브워드, 또는 단어 조각을 고정 차원의 실수 벡터로 매핑하는 행렬이며 각 행은 특정 토큰의 분산 표현을 담는다. 이 게시물에서는 임베딩 테이블 전체 또는 일부를 압축 표현으로 변환해 시각화에 사용했고, 임베딩 간의 유사도 계산과 차원 축소의 입력으로 직접 활용했다. 임베딩 테이블은 단어 의미의 분포적 관계를 반영하므로 시각화는 해당 분포 구조를 탐색하는 수단이 된다.
AI 요약 · 북마크 · 개인 피드 설정 — 무료
출처 · 인용 안내
인용 시 "요약 출처: AI Trends (aitrends.kr)"를 표기하고, 사실 확인은 원문 보기 기준으로 진행해 주세요. 자세한 기준은 운영 정책을 참고해 주세요.