← Search

Nishant Yadav

6 accepted papers

2024

Adaptive Retrieval and Scalable Indexing for k-NN Search with Cross-Encoders

ICLR 2024poster

Cross-encoder (CE) models which compute similarity by jointly encoding a query-item pair perform better than using dot-product with embedding-based models (dual-encoders) at estimating query-item relevance. Existing approaches perform k-NN search with cross-encoders by approximating the CE similarit…

Cited by 0SourcePDFScholar
2023

Efficient k-NN Search with Cross-Encoders using Adaptive Multi-Round CUR Decomposition

EMNLP 2023short findings

Cross-encoder models, which jointly encode and score a query-item pair, are prohibitively expensive for direct k-nearest neighbor (k-NN) search. Consequently, k-NN search typically employs a fast approximate retrieval (e.g. using BM25 or dual-encoder vectors), followed by reranking with a cross-enco…

Cited by 0SourcecodeScholar
2022

Efficient Nearest Neighbor Search for Cross-Encoder Models using Matrix Factorization

EMNLP 2022main

Efficient k-nearest neighbor search is a fundamental task, foundational for many problems in NLP. When the similarity is measured by dot-product between dual-encoder vectors or L2-distance, there already exist many scalable and efficient search methods. But not so when similarity is measured by more…

2022

Interactive Correlation Clustering with Existential Cluster Constraints

ICML 2022spotlight

We consider the problem of clustering with user feedback. Existing methods express constraints about the input data points, most commonly through must-link and cannot-link constraints on data point pairs. In this paper, we introduce existential cluster constraints: a new form of feedback where users…

2021

Clustering-based Inference for Biomedical Entity Linking

NAACL 2021long

Due to large number of entities in biomedical knowledge bases, only a small fraction of entities have corresponding labelled training data. This necessitates entity linking models which are able to link mentions of unseen entities using learned representations of entities. Previous approaches link e…

Cited by 70SourcePDFScholar
2019

Supervised Hierarchical Clustering with Exponential Linkage

ICML 2019oral

In supervised clustering, standard techniques for learning a pairwise dissimilarity function often suffer from a discrepancy between the training and clustering objectives, leading to poor cluster quality. Rectifying this discrepancy necessitates matching the procedure for training the dissimilarity…