Interactive Visualization for GPT-2 Embeddings

Limp-Contest-7309 · reddit · 2026-07-19

A developer created an interactive map tool for exploring the word embedding space of GPT-2-small. - **Technical Implementation**: Extracted 32,070 letter tokens and used the t-SNE algorithm to reduce the dimensionality of the compressed embedding table representations, without performing a forward pass or providing context. - **Interactive Experience**: Supports pinch-to-zoom on mobile. Users can click on any token to view its nearest neighbor connections or jump to it via the search box. The edges in the graph are constructed based on the minimum spanning tree of this space, representing true nearest affinities.

Related event: Interactive Tool Visualizes GPT-2 Token Embeddings(4 posts)→

Original post →

More from Research

Research channel →