{"id":"https://openalex.org/W7157110526","doi":"https://doi.org/10.48550/arxiv.2604.24608","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-04-27","ids":{"openalex":"https://openalex.org/W7157110526","doi":"https://doi.org/10.48550/arxiv.2604.24608"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2604.24608","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.24608","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":null,"license_id":null,"version":null,"is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Preprint"},"type":"preprint","indexed_in":["datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://doi.org/10.48550/arxiv.2604.24608","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5134774319","display_name":"Yuxing Tian","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Tian, Yuxing","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5134764355","display_name":"Fengran Mo","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Mo, Fengran","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5134762045","display_name":"Zhiqi Huang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Huang, Zhiqi","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5134775530","display_name":"Weixu Zhang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhang, Weixu","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5134777942","display_name":"Jian-Yun Nie","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Nie, Jian-Yun","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":0,"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":null,"last_page":null},"is_retracted":false,"is_paratext":false,"is_xpac":false,"primary_topic":{"id":"https://openalex.org/T10286","display_name":"Information Retrieval and Search Behavior","score":0.24390000104904175,"subfield":{"id":"https://openalex.org/subfields/1710","display_name":"Information Systems"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},"topics":[{"id":"https://openalex.org/T10286","display_name":"Information Retrieval and Search Behavior","score":0.24390000104904175,"subfield":{"id":"https://openalex.org/subfields/1710","display_name":"Information Systems"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T10028","display_name":"Topic Modeling","score":0.16539999842643738,"subfield":{"id":"https://openalex.org/subfields/1702","display_name":"Artificial Intelligence"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T10664","display_name":"Sentiment Analysis and Opinion Mining","score":0.09889999777078629,"subfield":{"id":"https://openalex.org/subfields/1702","display_name":"Artificial Intelligence"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/embedding","display_name":"Embedding","score":0.6261000037193298},{"id":"https://openalex.org/keywords/relevance","display_name":"Relevance (law)","score":0.5841000080108643},{"id":"https://openalex.org/keywords/redundancy","display_name":"Redundancy (engineering)","score":0.5740000009536743},{"id":"https://openalex.org/keywords/aggregate","display_name":"Aggregate (composite)","score":0.5450000166893005},{"id":"https://openalex.org/keywords/ranking","display_name":"Ranking (information retrieval)","score":0.5321999788284302},{"id":"https://openalex.org/keywords/selection","display_name":"Selection (genetic algorithm)","score":0.4900999963283539},{"id":"https://openalex.org/keywords/heuristic","display_name":"Heuristic","score":0.4560000002384186},{"id":"https://openalex.org/keywords/language-model","display_name":"Language model","score":0.45010000467300415},{"id":"https://openalex.org/keywords/construct","display_name":"Construct (python library)","score":0.4431000053882599}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.8047000169754028},{"id":"https://openalex.org/C41608201","wikidata":"https://www.wikidata.org/wiki/Q980509","display_name":"Embedding","level":2,"score":0.6261000037193298},{"id":"https://openalex.org/C158154518","wikidata":"https://www.wikidata.org/wiki/Q7310970","display_name":"Relevance (law)","level":2,"score":0.5841000080108643},{"id":"https://openalex.org/C152124472","wikidata":"https://www.wikidata.org/wiki/Q1204361","display_name":"Redundancy (engineering)","level":2,"score":0.5740000009536743},{"id":"https://openalex.org/C4679612","wikidata":"https://www.wikidata.org/wiki/Q866298","display_name":"Aggregate (composite)","level":2,"score":0.5450000166893005},{"id":"https://openalex.org/C189430467","wikidata":"https://www.wikidata.org/wiki/Q7293293","display_name":"Ranking (information retrieval)","level":2,"score":0.5321999788284302},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.52920001745224},{"id":"https://openalex.org/C81917197","wikidata":"https://www.wikidata.org/wiki/Q628760","display_name":"Selection (genetic algorithm)","level":2,"score":0.4900999963283539},{"id":"https://openalex.org/C173801870","wikidata":"https://www.wikidata.org/wiki/Q201413","display_name":"Heuristic","level":2,"score":0.4560000002384186},{"id":"https://openalex.org/C137293760","wikidata":"https://www.wikidata.org/wiki/Q3621696","display_name":"Language model","level":2,"score":0.45010000467300415},{"id":"https://openalex.org/C2780801425","wikidata":"https://www.wikidata.org/wiki/Q5164392","display_name":"Construct (python library)","level":2,"score":0.4431000053882599},{"id":"https://openalex.org/C99016210","wikidata":"https://www.wikidata.org/wiki/Q5488129","display_name":"Query expansion","level":2,"score":0.4120999872684479},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.39399999380111694},{"id":"https://openalex.org/C204321447","wikidata":"https://www.wikidata.org/wiki/Q30642","display_name":"Natural language processing","level":1,"score":0.37119999527931213},{"id":"https://openalex.org/C127705205","wikidata":"https://www.wikidata.org/wiki/Q5748245","display_name":"Heuristics","level":2,"score":0.33649998903274536},{"id":"https://openalex.org/C51632099","wikidata":"https://www.wikidata.org/wiki/Q3985153","display_name":"Training set","level":2,"score":0.33559998869895935},{"id":"https://openalex.org/C86037889","wikidata":"https://www.wikidata.org/wiki/Q4330127","display_name":"Learning to rank","level":3,"score":0.32260000705718994},{"id":"https://openalex.org/C2775896111","wikidata":"https://www.wikidata.org/wiki/Q642560","display_name":"Router","level":2,"score":0.3160000145435333},{"id":"https://openalex.org/C124101348","wikidata":"https://www.wikidata.org/wiki/Q172491","display_name":"Data mining","level":1,"score":0.2935999929904938},{"id":"https://openalex.org/C2776145971","wikidata":"https://www.wikidata.org/wiki/Q30673951","display_name":"Labeled data","level":2,"score":0.28859999775886536},{"id":"https://openalex.org/C23123220","wikidata":"https://www.wikidata.org/wiki/Q816826","display_name":"Information retrieval","level":1,"score":0.2847000062465668},{"id":"https://openalex.org/C2780312720","wikidata":"https://www.wikidata.org/wiki/Q5689100","display_name":"Head (geology)","level":2,"score":0.28040000796318054},{"id":"https://openalex.org/C165064840","wikidata":"https://www.wikidata.org/wiki/Q1321061","display_name":"Matching (statistics)","level":2,"score":0.2799000144004822},{"id":"https://openalex.org/C75553542","wikidata":"https://www.wikidata.org/wiki/Q178161","display_name":"A priori and a posteriori","level":2,"score":0.2702000141143799},{"id":"https://openalex.org/C195324797","wikidata":"https://www.wikidata.org/wiki/Q33742","display_name":"Natural language","level":2,"score":0.2653999924659729},{"id":"https://openalex.org/C192028432","wikidata":"https://www.wikidata.org/wiki/Q845739","display_name":"Query language","level":2,"score":0.259799987077713},{"id":"https://openalex.org/C175154964","wikidata":"https://www.wikidata.org/wiki/Q380077","display_name":"Task analysis","level":3,"score":0.2554999887943268}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2604.24608","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.24608","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":null,"license_id":null,"version":null,"is_accepted":false,"is_published":null,"raw_source_name":null,"raw_type":"Preprint"}],"best_oa_location":{"id":"doi:10.48550/arxiv.2604.24608","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.24608","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":null,"license_id":null,"version":null,"is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Preprint"},"sustainable_development_goals":[],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"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-01T06:00:48.157686","created_date":"2026-04-29T00:00:00"}