Vers un élagage de tokens sans coût dans les modèles de récupération à interaction tardive

Yuxuan Zong, Benjamin Piwowarski


Abstract
Les modèles de RI neuronaux à interaction tardive comme ColBERT offrent un compromis compétitif entre efficacité et efficience sur de nombreuses bases de référence. Cependant, ils nécessitent un espace mémoire considérable pour stocker les représentations contextuelles de tous les tokens des documents. Certains travaux ont proposé d’utiliser soit des heuristiques, soit des techniques basées sur les statistiques pour élaguer des tokens dans chaque document. Cependant, cela ne garantit pas que les tokens supprimés n’aient aucun impact sur le score de récupération. Notre travail utilise une approche méthodique pour définir comment élaguer des tokens sans affecter le score entre un document et une question. Nous introduisons trois coûts de régularisation, qui induisent une solution avec des taux d’élagage élevés, ainsi que deux stratégies d’élagage. Nous les étudions expérimentalement (en domaine interne et externe), démontrant que nous pouvons préserver les performances de ColBERT tout en n’utilisant que 30% des tokens.
Anthology ID:
2025.jeptalnrecital-coria.25
Volume:
Actes de la 20e Conférence en Recherche d’Information et Applications (CORIA)
Month:
6
Year:
2025
Address:
Marseille, France
Editors:
Frédéric Bechet, Adrian-Gabriel Chifu, Karen Pinel-sauvagnat, Benoit Favre, Eliot Maes, Diana Nurbakova
Venue:
JEP/TALN/RECITAL
SIG:
Publisher:
ATALA \\& ARIA
Note:
Pages:
294–309
Language:
French
URL:
https://aclanthology.org/2025.jeptalnrecital-coria.25/
DOI:
Bibkey:
Cite (ACL):
Yuxuan Zong and Benjamin Piwowarski. 2025. Vers un élagage de tokens sans coût dans les modèles de récupération à interaction tardive. In Actes de la 20e Conférence en Recherche d’Information et Applications (CORIA), pages 294–309, Marseille, France. ATALA \\& ARIA.
Cite (Informal):
Vers un élagage de tokens sans coût dans les modèles de récupération à interaction tardive (Zong & Piwowarski, JEP/TALN/RECITAL 2025)
Copy Citation:
PDF:
https://aclanthology.org/2025.jeptalnrecital-coria.25.pdf