{"id":"https://openalex.org/W7127602484","doi":"https://doi.org/10.48550/arxiv.2602.03306","title":"Learning to Select: Query-Aware Adaptive Dimension Selection for Dense Retrieval","display_name":"Learning to Select: Query-Aware Adaptive Dimension Selection for Dense Retrieval","publication_year":2026,"publication_date":"2026-02-03","ids":{"openalex":"https://openalex.org/W7127602484","doi":"https://doi.org/10.48550/arxiv.2602.03306"},"language":null,"primary_location":{"id":"pmh:doi:10.48550/arxiv.2602.03306","is_oa":true,"landing_page_url":null,"pdf_url":null,"source":{"id":"https://openalex.org/S4406922384","display_name":"Open MIND","issn_l":null,"issn":null,"is_oa":false,"is_in_doaj":false,"is_core":false,"host_organization":null,"host_organization_name":null,"host_organization_lineage":[],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Article"},"type":"preprint","indexed_in":["datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":null,"any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5085211031","display_name":"Zhanyu Wu","orcid":"https://orcid.org/0000-0003-0635-0071"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wu, Zhanyu","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5027015677","display_name":"Richong Zhang","orcid":"https://orcid.org/0000-0002-1207-0300"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhang, Richong","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5043077962","display_name":"Zhijie Nie","orcid":"https://orcid.org/0009-0004-3933-0522"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Nie, Zhijie","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"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":null,"last_page":null},"is_retracted":false,"is_paratext":false,"is_xpac":false,"primary_topic":{"id":"https://openalex.org/T11273","display_name":"Advanced Graph Neural Networks","score":0.4717999994754791,"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"}},"topics":[{"id":"https://openalex.org/T11273","display_name":"Advanced Graph Neural Networks","score":0.4717999994754791,"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/T10286","display_name":"Information Retrieval and Search Behavior","score":0.20550000667572021,"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/T11307","display_name":"Domain Adaptation and Few-Shot Learning","score":0.0738999992609024,"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.611299991607666},{"id":"https://openalex.org/keywords/leverage","display_name":"Leverage (statistics)","score":0.5450999736785889},{"id":"https://openalex.org/keywords/dimension","display_name":"Dimension (graph theory)","score":0.5351999998092651},{"id":"https://openalex.org/keywords/relevance-feedback","display_name":"Relevance feedback","score":0.4611999988555908},{"id":"https://openalex.org/keywords/relevance","display_name":"Relevance (law)","score":0.435699999332428},{"id":"https://openalex.org/keywords/oracle","display_name":"Oracle","score":0.42250001430511475},{"id":"https://openalex.org/keywords/dimensionality-reduction","display_name":"Dimensionality reduction","score":0.41029998660087585},{"id":"https://openalex.org/keywords/encode","display_name":"ENCODE","score":0.3885999917984009},{"id":"https://openalex.org/keywords/computation","display_name":"Computation","score":0.3797000050544739}],"concepts":[{"id":"https://openalex.org/C41608201","wikidata":"https://www.wikidata.org/wiki/Q980509","display_name":"Embedding","level":2,"score":0.611299991607666},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.5999000072479248},{"id":"https://openalex.org/C153083717","wikidata":"https://www.wikidata.org/wiki/Q6535263","display_name":"Leverage (statistics)","level":2,"score":0.5450999736785889},{"id":"https://openalex.org/C33676613","wikidata":"https://www.wikidata.org/wiki/Q13415176","display_name":"Dimension (graph theory)","level":2,"score":0.5351999998092651},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.4672999978065491},{"id":"https://openalex.org/C2779532271","wikidata":"https://www.wikidata.org/wiki/Q445558","display_name":"Relevance feedback","level":4,"score":0.4611999988555908},{"id":"https://openalex.org/C158154518","wikidata":"https://www.wikidata.org/wiki/Q7310970","display_name":"Relevance (law)","level":2,"score":0.435699999332428},{"id":"https://openalex.org/C55166926","wikidata":"https://www.wikidata.org/wiki/Q2892946","display_name":"Oracle","level":2,"score":0.42250001430511475},{"id":"https://openalex.org/C70518039","wikidata":"https://www.wikidata.org/wiki/Q16000077","display_name":"Dimensionality reduction","level":2,"score":0.41029998660087585},{"id":"https://openalex.org/C66746571","wikidata":"https://www.wikidata.org/wiki/Q1134833","display_name":"ENCODE","level":3,"score":0.3885999917984009},{"id":"https://openalex.org/C45374587","wikidata":"https://www.wikidata.org/wiki/Q12525525","display_name":"Computation","level":2,"score":0.3797000050544739},{"id":"https://openalex.org/C81917197","wikidata":"https://www.wikidata.org/wiki/Q628760","display_name":"Selection (genetic algorithm)","level":2,"score":0.37229999899864197},{"id":"https://openalex.org/C2777402240","wikidata":"https://www.wikidata.org/wiki/Q6783436","display_name":"Masking (illustration)","level":2,"score":0.36890000104904175},{"id":"https://openalex.org/C124101348","wikidata":"https://www.wikidata.org/wiki/Q172491","display_name":"Data mining","level":1,"score":0.33550000190734863},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.33480000495910645},{"id":"https://openalex.org/C153180895","wikidata":"https://www.wikidata.org/wiki/Q7148389","display_name":"Pattern recognition (psychology)","level":2,"score":0.32899999618530273},{"id":"https://openalex.org/C173801870","wikidata":"https://www.wikidata.org/wiki/Q201413","display_name":"Heuristic","level":2,"score":0.31839999556541443},{"id":"https://openalex.org/C136389625","wikidata":"https://www.wikidata.org/wiki/Q334384","display_name":"Supervised learning","level":3,"score":0.3165000081062317},{"id":"https://openalex.org/C103278499","wikidata":"https://www.wikidata.org/wiki/Q254465","display_name":"Similarity (geometry)","level":3,"score":0.31450000405311584},{"id":"https://openalex.org/C99016210","wikidata":"https://www.wikidata.org/wiki/Q5488129","display_name":"Query expansion","level":2,"score":0.30889999866485596},{"id":"https://openalex.org/C2780801425","wikidata":"https://www.wikidata.org/wiki/Q5164392","display_name":"Construct (python library)","level":2,"score":0.29649999737739563},{"id":"https://openalex.org/C93959086","wikidata":"https://www.wikidata.org/wiki/Q6888345","display_name":"Model selection","level":2,"score":0.2939999997615814},{"id":"https://openalex.org/C2777210771","wikidata":"https://www.wikidata.org/wiki/Q4927124","display_name":"Block (permutation group theory)","level":2,"score":0.28380000591278076},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.2816999852657318},{"id":"https://openalex.org/C23123220","wikidata":"https://www.wikidata.org/wiki/Q816826","display_name":"Information retrieval","level":1,"score":0.2702000141143799},{"id":"https://openalex.org/C87117476","wikidata":"https://www.wikidata.org/wiki/Q362383","display_name":"Cardinality (data modeling)","level":2,"score":0.263700008392334},{"id":"https://openalex.org/C2776145971","wikidata":"https://www.wikidata.org/wiki/Q30673951","display_name":"Labeled data","level":2,"score":0.25850000977516174},{"id":"https://openalex.org/C51632099","wikidata":"https://www.wikidata.org/wiki/Q3985153","display_name":"Training set","level":2,"score":0.257099986076355},{"id":"https://openalex.org/C115311070","wikidata":"https://www.wikidata.org/wiki/Q5347255","display_name":"Effective dimension","level":3,"score":0.2567000091075897},{"id":"https://openalex.org/C75564084","wikidata":"https://www.wikidata.org/wiki/Q5597085","display_name":"Graph embedding","level":3,"score":0.2500999867916107}],"mesh":[],"locations_count":2,"locations":[{"id":"pmh:doi:10.48550/arxiv.2602.03306","is_oa":true,"landing_page_url":null,"pdf_url":null,"source":{"id":"https://openalex.org/S4406922384","display_name":"Open MIND","issn_l":null,"issn":null,"is_oa":false,"is_in_doaj":false,"is_core":false,"host_organization":null,"host_organization_name":null,"host_organization_lineage":[],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Article"},{"id":"doi:10.48550/arxiv.2602.03306","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2602.03306","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":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":null,"is_accepted":false,"is_published":null,"raw_source_name":null,"raw_type":"article"}],"best_oa_location":{"id":"pmh:doi:10.48550/arxiv.2602.03306","is_oa":true,"landing_page_url":null,"pdf_url":null,"source":{"id":"https://openalex.org/S4406922384","display_name":"Open MIND","issn_l":null,"issn":null,"is_oa":false,"is_in_doaj":false,"is_core":false,"host_organization":null,"host_organization_name":null,"host_organization_lineage":[],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Article"},"sustainable_development_goals":[],"awards":[],"funders":[],"has_content":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Dense":[0],"retrieval":[1,176],"represents":[2],"queries":[3,84],"and":[4,61,85,125,167,187],"documents":[5],"as":[6,182,184],"high-dimensional":[7],"embeddings,":[8],"but":[9,54],"these":[10,136],"representations":[11],"can":[12,47],"be":[13],"redundant":[14],"at":[15],"the":[16,142,156,179],"query":[17,109,133,157],"level:":[18],"for":[19,32,150],"a":[20,25,95,128,132,145],"given":[21],"information":[22],"need,":[23],"only":[24],"subset":[26,147],"of":[27,148],"dimensions":[28,120,149],"is":[29],"consistently":[30],"helpful":[31],"ranking.":[33],"Prior":[34],"work":[35],"addresses":[36],"this":[37],"via":[38],"pseudo-relevance":[39,160],"feedback":[40],"(PRF)":[41],"based":[42,153],"dimension":[43,91,115,173],"importance":[44,106,116,138],"estimation,":[45],"which":[46],"produce":[48],"query-aware":[49,90,146],"masks":[50],"without":[51,159],"labeled":[52],"data":[53],"often":[55],"relies":[56],"on":[57,155],"noisy":[58],"pseudo":[59],"signals":[60],"heuristic":[62],"test-time":[63],"procedures.":[64],"In":[65],"contrast,":[66],"supervised":[67,122,188],"adapter":[68,189],"methods":[69],"leverage":[70],"relevance":[71,123],"labels":[72],"to":[73,103,130,135],"improve":[74],"embedding":[75,119,134],"quality,":[76],"yet":[77],"they":[78],"learn":[79],"global":[80],"transformations":[81],"shared":[82],"across":[83,163],"do":[86],"not":[87],"explicitly":[88],"model":[89],"importance.":[92],"We":[93,111],"propose":[94],"Query-Aware":[96],"Adaptive":[97],"Dimension":[98],"Selection":[99],"framework":[100],"that":[101,170],"\\emph{learns}":[102],"predict":[104],"per-dimension":[105],"directly":[107],"from":[108],"embedding.":[110],"first":[112],"construct":[113],"oracle":[114],"distributions":[117],"over":[118,178],"using":[121],"labels,":[124],"then":[126],"train":[127],"predictor":[129,143],"map":[131],"label-distilled":[137],"scores.":[139],"At":[140],"inference,":[141],"selects":[144],"similarity":[151],"computation":[152],"solely":[154],"embedding,":[158],"feedback.":[161],"Experiments":[162],"multiple":[164],"dense":[165],"retrievers":[166],"benchmarks":[168],"show":[169],"our":[171],"learned":[172],"selector":[174],"improves":[175],"effectiveness":[177],"full-dimensional":[180],"baseline":[181],"well":[183],"PRF-based":[185],"masking":[186],"baselines.":[190]},"counts_by_year":[],"updated_date":"2026-06-11T09:08:48.828518","created_date":"2026-02-06T00:00:00"}