{"id":"https://openalex.org/W7163388604","doi":"https://doi.org/10.48550/arxiv.2606.03329","title":"InfoMem: Training Long-Context Memory Agents with Answer-Conditioned Information Gain","display_name":"InfoMem: Training Long-Context Memory Agents with Answer-Conditioned Information Gain","publication_year":2026,"publication_date":"2026-06-02","ids":{"openalex":"https://openalex.org/W7163388604","doi":"https://doi.org/10.48550/arxiv.2606.03329"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2606.03329","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.03329","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.2606.03329","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5044488602","display_name":"Tiancheng Han","orcid":"https://orcid.org/0009-0003-4428-1262"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Han, Tiancheng","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137797252","display_name":"Yong Li (15029)","orcid":"https://orcid.org/0000-0001-5617-1659"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Li, Yong","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5102872336","display_name":"Wuzhou Yu","orcid":"https://orcid.org/0000-0002-5983-5796"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yu, Wuzhou","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137756651","display_name":"Qiaosheng Zhang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhang, Qiaosheng","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5137802683","display_name":"Wenqi Shao","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Shao, Wenqi","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/T10028","display_name":"Topic Modeling","score":0.34369999170303345,"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/T10028","display_name":"Topic Modeling","score":0.34369999170303345,"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/T11714","display_name":"Multimodal Machine Learning Applications","score":0.2150000035762787,"subfield":{"id":"https://openalex.org/subfields/1707","display_name":"Computer Vision and Pattern Recognition"},"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.09589999914169312,"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/task","display_name":"Task (project management)","score":0.7567999958992004},{"id":"https://openalex.org/keywords/encoding","display_name":"Encoding (memory)","score":0.4880000054836273},{"id":"https://openalex.org/keywords/code","display_name":"Code (set theory)","score":0.4375},{"id":"https://openalex.org/keywords/training","display_name":"Training (meteorology)","score":0.43149998784065247},{"id":"https://openalex.org/keywords/reading","display_name":"Reading (process)","score":0.42989999055862427},{"id":"https://openalex.org/keywords/mechanism","display_name":"Mechanism (biology)","score":0.41499999165534973},{"id":"https://openalex.org/keywords/key","display_name":"Key (lock)","score":0.3882000148296356}],"concepts":[{"id":"https://openalex.org/C2780451532","wikidata":"https://www.wikidata.org/wiki/Q759676","display_name":"Task (project management)","level":2,"score":0.7567999958992004},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7098000049591064},{"id":"https://openalex.org/C125411270","wikidata":"https://www.wikidata.org/wiki/Q18653","display_name":"Encoding (memory)","level":2,"score":0.4880000054836273},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.45649999380111694},{"id":"https://openalex.org/C2776760102","wikidata":"https://www.wikidata.org/wiki/Q5139990","display_name":"Code (set theory)","level":3,"score":0.4375},{"id":"https://openalex.org/C2777211547","wikidata":"https://www.wikidata.org/wiki/Q17141490","display_name":"Training (meteorology)","level":2,"score":0.43149998784065247},{"id":"https://openalex.org/C554936623","wikidata":"https://www.wikidata.org/wiki/Q199657","display_name":"Reading (process)","level":2,"score":0.42989999055862427},{"id":"https://openalex.org/C89611455","wikidata":"https://www.wikidata.org/wiki/Q6804646","display_name":"Mechanism (biology)","level":2,"score":0.41499999165534973},{"id":"https://openalex.org/C26517878","wikidata":"https://www.wikidata.org/wiki/Q228039","display_name":"Key (lock)","level":2,"score":0.3882000148296356},{"id":"https://openalex.org/C100660578","wikidata":"https://www.wikidata.org/wiki/Q18733","display_name":"Recall","level":2,"score":0.3882000148296356},{"id":"https://openalex.org/C180747234","wikidata":"https://www.wikidata.org/wiki/Q23373","display_name":"Cognitive psychology","level":1,"score":0.3808000087738037},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.36820000410079956},{"id":"https://openalex.org/C2779843651","wikidata":"https://www.wikidata.org/wiki/Q7390335","display_name":"SIGNAL (programming language)","level":2,"score":0.35679998993873596},{"id":"https://openalex.org/C2983203078","wikidata":"https://www.wikidata.org/wiki/Q255166","display_name":"Information gain","level":2,"score":0.31290000677108765},{"id":"https://openalex.org/C175154964","wikidata":"https://www.wikidata.org/wiki/Q380077","display_name":"Task analysis","level":3,"score":0.30889999866485596},{"id":"https://openalex.org/C21963081","wikidata":"https://www.wikidata.org/wiki/Q11337567","display_name":"Working memory","level":3,"score":0.2851000130176544},{"id":"https://openalex.org/C107457646","wikidata":"https://www.wikidata.org/wiki/Q207434","display_name":"Human\u2013computer interaction","level":1,"score":0.28349998593330383},{"id":"https://openalex.org/C15744967","wikidata":"https://www.wikidata.org/wiki/Q9418","display_name":"Psychology","level":0,"score":0.2799000144004822},{"id":"https://openalex.org/C176649486","wikidata":"https://www.wikidata.org/wiki/Q2308807","display_name":"Memory management","level":3,"score":0.2736999988555908},{"id":"https://openalex.org/C43126263","wikidata":"https://www.wikidata.org/wiki/Q128751","display_name":"Source code","level":2,"score":0.2565000057220459}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2606.03329","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.03329","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.2606.03329","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.03329","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":[{"display_name":"Quality Education","id":"https://metadata.un.org/sdg/4","score":0.5477713942527771}],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Long-context":[0],"tasks":[1],"require":[2],"LLMs":[3],"to":[4,122],"identify":[5],"and":[6,28,55,125,136,165],"preserve":[7],"answer-relevant":[8],"information":[9],"from":[10,33],"large":[11],"contexts.":[12],"Chunk-wise":[13],"memory":[14,54,74,88,103],"agents":[15,41,89],"address":[16],"this":[17,119],"issue":[18],"by":[19],"sequentially":[20],"reading":[21],"document":[22],"chunks,":[23],"updating":[24],"a":[25,82],"compact":[26],"memory,":[27],"generating":[29],"the":[30,34,72,76,101,105,110,132,169,173],"final":[31,73,102],"answer":[32,170],"accumulated":[35],"memory.":[36],"However,":[37],"existing":[38],"RL-based":[39],"chunk-wise":[40,87],"either":[42],"rely":[43],"on":[44,157,168],"sparse":[45],"final-answer":[46],"rewards":[47,52,154],"or":[48,63],"use":[49],"lexical":[50],"intermediate":[51],"for":[53,85],"retrieval":[56],"actions.":[57],"These":[58],"signals":[59],"supervise":[60],"task":[61],"success":[62],"local":[64],"overlap,":[65],"but":[66],"do":[67],"not":[68],"directly":[69],"evaluate":[70],"whether":[71],"supports":[75],"ground-truth":[77,111],"answer.":[78,112],"We":[79],"propose":[80],"InfoMem,":[81],"reward":[83,129,163],"mechanism":[84],"training":[86,137],"that":[90,151],"evaluates":[91],"final-memory":[92,153],"utility":[93],"using":[94],"answer-conditioned":[95],"information.":[96],"InfoMem":[97,117,139],"measures":[98],"how":[99],"much":[100],"increases":[104],"model's":[106],"per-token":[107],"log-likelihood":[108],"of":[109],"To":[113],"stabilize":[114],"RL":[115,147],"optimization,":[116],"applies":[118],"signal":[120],"only":[121],"successful":[123,158],"trajectories":[124],"normalizes":[126],"it":[127],"before":[128,162],"composition.":[130],"Under":[131],"same":[133],"GRPO":[134],"framework":[135],"budget,":[138],"improves":[140],"long-context":[141],"memory-agent":[142,146],"performance":[143],"over":[144],"comparable":[145],"baselines.":[148],"Analyses":[149],"show":[150],"effective":[152],"should":[155],"operate":[156],"trajectories,":[159],"be":[160,166],"normalized":[161],"composition,":[164],"conditioned":[167],"rather":[171],"than":[172],"query.":[174],"Our":[175],"code":[176],"is":[177],"available":[178],"at":[179],"https://github.com/GenSouKa1/InfoMem.":[180]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-06-04T00:00:00"}