Inside the LLM: Visualizing the Embeddings Layer of Mistral-7B and Gemma-2B

Рет қаралды 5,323

3 ай бұрын

We look deep into the AI and look at how the embeddings layer of a Large Language Model such as Mistral-7B and Gemma-2B actually works.
You will learn how tokens and embeddings work and even extract out and load the embeddings layer from Gemma and Mistral into your own simple model, which we will use to visualize the model
You will see how an AI clusters terms together and how it can cluster similar words, build connections which cover not just similar words but also grouping of concepts such as colors, hotel chains, programming terms.
If you really want to understand how an LLM's works or even build your own LLM then starting with the first layer of a Generative AI model is the best place to start.
Github
-----------
github.com/chrishayuk/embeddings

Пікірлер: 32

@chrishayuk 3 ай бұрын

this is the github repo: github.com/chrishayuk/embeddings

@guaranamedia 2 күн бұрын

Excellent explanation. Thanks for making these examples.

@NERDDISCO 3 ай бұрын

This came to the absolute right time! Thank you very much! I was just trying to understand this. Now I know how it works ❤

@chrishayuk 3 ай бұрын

Glad it was helpful!

@rajneesh31 10 күн бұрын

Damn, thank you KZfaq for recommending this channel. @chrishayuk is a gun. Thanks Chris

@chrishayuk 3 күн бұрын

Very kind, glad you like the channel

@sumandawnmobile 3 ай бұрын

Its an great video to understand the internals via the visualization. Thanks Chris.

@scitechtalktv9742 2 ай бұрын

Fantastic video ! I am wondering: I think it would also be very interesting to also be able have a visualization of not only the static embeddings you already did, but also a visualization of the so-called contextualized embeddings in a later layer of the model! These are the embeddings that are exposed to the attention mechanism. That why they are also called dynamic embeddings. It adds another layer of abstraction, but are better embeddings because they are able to distinguish between homonyms: words that are the same but have completely other meanings if used in another context. A good example is the word “bank”, that has several different meanings when used in another context (for example financial institution or river bank and several other meanings! ). As a consequence the word “bank” will be represented by several different vectors in embedding space, depending on the context it is used in! This technique is called Word Sense Disambiguation (WSD). Would it be possible to visualize that too? I am curious….

@chrishayuk 2 ай бұрын

yep, you got what i'm doing... i'm literally walking the stack

@chrishayuk 2 ай бұрын

so those videos will be coming

@scitechtalktv9742 2 ай бұрын

@@chrishayukFantastic ! Those embeddings are crucially important for the workings of Large Language Models !

@johntdavies 3 ай бұрын

Great insight, thanks for posting this. It would be interesting to show how a fine-tuned model differs in similarities and "vocabulary". I'm also curious on the effects of quantisation, i.e. Q4, Q6, Q8, fp16 etc. on the internal "workings" of the LLM. Thanks again.

@chrishayuk 3 ай бұрын

It’s almost like you’re reading my roadmap

@kenchang3456 3 ай бұрын

Thanks the visualization really helped me.

@chrishayuk 3 ай бұрын

so glad, seeing it at a lower level really demystifies what's going on

@andypai 2 ай бұрын

Thank you! Great video!

@chrishayuk 23 күн бұрын

thank you, glad it was useful

@khalilbenzineb 3 ай бұрын

I was playing a bit with finetuning to force an output schema for some 7B Models, but lately I discovered schema grammar, which is a way to dynamically play with the EOS tokens, by limiting them to a specific set of tokens, to generate the output you want, This is very stable and way efficient for many cases that we may think it requires finetuning, For me it felt like a new dimension to get the model intentions inline, I loved the unique and efficient way you create your videos, So I wanted to ask you if possible to create a video for us about this, I feel it's very important

@chrishayuk 3 ай бұрын

that's a good shout

@khalilbenzineb 3 ай бұрын

Thx@@chrishayuk