ArvinZhuang / DSI-transformers
A huggingface transformers implementation of "Transformer Memory as a Differentiable Search Index"
☆164Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for DSI-transformers
- The official repository for "Bridging the Gap Between Indexing and Retrieval for Differentiable Search Index with Query Generation", Shen…☆114Updated last year
- ☆154Updated last year
- A curated list of awesome papers related to generative retrieval models.☆45Updated last year
- SIGIR'21: Optimizing DR with hard negatives and achieving SOTA first-stage retrieval performance on TREC DL Track.☆126Updated 2 years ago
- EMNLP 2021 - Pre-training architectures for dense retrieval☆243Updated 2 years ago
- ☆202Updated last year
- WSDM'22 Best Paper: Learning Discrete Representations via Constrained Clustering for Effective and Efficient Dense Retrieval☆118Updated 3 months ago
- Learning to Tokenize for Generative Retrieval (NeurIPS 2023)☆42Updated 2 weeks ago
- An Open-Source Package for Information Retrieval☆153Updated last month
- Code, datasets, and checkpoints for the paper "Improving Passage Retrieval with Zero-Shot Question Generation (EMNLP 2022)"☆96Updated last year
- Codebase for RetroMAE and beyond.☆240Updated 5 months ago
- CIKM 2022: CorpusBrain: Pre-train a Generative Retrieval Model for Knowledge-Intensive Language Tasks☆32Updated 2 years ago
- https://acl2023-retrieval-lm.github.io/☆154Updated last year
- Build Text Rerankers with Deep Language Models☆251Updated 9 months ago
- ☆66Updated 2 years ago
- [EMNLP 2022] Training Language Models with Memory Augmentation https://arxiv.org/abs/2205.12674☆194Updated last year
- Zero-shot Document Ranking with Large Language Models.☆96Updated 4 months ago
- Scalable training for dense retrieval models.☆271Updated last year
- Code and models for the paper "Questions Are All You Need to Train a Dense Passage Retriever (TACL 2023)"☆60Updated last year
- ☆45Updated 3 months ago
- Code for Editing Factual Knowledge in Language Models☆135Updated 2 years ago
- ☆265Updated 11 months ago
- Train Dense Passage Retriever (DPR) with a single GPU☆127Updated 3 years ago
- Code and resources for papers "Generation-Augmented Retrieval for Open-Domain Question Answering" and "Reader-Guided Passage Reranking fo…☆73Updated 2 years ago
- Tevatron - A flexible toolkit for neural retrieval research and development.☆524Updated last month
- Code for the paper "Open Domain Question Answering with A Unified Knowledge Interface" (ACL 2022)☆57Updated last year
- Trans-Encoder: Unsupervised sentence-pair modelling through self- and mutual-distillations☆133Updated 5 months ago
- Paper collections of retrieval-based (augmented) language model.☆230Updated 5 months ago
- CIKM'21: JPQ substantially improves the efficiency of Dense Retrieval with 30x compression ratio, 10x CPU speedup and 2x GPU speedup.☆50Updated 2 years ago
- An original implementation of "MetaICL Learning to Learn In Context" by Sewon Min, Mike Lewis, Luke Zettlemoyer and Hannaneh Hajishirzi☆254Updated last year