{"id":"https://openalex.org/W7162429073","doi":"https://doi.org/10.48550/arxiv.2605.24749","title":"How Neural Reward Models Learn Features for Policy Optimization: A Single-Index Analysis","display_name":"How Neural Reward Models Learn Features for Policy Optimization: A Single-Index Analysis","publication_year":2026,"publication_date":"2026-05-23","ids":{"openalex":"https://openalex.org/W7162429073","doi":"https://doi.org/10.48550/arxiv.2605.24749"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2605.24749","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.24749","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"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":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.24749","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5137031073","display_name":"Rei Higuchi","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Higuchi, Rei","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5055978970","display_name":"Ryotaro Kawata","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Kawata, Ryotaro","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137079119","display_name":"Akifumi Wachi","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wachi, Akifumi","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5119989939","display_name":"Shokichi Takakura","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Takakura, Shokichi","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5001545412","display_name":"Kohei Miyaguchi","orcid":"https://orcid.org/0000-0002-6702-7780"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Miyaguchi, Kohei","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5137001253","display_name":"Taiji Suzuki","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Suzuki, Taiji","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/T11612","display_name":"Stochastic Gradient Optimization Techniques","score":0.22300000488758087,"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/T11612","display_name":"Stochastic Gradient Optimization Techniques","score":0.22300000488758087,"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/T10462","display_name":"Reinforcement Learning in Robotics","score":0.17890000343322754,"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/T11689","display_name":"Adversarial Robustness in Machine Learning","score":0.1005999967455864,"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/weighting","display_name":"Weighting","score":0.6409000158309937},{"id":"https://openalex.org/keywords/set","display_name":"Set (abstract data type)","score":0.5910000205039978},{"id":"https://openalex.org/keywords/constant","display_name":"Constant (computer programming)","score":0.5755000114440918},{"id":"https://openalex.org/keywords/feature","display_name":"Feature (linguistics)","score":0.5393999814987183},{"id":"https://openalex.org/keywords/exponential-function","display_name":"Exponential function","score":0.5062000155448914},{"id":"https://openalex.org/keywords/gaussian","display_name":"Gaussian","score":0.46810001134872437},{"id":"https://openalex.org/keywords/generative-model","display_name":"Generative model","score":0.4645000100135803},{"id":"https://openalex.org/keywords/fraction","display_name":"Fraction (chemistry)","score":0.42500001192092896},{"id":"https://openalex.org/keywords/artificial-neural-network","display_name":"Artificial neural network","score":0.420199990272522},{"id":"https://openalex.org/keywords/key","display_name":"Key (lock)","score":0.35589998960494995}],"concepts":[{"id":"https://openalex.org/C183115368","wikidata":"https://www.wikidata.org/wiki/Q856577","display_name":"Weighting","level":2,"score":0.6409000158309937},{"id":"https://openalex.org/C177264268","wikidata":"https://www.wikidata.org/wiki/Q1514741","display_name":"Set (abstract data type)","level":2,"score":0.5910000205039978},{"id":"https://openalex.org/C2777027219","wikidata":"https://www.wikidata.org/wiki/Q1284190","display_name":"Constant (computer programming)","level":2,"score":0.5755000114440918},{"id":"https://openalex.org/C2776401178","wikidata":"https://www.wikidata.org/wiki/Q12050496","display_name":"Feature (linguistics)","level":2,"score":0.5393999814987183},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.525600016117096},{"id":"https://openalex.org/C151376022","wikidata":"https://www.wikidata.org/wiki/Q168698","display_name":"Exponential function","level":2,"score":0.5062000155448914},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.4796999990940094},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.4787999987602234},{"id":"https://openalex.org/C163716315","wikidata":"https://www.wikidata.org/wiki/Q901177","display_name":"Gaussian","level":2,"score":0.46810001134872437},{"id":"https://openalex.org/C167966045","wikidata":"https://www.wikidata.org/wiki/Q5532625","display_name":"Generative model","level":3,"score":0.4645000100135803},{"id":"https://openalex.org/C149629883","wikidata":"https://www.wikidata.org/wiki/Q660926","display_name":"Fraction (chemistry)","level":2,"score":0.42500001192092896},{"id":"https://openalex.org/C50644808","wikidata":"https://www.wikidata.org/wiki/Q192776","display_name":"Artificial neural network","level":2,"score":0.420199990272522},{"id":"https://openalex.org/C26517878","wikidata":"https://www.wikidata.org/wiki/Q228039","display_name":"Key (lock)","level":2,"score":0.35589998960494995},{"id":"https://openalex.org/C2779843651","wikidata":"https://www.wikidata.org/wiki/Q7390335","display_name":"SIGNAL (programming language)","level":2,"score":0.34599998593330383},{"id":"https://openalex.org/C61326573","wikidata":"https://www.wikidata.org/wiki/Q1496376","display_name":"Gaussian process","level":3,"score":0.3375999927520752},{"id":"https://openalex.org/C151201525","wikidata":"https://www.wikidata.org/wiki/Q177239","display_name":"Limit (mathematics)","level":2,"score":0.3343999981880188},{"id":"https://openalex.org/C2776291640","wikidata":"https://www.wikidata.org/wiki/Q2912517","display_name":"Value (mathematics)","level":2,"score":0.3314000070095062},{"id":"https://openalex.org/C61224824","wikidata":"https://www.wikidata.org/wiki/Q2260434","display_name":"Mixture model","level":2,"score":0.3287999927997589},{"id":"https://openalex.org/C97541855","wikidata":"https://www.wikidata.org/wiki/Q830687","display_name":"Reinforcement learning","level":2,"score":0.325300008058548},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.3246000111103058},{"id":"https://openalex.org/C167085575","wikidata":"https://www.wikidata.org/wiki/Q6803654","display_name":"Mean squared prediction error","level":2,"score":0.320499986410141},{"id":"https://openalex.org/C153180895","wikidata":"https://www.wikidata.org/wiki/Q7148389","display_name":"Pattern recognition (psychology)","level":2,"score":0.30790001153945923},{"id":"https://openalex.org/C48677424","wikidata":"https://www.wikidata.org/wiki/Q6888088","display_name":"Mode (computer interface)","level":2,"score":0.3050000071525574},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.295199990272522},{"id":"https://openalex.org/C75235859","wikidata":"https://www.wikidata.org/wiki/Q582659","display_name":"Exponential growth","level":2,"score":0.2858999967575073},{"id":"https://openalex.org/C3832189","wikidata":"https://www.wikidata.org/wiki/Q8588916","display_name":"Models of neural computation","level":3,"score":0.27799999713897705},{"id":"https://openalex.org/C196340769","wikidata":"https://www.wikidata.org/wiki/Q7698910","display_name":"Temporal difference learning","level":3,"score":0.26089999079704285},{"id":"https://openalex.org/C32277403","wikidata":"https://www.wikidata.org/wiki/Q740445","display_name":"Ridge","level":2,"score":0.26080000400543213},{"id":"https://openalex.org/C2780813799","wikidata":"https://www.wikidata.org/wiki/Q3274237","display_name":"Zero (linguistics)","level":2,"score":0.26030001044273376},{"id":"https://openalex.org/C2778112365","wikidata":"https://www.wikidata.org/wiki/Q3511065","display_name":"Sequence (biology)","level":2,"score":0.2599000036716461},{"id":"https://openalex.org/C2164484","wikidata":"https://www.wikidata.org/wiki/Q5170150","display_name":"Core (optical fiber)","level":2,"score":0.2597000002861023},{"id":"https://openalex.org/C28826006","wikidata":"https://www.wikidata.org/wiki/Q33521","display_name":"Applied mathematics","level":1,"score":0.2535000145435333},{"id":"https://openalex.org/C2779227376","wikidata":"https://www.wikidata.org/wiki/Q6505497","display_name":"Layer (electronics)","level":2,"score":0.2515999972820282}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2605.24749","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.24749","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"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":"Preprint"}],"best_oa_location":{"id":"doi:10.48550/arxiv.2605.24749","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.24749","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"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":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":{"Reward":[0],"modeling":[1],"is":[2,15],"not":[3],"only":[4],"a":[5,36,53,96,100,134],"prediction":[6],"problem:":[7],"in":[8,28,35,167],"KL-regularized":[9],"policy":[10],"optimization,":[11],"the":[12,19,61,71,82,87,106,114,143,154,160,168],"learned":[13],"reward":[14,56,79],"exponentiated":[16],"to":[17,86],"define":[18],"deployed":[20],"policy,":[21],"so":[22],"downstream":[23],"value":[24],"depends":[25],"on":[26],"errors":[27],"reward-tilted":[29],"regions.":[30],"We":[31,51],"study":[32],"this":[33,174],"feedback":[34],"Gaussian":[37],"single-index":[38],"model":[39,57],"with":[40,109,130,139],"$r^*(x)":[41],"=":[42],"\u03c3^*(\\langle":[43],"\u03b8^*,":[44],"x\\rangle)$":[45],"and":[46,68,133],"$x":[47],"\\sim":[48],"N(0,":[49],"I_d)$.":[50],"analyze":[52],"two-stage":[54],"neural":[55],"that":[58],"first":[59,88],"learns":[60],"hidden":[62,107],"direction":[63],"$\u03b8^*$":[64],"from":[65,156],"reward-weighted":[66],"samples":[67],"then":[69],"fits":[70],"readout":[72],"layer":[73],"by":[74,113,164],"weighted":[75],"ridge":[76],"regression.":[77],"Exponential":[78],"weighting":[80],"changes":[81],"Hermite":[83],"signal":[84],"available":[85],"layer;":[89],"for":[90,125],"any":[91],"feature-learning":[92],"temperature":[93],"$\u03b2_1$":[94],"above":[95],"dimension-free":[97],"$O(1)$":[98],"threshold,":[99],"constant":[101],"fraction":[102],"of":[103,150],"neurons":[104],"recover":[105],"direction,":[108],"weak-recovery":[110],"complexity":[111],"governed":[112],"generative":[115],"exponent.":[116],"After":[117],"feature":[118],"recovery,":[119],"we":[120],"derive":[121],"tilted-policy":[122],"value-gap":[123],"bounds":[124],"an":[126,147],"idealized":[127],"label-weighted":[128],"fit":[129,138],"weights":[131,140],"$e^{y/\u03b2_2}$":[132],"more":[135],"practical":[136],"surrogate-weighted":[137,169],"$e^{r_{a_0}(x)/\u03b2_2}$.":[141],"Keeping":[142],"$\u03b2_2$-dependence":[144],"explicit":[145],"yields":[146],"admissible":[148,175],"set":[149],"deployment":[151],"temperatures,":[152],"balancing":[153],"gain":[155],"lowering":[157],"$\u03b2_2$":[158],"against":[159],"learning":[161],"cost":[162],"amplified":[163],"exponential":[165],"weighting;":[166],"case,":[170],"proxy-dependent":[171],"factors":[172],"shrink":[173],"set.":[176]},"counts_by_year":[],"updated_date":"2026-07-28T07:46:37.118299","created_date":"2026-05-27T00:00:00"}