Re:Show HN: Wordllama – Things you can do with the token emb...
by lennxa on Sunday September 15, 2024 @02:47AM
(#41545720)
looks like it's the size of the model itself, more lightweight and faster. mini-lm is 80mb while the smallest one here is 16mb.
[ Parent
| on HN ]
Re:Show HN: Wordllama – Things you can do with the token emb...
by authorfly on Sunday September 15, 2024 @03:07AM
(#41545797)
Mini-lm isn't optimized to be as small as possible though, and is kind of dated. It was trained on a tiny amount of similarity pairs compared to what we have available today.
As of the last time I did it in 2022, Mini-lm can be distilled to 40mb with only limited loss in accuracy, so can paraphrase-MiniLM-L3-v1 (down to 21mb), by reducing the dimensions by half or more and projecting a custom matrix optimization(optionally, including domain specific or more recent training pairs). I imagine today you could get it down to 32mb (= project to ~156 dim) without accuracy loss.