{"id":"https://openalex.org/W7162410668","doi":"https://doi.org/10.48550/arxiv.2605.24425","title":"Momentum Streams for Optimizer-Inspired Transformers","display_name":"Momentum Streams for Optimizer-Inspired Transformers","publication_year":2026,"publication_date":"2026-05-23","ids":{"openalex":"https://openalex.org/W7162410668","doi":"https://doi.org/10.48550/arxiv.2605.24425"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2605.24425","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.24425","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":"article"},"type":"preprint","indexed_in":["datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://doi.org/10.48550/arxiv.2605.24425","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5137045477","display_name":"Jingchu Gai","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Gai, Jingchu","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137017314","display_name":"Nai-Chieh Huang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Huang, Nai-Chieh","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5137042296","display_name":"Jiayun Wu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wu, Jiayun","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/T10048","display_name":"Particle physics theoretical and experimental studies","score":0.24940000474452972,"subfield":{"id":"https://openalex.org/subfields/3106","display_name":"Nuclear and High Energy Physics"},"field":{"id":"https://openalex.org/fields/31","display_name":"Physics and Astronomy"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},"topics":[{"id":"https://openalex.org/T10048","display_name":"Particle physics theoretical and experimental studies","score":0.24940000474452972,"subfield":{"id":"https://openalex.org/subfields/3106","display_name":"Nuclear and High Energy Physics"},"field":{"id":"https://openalex.org/fields/31","display_name":"Physics and Astronomy"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T10682","display_name":"Quantum Computing Algorithms and Architecture","score":0.08879999816417694,"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/T13650","display_name":"Computational Physics and Python Applications","score":0.0754999965429306,"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/transformer","display_name":"Transformer","score":0.6442999839782715},{"id":"https://openalex.org/keywords/maxima-and-minima","display_name":"Maxima and minima","score":0.5472999811172485},{"id":"https://openalex.org/keywords/residual","display_name":"Residual","score":0.5116000175476074},{"id":"https://openalex.org/keywords/control-theory","display_name":"Control theory (sociology)","score":0.4634000062942505},{"id":"https://openalex.org/keywords/streams","display_name":"STREAMS","score":0.38370001316070557},{"id":"https://openalex.org/keywords/security-token","display_name":"Security token","score":0.36039999127388}],"concepts":[{"id":"https://openalex.org/C66322947","wikidata":"https://www.wikidata.org/wiki/Q11658","display_name":"Transformer","level":3,"score":0.6442999839782715},{"id":"https://openalex.org/C186633575","wikidata":"https://www.wikidata.org/wiki/Q845060","display_name":"Maxima and minima","level":2,"score":0.5472999811172485},{"id":"https://openalex.org/C155512373","wikidata":"https://www.wikidata.org/wiki/Q287450","display_name":"Residual","level":2,"score":0.5116000175476074},{"id":"https://openalex.org/C47446073","wikidata":"https://www.wikidata.org/wiki/Q5165890","display_name":"Control theory (sociology)","level":3,"score":0.4634000062942505},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.43970000743865967},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.42590001225471497},{"id":"https://openalex.org/C42090638","wikidata":"https://www.wikidata.org/wiki/Q4048907","display_name":"STREAMS","level":2,"score":0.38370001316070557},{"id":"https://openalex.org/C48145219","wikidata":"https://www.wikidata.org/wiki/Q1335365","display_name":"Security token","level":2,"score":0.36039999127388},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.3529999852180481},{"id":"https://openalex.org/C24326235","wikidata":"https://www.wikidata.org/wiki/Q126095","display_name":"Electronic engineering","level":1,"score":0.3379000127315521},{"id":"https://openalex.org/C39927690","wikidata":"https://www.wikidata.org/wiki/Q11197","display_name":"Logarithm","level":2,"score":0.33180001378059387},{"id":"https://openalex.org/C127413603","wikidata":"https://www.wikidata.org/wiki/Q11023","display_name":"Engineering","level":0,"score":0.3312999904155731},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.31949999928474426},{"id":"https://openalex.org/C7149132","wikidata":"https://www.wikidata.org/wiki/Q1377840","display_name":"Forgetting","level":2,"score":0.30550000071525574},{"id":"https://openalex.org/C119599485","wikidata":"https://www.wikidata.org/wiki/Q43035","display_name":"Electrical engineering","level":1,"score":0.30469998717308044},{"id":"https://openalex.org/C184720557","wikidata":"https://www.wikidata.org/wiki/Q7825049","display_name":"Topology (electrical circuits)","level":2,"score":0.28630000352859497},{"id":"https://openalex.org/C121332964","wikidata":"https://www.wikidata.org/wiki/Q413","display_name":"Physics","level":0,"score":0.2831000089645386},{"id":"https://openalex.org/C50644808","wikidata":"https://www.wikidata.org/wiki/Q192776","display_name":"Artificial neural network","level":2,"score":0.2556000053882599},{"id":"https://openalex.org/C117323899","wikidata":"https://www.wikidata.org/wiki/Q1907989","display_name":"Distribution transformer","level":4,"score":0.25429999828338623}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2605.24425","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.24425","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":"article"}],"best_oa_location":{"id":"doi:10.48550/arxiv.2605.24425","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.24425","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":"article"},"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":{"The":[0],"residual":[1],"update":[2],"of":[3,14,42,91],"a":[4,15,20,40],"pre-norm":[5],"Transformer":[6,71],"layer":[7],"admits":[8],"an":[9],"interpretation":[10],"as":[11,31],"one":[12],"step":[13],"first-order":[16],"optimizer":[17],"acting":[18],"on":[19,35],"surrogate":[21],"token":[22],"energy,":[23],"wherein":[24],"the":[25,60,64,69,88,92,107],"attention":[26],"and":[27,49,72,79,99,115],"MLP":[28],"sublayers":[29],"function":[30],"gradient":[32],"oracles.":[33],"Based":[34],"this":[36],"observation,":[37],"we":[38],"build":[39],"family":[41],"optimizer-inspired":[43],"Transformers":[44],"(triple-momentum,":[45],"Adam/AdamW,":[46],"Muon,":[47],"SOAP)":[48],"compare":[50],"them":[51],"under":[52],"matched":[53],"compute.":[54],"In":[55],"our":[56],"main":[57,89],"pretraining":[58],"experiment,":[59],"triple-momentum":[61],"TMMFormer":[62,98],"achieves":[63],"lowest":[65],"validation":[66],"loss,":[67],"outperforming":[68],"vanilla":[70,108],"prior":[73],"architectural":[74],"variants.":[75],"A":[76],"controlled":[77],"ablation":[78],"supporting":[80],"theory":[81],"show":[82,96],"that":[83,97],"momentum,":[84],"not":[85],"preconditioning,":[86],"is":[87],"source":[90],"gain.":[93],"We":[94],"further":[95],"other":[100],"momentum-based":[101],"designs":[102],"reach":[103],"flatter":[104],"minima":[105],"than":[106],"Transformer,":[109],"which":[110],"leads":[111],"to":[112],"less":[113],"forgetting":[114],"better":[116],"generalization.":[117]},"counts_by_year":[],"updated_date":"2026-06-11T09:08:48.828518","created_date":"2026-05-27T00:00:00"}