논리적 사고

같은 토큰 ID는 언제나 같은 임베딩 행을 읽을까요?

하나의 고정된 임베딩 표에서는 같은 정수 ID가 같은 행을 고르지만, 뒤의 문맥 연산까지 같은 표현을 보장하지는 않습니다.

토큰 ID 2, 1, 2가 임베딩 표의 2번, 1번, 2번 행을 골라 첫째와 셋째 결과가 같은 벡터가 되는 조회.
토큰 ID 2, 1, 2가 임베딩 표의 2번, 1번, 2번 행을 골라 첫째와 셋째 결과가 같은 벡터가 되는 조회.
그림 크게 열기 ↗SVG 저장

그림에서 무엇을 볼까요?

토큰 ID 2, 1, 2가 임베딩 표의 2번, 1번, 2번 행을 골라 첫째와 셋째 결과가 같은 벡터가 되는 조회.

PyTorch의 nn.Embedding은 정수 인덱스로 학습 가능한 표의 행을 고르는 조회입니다. 표를 0:[0.1,0.0], 1:[0.0,0.8], 2:[0.6,0.2]로 정하고 ID [2,1,2]를 넣으면 결과는 [[0.6,0.2],[0.0,0.8],[0.6,0.2]]입니다.

생각해 볼 질문

내가 비교하는 벡터는 임베딩 표에서 막 꺼낸 값인가요, 문맥 층을 통과한 값인가요?

← 논리적 사고 그림 더 보기