CoT-MAE MS-Marco Passage Reranker

caskcsg
Similitud de oraciones

CoT-MAE es una arquitectura de preentrenamiento de Mask Auto-Encoder basada en transformers, diseñada para la Recuperación Densa de Pasajes. CoT-MAE MS-Marco Passage Reranker es un reranker entrenado con negativos duros extraídos por el recuperador CoT-MAE utilizando el kit de herramientas Tevatron. Los detalles se pueden encontrar en nuestro artículo y códigos.

Como usar

Primero recuperamos utilizando CoT-MAE MS-Marco Passage Retriever (nombrado cotmae_base_msmarco_retriever), luego usamos el reranker para re-calificar los 200 mejores resultados de recuperación. Las actuaciones son las siguientes.

MRR @10: 0.43884 Recuerdo@1: 0.304871 Recuerdo@50: 0.903582 Recuerdo@200: 0.956734 Consultas calificadas: 6980

Funcionalidades

Clasificación de textos
Extracción de características
Embeddings de textos-inferencia
Puntos de inferencia
MRR @10 de 0.43884
Recuerdo@1 de 0.304871
Recuerdo@50 de 0.903582
Recuerdo@200 de 0.956734

Casos de uso

Recuperación densa de pasajes
Recalificación de resultados de búsqueda