stanford-futuredata / ColBERT

ColBERT: state-of-the-art neural search (SIGIR'20, TACL'21, NeurIPS'21, NAACL'22, CIKM'22, ACL'23, EMNLP'23)
MIT License
2.67k stars 355 forks source link

Add support for (optional) hierarchical pooling #347

Closed bclavie closed 3 weeks ago

bclavie commented 1 month ago

@okhat 👀 It works pretty well! Blog post is coming in a few days, and future refinement in a bit longer.

Thanks @NohTow for helping with testing/code clean-up from my prototype to workable version(s)!