29 citations · 130 across the 17 of their papers we have counts for
Showing cs.LGShow all
2 papers · 1 filter
cs.LG2026
FlashTrie: A GPU-Accelerated Constrained Beam Search for Generative Retrieval
Dakshitha Anandakumar, Anurag Mukkara, Wenxiang Hu +5
Constrained decoding is essential in generative retrieval, where document identifiers generated directly from a query must exactly match a predefined library of valid IDs. At scale…
cs.LG2022★ 1 cited
NGAME: Negative Mining-aware Mini-batching for Extreme Classification
Kunal Dahiya, Nilesh Gupta, Deepak Saini +16
Extreme Classification (XC) seeks to tag data points with the most relevant subset of labels from an extremely large label set. Performing deep XC with dense, learnt representation…