r/MachineLearning · · 1 min read

I posted my embedding migration project here, it got a lot of attention, so I added the features you guys said were missing [R]

Mirrored from r/MachineLearning for archival readability. Support the source by reading on the original site.

A little while ago, I posted about embedflow

https://github.com/arnsri33/embedflow

and it got a lot of attention. The basic idea was pretty simple

keep your existing embedding index for candidate retrieval → rerank "k" candidates with the new embedding model → progressively materialize the new embeddings instead of requiring a full re-embed before you can use the new model.

The project got way more attention than I expected, and a bunch of comments were variations of

-cool idea, but how do I actually know what K to use?
-what happens with a cold cache?
-how do I test this safely on production traffic?
-can this work with my vector DB?

So I spent the last week turning it from a migration engine into something that can actually support a real migration workflow

Additions:

1)FAISS, Qdrant, pgvector, Pinecone, Milvus, and Weaviate support.

2) Migration planner -> embedflow plan looks at the source index, source/target model contracts, probe queries, + other evidence, and reccomends a candidate K + migration plan

3)Shadow mode -> you can run the new embedding path against real traffic while the old/source retrieval remains authoritative. The shadow path can crash, time out, or get saturated and it still doesn't change or block the response users receive.

4)Traffic aware prewarming 5) Persistent Target Cache + background materialization 6) extensive reports

You can use it with python with just

pip install embedflow

submitted by /u/Potential_Low_1183
[link] [comments]

Discussion (0)

Sign in to join the discussion. Free account, 30 seconds — email code or GitHub.

Sign in →

No comments yet. Sign in and be the first to say something.

More from r/MachineLearning