←back to thread

76 points unixpickle | 1 comments | | HN request time: 0.209s | source

I made this website with my wife in mind; it makes it possible to browse for similar fashion products over many different retailers at once.

The backend is written in Swift, and is hosted on a single Mac Mini. It performs nearest neighbors on the GPU over ~3M product images.

No vector DB, just pure matrix multiplications. Since we aren't just doing approximate nearest neighbors but rather sorting all results by distance, it's possible to show different "variety" levels by changing the stride over the sorted search results.

Nearest neighbors are computed in a latent vector space. The model which produces the vectors is also something I trained in pure Swift.

The underlying data is about 2TB scraped from https://www.shopltk.com/.

All the code is at https://github.com/unixpickle/LTKlassifier

1. binarymax ◴[] No.43375116[source]
This is really cool. You might be interested in this talk which shows you can incrementally add some preference vectors as a way to improve the recommendation.

https://haystackconf.com/us2023/talk-20/