{"id":"https://openalex.org/W7168410131","doi":"https://doi.org/10.1145/3805712.3809945","title":"Learning to Route Queries to Heads for Attention-based Re-ranking with Large Language Models","display_name":"Learning to Route Queries to Heads for Attention-based Re-ranking with Large Language Models","publication_year":2026,"publication_date":"2026-07-15","ids":{"openalex":"https://openalex.org/W7168410131","doi":"https://doi.org/10.1145/3805712.3809945"},"language":null,"primary_location":{"id":"doi:10.1145/3805712.3809945","is_oa":true,"landing_page_url":"https://doi.org/10.1145/3805712.3809945","pdf_url":null,"source":null,"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Proceedings of the 49th International ACM SIGIR Conference on Research and Development in Information Retrieval","raw_type":"proceedings-article"},"type":"conference-paper","indexed_in":["crossref"],"open_access":{"is_oa":true,"oa_status":"gold","oa_url":"https://doi.org/10.1145/3805712.3809945","any_repository_has_fulltext":null},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5140799509","display_name":"Yuxing Tian","orcid":"https://orcid.org/0009-0008-0313-5089"},"institutions":[{"id":"https://openalex.org/I70931966","display_name":"Universit\u00e9 de Montr\u00e9al","ror":"https://ror.org/0161xgx34","country_code":"CA","type":"education","lineage":["https://openalex.org/I70931966"]}],"countries":["CA"],"is_corresponding":false,"raw_author_name":"Yuxing Tian","raw_affiliation_strings":["Universit\u00e9 de Montr\u00e9al, Montreal, QC, Canada"],"raw_orcid":"https://orcid.org/0009-0008-0313-5089","affiliations":[{"raw_affiliation_string":"Universit\u00e9 de Montr\u00e9al, Montreal, QC, Canada","institution_ids":["https://openalex.org/I70931966"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5140877277","display_name":"Fengran Mo","orcid":"https://orcid.org/0000-0002-0838-6994"},"institutions":[{"id":"https://openalex.org/I70931966","display_name":"Universit\u00e9 de Montr\u00e9al","ror":"https://ror.org/0161xgx34","country_code":"CA","type":"education","lineage":["https://openalex.org/I70931966"]}],"countries":["CA"],"is_corresponding":false,"raw_author_name":"Fengran Mo","raw_affiliation_strings":["Universit\u00e9 de Montr\u00e9al, Montreal, QC, Canada"],"raw_orcid":"https://orcid.org/0000-0002-0838-6994","affiliations":[{"raw_affiliation_string":"Universit\u00e9 de Montr\u00e9al, Montreal, QC, Canada","institution_ids":["https://openalex.org/I70931966"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5140949848","display_name":"Zhiqi Huang","orcid":"https://orcid.org/0000-0002-2939-1936"},"institutions":[{"id":"https://openalex.org/I4210119705","display_name":"Boston Engineering (United States)","ror":"https://ror.org/02cxh9170","country_code":"US","type":"company","lineage":["https://openalex.org/I4210119705"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Zhiqi Huang","raw_affiliation_strings":["CapitalOne, Boston, MA, USA"],"raw_orcid":"https://orcid.org/0000-0002-2939-1936","affiliations":[{"raw_affiliation_string":"CapitalOne, Boston, MA, USA","institution_ids":["https://openalex.org/I4210119705"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5140814624","display_name":"Weixu Zhang","orcid":"https://orcid.org/0009-0003-2000-7891"},"institutions":[{"id":"https://openalex.org/I4210164802","display_name":"Mila - Quebec Artificial Intelligence Institute","ror":"https://ror.org/05c22rx21","country_code":"CA","type":"facility","lineage":["https://openalex.org/I4210164802"]}],"countries":["CA"],"is_corresponding":false,"raw_author_name":"Weixu Zhang","raw_affiliation_strings":["McGill University, Montreal, QC, Canada and Mila - Quebec Artificial Intelligence Institute, Montreal, QC, Canada"],"raw_orcid":"https://orcid.org/0009-0003-2000-7891","affiliations":[{"raw_affiliation_string":"McGill University, Montreal, QC, Canada and Mila - Quebec Artificial Intelligence Institute, Montreal, QC, Canada","institution_ids":["https://openalex.org/I4210164802"]}]},{"author_position":"last","author":{"id":"https://openalex.org/A5140880418","display_name":"Jian-Yun Nie","orcid":"https://orcid.org/0000-0003-1556-3335"},"institutions":[{"id":"https://openalex.org/I70931966","display_name":"Universit\u00e9 de Montr\u00e9al","ror":"https://ror.org/0161xgx34","country_code":"CA","type":"education","lineage":["https://openalex.org/I70931966"]}],"countries":["CA"],"is_corresponding":false,"raw_author_name":"Jian-Yun Nie","raw_affiliation_strings":["Universit\u00e9 de Montr\u00e9al, Montreal, QC, Canada"],"raw_orcid":"https://orcid.org/0000-0003-1556-3335","affiliations":[{"raw_affiliation_string":"Universit\u00e9 de Montr\u00e9al, Montreal, QC, Canada","institution_ids":["https://openalex.org/I70931966"]}]}],"institutions":[],"countries_distinct_count":2,"institutions_distinct_count":3,"corresponding_author_ids":[],"corresponding_institution_ids":[],"apc_list":null,"apc_paid":null,"fwci":null,"has_fulltext":false,"cited_by_count":0,"citation_normalized_percentile":null,"cited_by_percentile_year":null,"biblio":{"volume":null,"issue":null,"first_page":"4181","last_page":"4186"},"is_retracted":false,"is_paratext":false,"is_xpac":false,"primary_topic":null,"topics":[],"keywords":[{"id":"https://openalex.org/keywords/feature","display_name":"Feature (linguistics)","score":0.3176000118255615},{"id":"https://openalex.org/keywords/natural-language","display_name":"Natural language","score":0.3009999990463257},{"id":"https://openalex.org/keywords/matching","display_name":"Matching (statistics)","score":0.27619999647140503},{"id":"https://openalex.org/keywords/language-model","display_name":"Language model","score":0.26910001039505005},{"id":"https://openalex.org/keywords/semantics","display_name":"Semantics (computer science)","score":0.2651999890804291}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.6969000101089478},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.4399000108242035},{"id":"https://openalex.org/C204321447","wikidata":"https://www.wikidata.org/wiki/Q30642","display_name":"Natural language processing","level":1,"score":0.4260999858379364},{"id":"https://openalex.org/C2776401178","wikidata":"https://www.wikidata.org/wiki/Q12050496","display_name":"Feature (linguistics)","level":2,"score":0.3176000118255615},{"id":"https://openalex.org/C195324797","wikidata":"https://www.wikidata.org/wiki/Q33742","display_name":"Natural language","level":2,"score":0.3009999990463257},{"id":"https://openalex.org/C165064840","wikidata":"https://www.wikidata.org/wiki/Q1321061","display_name":"Matching (statistics)","level":2,"score":0.27619999647140503},{"id":"https://openalex.org/C137293760","wikidata":"https://www.wikidata.org/wiki/Q3621696","display_name":"Language model","level":2,"score":0.26910001039505005},{"id":"https://openalex.org/C184337299","wikidata":"https://www.wikidata.org/wiki/Q1437428","display_name":"Semantics (computer science)","level":2,"score":0.2651999890804291},{"id":"https://openalex.org/C116834253","wikidata":"https://www.wikidata.org/wiki/Q2039217","display_name":"Identification (biology)","level":2,"score":0.260699987411499},{"id":"https://openalex.org/C177264268","wikidata":"https://www.wikidata.org/wiki/Q1514741","display_name":"Set (abstract data type)","level":2,"score":0.2581999897956848}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.1145/3805712.3809945","is_oa":true,"landing_page_url":"https://doi.org/10.1145/3805712.3809945","pdf_url":null,"source":null,"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Proceedings of the 49th International ACM SIGIR Conference on Research and Development in Information Retrieval","raw_type":"proceedings-article"}],"best_oa_location":{"id":"doi:10.1145/3805712.3809945","is_oa":true,"landing_page_url":"https://doi.org/10.1145/3805712.3809945","pdf_url":null,"source":null,"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Proceedings of the 49th International ACM SIGIR Conference on Research and Development in Information Retrieval","raw_type":"proceedings-article"},"sustainable_development_goals":[],"awards":[],"funders":[],"has_content":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":25,"referenced_works":["https://openalex.org/W2470818894","https://openalex.org/W2740492458","https://openalex.org/W2798658104","https://openalex.org/W2946794439","https://openalex.org/W2970641574","https://openalex.org/W3021397474","https://openalex.org/W3099700870","https://openalex.org/W3099977667","https://openalex.org/W4385569686","https://openalex.org/W4385569963","https://openalex.org/W4385573057","https://openalex.org/W4385573970","https://openalex.org/W4389523765","https://openalex.org/W4396819963","https://openalex.org/W4400531953","https://openalex.org/W4401043313","https://openalex.org/W4409362556","https://openalex.org/W4410502462","https://openalex.org/W4412377183","https://openalex.org/W4412945661","https://openalex.org/W4413243547","https://openalex.org/W4416035758","https://openalex.org/W7140078186","https://openalex.org/W7166828044","https://openalex.org/W7168424163"],"related_works":[],"abstract_inverted_index":{"Large":[0],"Language":[1],"Models":[2],"(LLMs)":[3],"have":[4],"recently":[5],"been":[6],"explored":[7],"as":[8],"fine-grained":[9],"zero-shot":[10],"re-rankers":[11],"by":[12,38,108],"leveraging":[13],"attention":[14,25,110],"signals":[15,26,111],"to":[16,65,98],"estimate":[17],"document":[18],"relevance.":[19],"However,":[20],"existing":[21],"methods":[22],"either":[23],"aggregate":[24],"across":[27,52],"all":[28],"heads":[29,49,60],"or":[30,54,67],"rely":[31],"on":[32,160,169],"a":[33,76,90,137,165],"statically":[34],"selected":[35],"subset":[36],"identified":[37],"heuristic":[39],"rules.":[40],"This":[41],"solution":[42],"can":[43,50,61,94],"be":[44],"suboptimal":[45],"because":[46],"the":[47,149,153,161,178],"informative":[48],"vary":[51],"queries":[53],"domains.":[55],"Moreover,":[56],"naively":[57],"combining":[58],"multiple":[59,173],"degrade":[62],"performance":[63],"due":[64],"redundancy":[66],"conflicting":[68],"ranking":[69],"signals.":[70],"In":[71],"this":[72],"paper,":[73],"we":[74,88,122],"propose":[75],"query-dependent":[77],"head":[78,101,135],"selection":[79],"method,":[80],"RouteHead,":[81],"for":[82],"attention-based":[83],"re-ranking":[84],"with":[85,136,164],"LLMs.":[86],"Specifically,":[87],"learn":[89],"lightweight":[91],"router":[92,132],"that":[93,177],"map":[95],"each":[96,134,142],"query":[97,143],"an":[99,128,145],"optimal":[100,118],"set,":[102],"and":[103,140,172],"relevance":[104],"scores":[105],"are":[106,120],"computed":[107],"aggregating":[109],"only":[112],"from":[113,148],"these":[114],"heads.":[115],"Since":[116],"query-to-head":[117],"labels":[119,126,163],"unavailable,":[121],"first":[123],"construct":[124],"pseudo":[125,162],"via":[127],"offline":[129],"search.":[130],"The":[131],"represents":[133,141],"learnable":[138],"embedding":[139,146],"using":[144],"extracted":[147],"hidden":[150],"states":[151],"of":[152],"frozen":[154],"LLM.":[155],"Then":[156],"it":[157],"is":[158],"trained":[159],"sparsity":[166],"regularizer.":[167],"Experiments":[168],"diverse":[170],"benchmarks":[171],"LLM":[174],"backbones":[175],"show":[176],"proposed":[179],"method":[180],"consistently":[181],"outperforms":[182],"strong":[183],"baselines.":[184]},"counts_by_year":[],"updated_date":"2026-07-17T05:52:16.776730","created_date":"2026-07-16T00:00:00"}