search

jingtaozhan / JPQ

CIKM'21: JPQ substantially improves the efficiency of Dense Retrieval with 30x compression ratio, 10x CPU speedup and 2x GPU speedup.
MIT License
50 stars 11 forks source link

PQ Centroid Optimization implement #6

Open 592375753 opened 2 years ago

592375753 commented 2 years ago

hi, jingtaozhan,I am glad to read your paper.However,I don't find the code of PQ Centroid Optimization in the paper.Maybe I misunderstood your paper or code. image

jingtaozhan commented 2 years ago

Hi, Danfeng. This section illustrates how PQ centroids are updated with gradient descent at each training step. In our implementation, this step is automatically done by the optimizer and we do not need to explicitly write the code.