hirooms2/ir

ConTextual Mask Auto-Encoder for Dense Passage Retrieval

Information Retrieval Researches

Codes and models for our information retrieval research papers.

Knowledge Computing and Service Group, Institute of Information Engineering, Chinese Academy of Sciences.

Releases

CoT-MAE: ConTextual Mask Auto-Encoder for Dense Passage Retrieval. CoT-MAE is a transformers based Mask Auto-Encoder pre-training architecture designed for Dense Passage Retrieval.

CoT-MAE-qc: Query-as-context Pre-training for Dense Passage Retrieval. A simple yet effective pre-training scheme for single vector Dense Passage Retrieval.