{"id":"https://openalex.org/W7162532412","doi":"https://doi.org/10.48550/arxiv.2605.26385","title":"Credit-assigned Policy Gradient for Early Stage Retrieval in Two-stage Ranking","display_name":"Credit-assigned Policy Gradient for Early Stage Retrieval in Two-stage Ranking","publication_year":2026,"publication_date":"2026-05-25","ids":{"openalex":"https://openalex.org/W7162532412","doi":"https://doi.org/10.48550/arxiv.2605.26385"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2605.26385","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.26385","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.2605.26385","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5001514927","display_name":"Haruka Kiyohara","orcid":"https://orcid.org/0009-0000-6378-4365"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Kiyohara, Haruka","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5025723886","display_name":"Mihaela Curmei","orcid":"https://orcid.org/0000-0002-3918-3065"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Curmei, Mihaela","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5106708657","display_name":"Ariel Evnine","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Evnine, Ariel","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5090033474","display_name":"Shankar Kalyanaraman","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Kalyanaraman, Shankar","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5059111932","display_name":"Israel Nir","orcid":"https://orcid.org/0000-0001-8698-4730"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Nir, Israel","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5085991817","display_name":"Ana-Roxana Pop","orcid":"https://orcid.org/0000-0003-3482-2734"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Pop, Ana-Roxana","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5078520797","display_name":"Nitzan Razin","orcid":"https://orcid.org/0000-0001-7571-5658"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Razin, Nitzan","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137181248","display_name":"Sarah Dean","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Dean, Sarah","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5014687727","display_name":"Thorsten Joachims","orcid":"https://orcid.org/0000-0003-3654-3683"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Joachims, Thorsten","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5000980244","display_name":"Udi Weinsberg","orcid":"https://orcid.org/0000-0002-6966-1945"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Weinsberg, Udi","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/T10203","display_name":"Recommender Systems and Techniques","score":0.4708999991416931,"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/T10203","display_name":"Recommender Systems and Techniques","score":0.4708999991416931,"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/T10286","display_name":"Information Retrieval and Search Behavior","score":0.41920000314712524,"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/T13274","display_name":"Expert finding and Q&A systems","score":0.012000000104308128,"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"}}],"keywords":[{"id":"https://openalex.org/keywords/ranking","display_name":"Ranking (information retrieval)","score":0.661899983882904},{"id":"https://openalex.org/keywords/set","display_name":"Set (abstract data type)","score":0.5554999709129333},{"id":"https://openalex.org/keywords/training-set","display_name":"Training set","score":0.5249000191688538},{"id":"https://openalex.org/keywords/stability","display_name":"Stability (learning theory)","score":0.5024999976158142},{"id":"https://openalex.org/keywords/scalability","display_name":"Scalability","score":0.4860999882221222},{"id":"https://openalex.org/keywords/convergence","display_name":"Convergence (economics)","score":0.4627000093460083},{"id":"https://openalex.org/keywords/variance","display_name":"Variance (accounting)","score":0.4065999984741211}],"concepts":[{"id":"https://openalex.org/C189430467","wikidata":"https://www.wikidata.org/wiki/Q7293293","display_name":"Ranking (information retrieval)","level":2,"score":0.661899983882904},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.6118999719619751},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.61080002784729},{"id":"https://openalex.org/C177264268","wikidata":"https://www.wikidata.org/wiki/Q1514741","display_name":"Set (abstract data type)","level":2,"score":0.5554999709129333},{"id":"https://openalex.org/C51632099","wikidata":"https://www.wikidata.org/wiki/Q3985153","display_name":"Training set","level":2,"score":0.5249000191688538},{"id":"https://openalex.org/C112972136","wikidata":"https://www.wikidata.org/wiki/Q7595718","display_name":"Stability (learning theory)","level":2,"score":0.5024999976158142},{"id":"https://openalex.org/C48044578","wikidata":"https://www.wikidata.org/wiki/Q727490","display_name":"Scalability","level":2,"score":0.4860999882221222},{"id":"https://openalex.org/C2777303404","wikidata":"https://www.wikidata.org/wiki/Q759757","display_name":"Convergence (economics)","level":2,"score":0.4627000093460083},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.45329999923706055},{"id":"https://openalex.org/C196083921","wikidata":"https://www.wikidata.org/wiki/Q7915758","display_name":"Variance (accounting)","level":2,"score":0.4065999984741211},{"id":"https://openalex.org/C153180895","wikidata":"https://www.wikidata.org/wiki/Q7148389","display_name":"Pattern recognition (psychology)","level":2,"score":0.40130001306533813},{"id":"https://openalex.org/C124101348","wikidata":"https://www.wikidata.org/wiki/Q172491","display_name":"Data mining","level":1,"score":0.37310001254081726},{"id":"https://openalex.org/C61797465","wikidata":"https://www.wikidata.org/wiki/Q1188986","display_name":"Term (time)","level":2,"score":0.3580000102519989},{"id":"https://openalex.org/C58489278","wikidata":"https://www.wikidata.org/wiki/Q1172284","display_name":"Data set","level":2,"score":0.3483999967575073},{"id":"https://openalex.org/C26517878","wikidata":"https://www.wikidata.org/wiki/Q228039","display_name":"Key (lock)","level":2,"score":0.3303999900817871},{"id":"https://openalex.org/C97541855","wikidata":"https://www.wikidata.org/wiki/Q830687","display_name":"Reinforcement learning","level":2,"score":0.32100000977516174},{"id":"https://openalex.org/C115680565","wikidata":"https://www.wikidata.org/wiki/Q5977448","display_name":"Gradient method","level":2,"score":0.31450000405311584},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.3003000020980835},{"id":"https://openalex.org/C185798385","wikidata":"https://www.wikidata.org/wiki/Q1161707","display_name":"Benchmark (surveying)","level":2,"score":0.2775999903678894}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2605.26385","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.26385","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.2605.26385","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.26385","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":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Large-scale":[0],"search,":[1],"recommendation,":[2],"and":[3,185,195],"retrieval-augmented":[4],"generation":[5],"(RAG)":[6],"systems":[7],"typically":[8],"employ":[9],"a":[10,18,26,109,176],"two-stage":[11],"architecture:":[12],"an":[13],"early-stage":[14],"ranker":[15,28],"(ESR)":[16],"generates":[17],"candidate":[19,87,98,131,137,162],"set,":[20,132,163],"which":[21,115],"is":[22,59,127,210],"subsequently":[23],"re-ranked":[24],"by":[25,154],"late-stage":[27],"(LSR).":[29],"While":[30],"there":[31],"are":[32],"many":[33],"reinforcement":[34],"learning":[35],"(RL)":[36],"methods":[37],"for":[38,62,66,198],"training":[39,43,196],"the":[40,45,79,82,86,90,97,101,121,124,150,157,161,166,170,192,201,207],"LSR,":[41],"end-to-end":[42],"of":[44,54,85,92,152,160,173],"ESR":[46],"has":[47],"proven":[48],"challenging.":[49],"In":[50],"particular,":[51],"naive":[52],"application":[53],"\"vanilla\"":[55],"policy":[56,112],"gradient":[57,80,113],"(V-PG)":[58],"not":[60],"scalable":[61],"candidate-set":[63,208],"sizes":[64],"relevant":[65],"practical":[67],"use":[68],"due":[69],"to":[70,81,100,120,168],"exploding":[71],"variance.":[72],"This":[73],"issue":[74],"arises":[75],"because":[76],"V-PG":[77,153],"propagates":[78],"joint":[83],"probability":[84,122],"sets,":[88],"ignoring":[89],"contribution":[91],"each":[93],"specific":[94,158],"item":[95,126],"in":[96,129],"set":[99],"reward.":[102],"To":[103],"mitigate":[104],"this":[105],"issue,":[106],"we":[107],"propose":[108],"novel":[110],"\"credit-assigned\"":[111],"(CA-PG),":[114],"computes":[116],"gradients":[117],"with":[118],"respect":[119],"that":[123,139,146,189],"target":[125],"chosen":[128],"any":[130],"i.e.":[133],"marginalizing":[134,155],"over":[135,156],"all":[136],"sets":[138],"contain":[140],"it.":[141],"Our":[142],"theoretical":[143],"analysis":[144],"reveals":[145],"CA-PG":[147,190],"significantly":[148],"reduces":[149],"variance":[151],"composition":[159],"while":[164],"preserving":[165],"ability":[167],"learn":[169],"correct":[171],"ranking":[172],"items":[174],"under":[175],"reasonably":[177],"aligned":[178],"LSR":[179],"policy.":[180],"Experiments":[181],"on":[182],"both":[183],"synthetic":[184],"real-world":[186],"data":[187],"demonstrate":[188],"improves":[191],"convergence":[193],"speed":[194],"stability":[197],"ESRs":[199],"utilizing":[200],"canonical":[202],"Plackett-Luce":[203],"model,":[204],"especially":[205],"when":[206],"size":[209],"large.":[211]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-05-28T00:00:00"}