recommenders icon indicating copy to clipboard operation
recommenders copied to clipboard

Has anyone implemented retrieval model in ways other than 2 tower model?

Open amit-timalsina opened this issue 3 years ago • 4 comments

I have seen most of the people are implementing just the 2 tower architecture for the retrieval model. I wanted to try other architecture. So, if you have implemented or have an idea about ways to do that, please help me.

Also, Like for this implementation, how can we use TensorFlow recommenders? image

amit-timalsina avatar Jan 17 '22 05:01 amit-timalsina

hey @amit-timalsina, the picture above describes the serving case for a two tower retrieval model.

patrickorlando avatar Jan 17 '22 06:01 patrickorlando

hey @amit-timalsina, the picture above describes the serving case for a two-tower retrieval model.

How, can you please elaborate?

amit-timalsina avatar Jan 17 '22 06:01 amit-timalsina

@amit-timalsina I guess, that model is implemented in softmax model section of this colab

AzizIlyosov avatar Jan 17 '22 07:01 AzizIlyosov

The softmax model you picture here is a degenerate case of a two-tower model. The softmax is exactly equivalent to having a candidate tower with a single embedding layer, with one embedding vector per candidate.

maciejkula avatar Jun 05 '22 19:06 maciejkula