xmc-aalto / cascadexml
Code for our paper CascadeXML: Rethinking Transformers for End-to-end Multi-resolution Training in Extreme Multi-label Classification
☆7Updated 11 months ago
Alternatives and similar repositories for cascadexml:
Users that are interested in cascadexml are comparing it to the libraries listed below
- LightXML: Transformer with dynamic negative sampling for High-Performance Extreme Multi-label Text Classification☆96Updated 2 years ago
- PECOS - Prediction for Enormous and Correlated Spaces☆523Updated 3 months ago
- Paper List for Contrastive Learning for Natural Language Processing☆548Updated last year
- Tools for multi-label classification problems.☆127Updated 3 months ago
- Implementation for "AttentionXML: Label Tree-based Attention-Aware Deep Model for High-Performance Extreme Multi-Label Text Classificatio…☆243Updated 2 years ago
- ☆12Updated last year
- [ACL 2022] LinkBERT: A Knowledgeable Language Model 😎 Pretrained with Document Links☆428Updated 2 years ago
- A curated list of awesome papers for Semantic Retrieval (TOIS Accepted: Semantic Models for the First-stage Retrieval: A Comprehensive Re…☆321Updated last year
- A Label Attention Model for ICD Coding from Clinical Text☆69Updated 2 years ago
- X-Transformer: Taming Pretrained Transformers for eXtreme Multi-label Text Classification☆137Updated 3 years ago
- Training & evaluation library for text-based neural re-ranking and dense retrieval models built with PyTorch☆261Updated 2 years ago
- auto icd coding with prompt☆47Updated 8 months ago
- ☆294Updated 2 years ago
- A huggingface transformers implementation of "Transformer Memory as a Differentiable Search Index"☆169Updated 2 years ago
- pytrec_eval is an Information Retrieval evaluation tool for Python, based on the popular trec_eval.☆296Updated last year
- Search Engines with Autoregressive Language models☆281Updated last year
- Provides a common interface to many IR ranking datasets.☆339Updated 2 weeks ago
- A curated list of graph data augmentation papers.☆304Updated 9 months ago
- Official implementation of GraphMask as presented in our paper Interpreting Graph Neural Networks for NLP With Differentiable Edge Maskin…☆43Updated 3 years ago
- TorchKGE: Knowledge Graph embedding in Python and PyTorch.☆382Updated 9 months ago
- [NAACL 2021] This is the code for our paper `Fine-Tuning Pre-trained Language Model with Weak Supervision: A Contrastive-Regularized Self…☆201Updated 2 years ago
- docTTTTTquery document expansion model☆360Updated last year
- ☆128Updated last year
- Open source code for paper "Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hypersphere" ICML 2…☆436Updated 2 years ago
- A novel embedding training algorithm leveraging ANN search and achieved SOTA retrieval on Trec DL 2019 and OpenQA benchmarks☆367Updated last year
- A simple toolkit to process TREC files in Python.☆167Updated 5 months ago
- Code for the paper "Taxonomy Completion via Triplet Matching Network" AAAI 2021☆28Updated 3 years ago
- GitHub Repository for KDD 2022 paper "Saliency-Regularized Deep Multi-Task Learning"☆12Updated last year
- The official repository for "Bridging the Gap Between Indexing and Retrieval for Differentiable Search Index with Query Generation", Shen…☆118Updated last year
- EMNLP 2021 - Pre-training architectures for dense retrieval☆244Updated 2 years ago