{"id":"https://openalex.org/W7165059670","doi":"https://doi.org/10.48550/arxiv.2606.17526","title":"MGUP: A Momentum-Gradient Alignment Update Policy for Stochastic Optimization","display_name":"MGUP: A Momentum-Gradient Alignment Update Policy for Stochastic Optimization","publication_year":2026,"publication_date":"2026-06-16","ids":{"openalex":"https://openalex.org/W7165059670","doi":"https://doi.org/10.48550/arxiv.2606.17526"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2606.17526","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.17526","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":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.17526","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5138778289","display_name":"Da Chang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Chang, Da","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5040011580","display_name":"Ganzhao Yuan","orcid":"https://orcid.org/0000-0002-2239-7315"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yuan, Ganzhao","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.26179999113082886,"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.26179999113082886,"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/T10775","display_name":"Generative Adversarial Networks and Image Synthesis","score":0.10459999740123749,"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/T10028","display_name":"Topic Modeling","score":0.10400000214576721,"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/convergence","display_name":"Convergence (economics)","score":0.6909999847412109},{"id":"https://openalex.org/keywords/downstream","display_name":"Downstream (manufacturing)","score":0.41990000009536743},{"id":"https://openalex.org/keywords/base","display_name":"Base (topology)","score":0.40720000863075256},{"id":"https://openalex.org/keywords/mechanism","display_name":"Mechanism (biology)","score":0.3982999920845032},{"id":"https://openalex.org/keywords/stochastic-optimization","display_name":"Stochastic optimization","score":0.39239999651908875},{"id":"https://openalex.org/keywords/bridge","display_name":"Bridge (graph theory)","score":0.38589999079704285},{"id":"https://openalex.org/keywords/code","display_name":"Code (set theory)","score":0.3792000114917755},{"id":"https://openalex.org/keywords/control","display_name":"Control (management)","score":0.37720000743865967}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7674000263214111},{"id":"https://openalex.org/C2777303404","wikidata":"https://www.wikidata.org/wiki/Q759757","display_name":"Convergence (economics)","level":2,"score":0.6909999847412109},{"id":"https://openalex.org/C126255220","wikidata":"https://www.wikidata.org/wiki/Q141495","display_name":"Mathematical optimization","level":1,"score":0.558899998664856},{"id":"https://openalex.org/C2776207758","wikidata":"https://www.wikidata.org/wiki/Q5303302","display_name":"Downstream (manufacturing)","level":2,"score":0.41990000009536743},{"id":"https://openalex.org/C42058472","wikidata":"https://www.wikidata.org/wiki/Q810214","display_name":"Base (topology)","level":2,"score":0.40720000863075256},{"id":"https://openalex.org/C89611455","wikidata":"https://www.wikidata.org/wiki/Q6804646","display_name":"Mechanism (biology)","level":2,"score":0.3982999920845032},{"id":"https://openalex.org/C194387892","wikidata":"https://www.wikidata.org/wiki/Q1747770","display_name":"Stochastic optimization","level":2,"score":0.39239999651908875},{"id":"https://openalex.org/C100776233","wikidata":"https://www.wikidata.org/wiki/Q2532492","display_name":"Bridge (graph theory)","level":2,"score":0.38589999079704285},{"id":"https://openalex.org/C2776760102","wikidata":"https://www.wikidata.org/wiki/Q5139990","display_name":"Code (set theory)","level":3,"score":0.3792000114917755},{"id":"https://openalex.org/C2775924081","wikidata":"https://www.wikidata.org/wiki/Q55608371","display_name":"Control (management)","level":2,"score":0.37720000743865967},{"id":"https://openalex.org/C2778112365","wikidata":"https://www.wikidata.org/wiki/Q3511065","display_name":"Sequence (biology)","level":2,"score":0.3490000069141388},{"id":"https://openalex.org/C97541855","wikidata":"https://www.wikidata.org/wiki/Q830687","display_name":"Reinforcement learning","level":2,"score":0.3197999894618988},{"id":"https://openalex.org/C26517878","wikidata":"https://www.wikidata.org/wiki/Q228039","display_name":"Key (lock)","level":2,"score":0.3057999908924103},{"id":"https://openalex.org/C137836250","wikidata":"https://www.wikidata.org/wiki/Q984063","display_name":"Optimization problem","level":2,"score":0.2888999879360199},{"id":"https://openalex.org/C112972136","wikidata":"https://www.wikidata.org/wiki/Q7595718","display_name":"Stability (learning theory)","level":2,"score":0.2842000126838684},{"id":"https://openalex.org/C8272713","wikidata":"https://www.wikidata.org/wiki/Q176737","display_name":"Stochastic process","level":2,"score":0.2808000147342682},{"id":"https://openalex.org/C137631369","wikidata":"https://www.wikidata.org/wiki/Q7617831","display_name":"Stochastic programming","level":2,"score":0.2694999873638153},{"id":"https://openalex.org/C127491075","wikidata":"https://www.wikidata.org/wiki/Q7617825","display_name":"Stochastic modelling","level":2,"score":0.2603999972343445},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.25429999828338623}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2606.17526","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.17526","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":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.17526","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.17526","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":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":{"Efficient":[0],"optimization":[1],"is":[2,27,166],"essential":[3],"for":[4,40,104,151],"training":[5,160],"large":[6],"language":[7],"models.":[8,163],"Although":[9],"intra-layer":[10,153],"selective":[11,41,154],"updates":[12],"have":[13],"been":[14],"explored,":[15],"a":[16,37,53,71,144],"general":[17],"mechanism":[18,39],"that":[19,126],"enables":[20],"fine-grained":[21],"control":[22],"while":[23,62],"ensuring":[24],"convergence":[25,102],"guarantees":[26,103],"still":[28],"lacking.":[29],"To":[30],"bridge":[31],"this":[32],"gap,":[33],"we":[34,99],"propose":[35],"\\textbf{MGUP},":[36],"novel":[38],"updates.":[42],"\\textbf{MGUP}":[43,75],"augments":[44],"standard":[45,97],"momentum-based":[46],"optimizers":[47,79,129],"by":[48],"applying":[49,63],"larger":[50],"step-sizes":[51,66],"to":[52,67,137],"selected":[54],"fixed":[55],"proportion":[56],"of":[57,161],"parameters":[58],"in":[59,109],"each":[60],"iteration,":[61],"smaller,":[64],"non-zero":[65],"the":[68,159],"rest.":[69],"As":[70],"nearly":[72],"{plug-and-play}":[73],"module,":[74],"seamlessly":[76],"integrates":[77],"with":[78],"such":[80,90],"as":[81,91],"AdamW,":[82],"Lion,":[83],"and":[84,94,122,147,157],"Muon.":[85],"This":[86],"yields":[87],"powerful":[88],"variants":[89],"\\textbf{MGUP-AdamW},":[92],"\\textbf{MGUP-Lion},":[93],"\\textbf{MGUP-Muon}.":[95],"Under":[96],"assumptions,":[98],"provide":[100],"theoretical":[101],"\\textbf{MGUP-AdamW}":[105],"(without":[106],"weight":[107],"decay)":[108],"stochastic":[110],"optimization.":[111],"Extensive":[112],"experiments":[113],"across":[114],"diverse":[115],"tasks,":[116],"including":[117],"MAE":[118],"pretraining,":[119,121],"LLM":[120],"downstream":[123],"fine-tuning,":[124],"demonstrate":[125],"our":[127],"\\textbf{MGUP}-enhanced":[128],"achieve":[130],"superior":[131],"or":[132],"more":[133],"stable":[134],"performance":[135],"compared":[136],"their":[138],"original":[139],"base":[140],"optimizers.":[141],"We":[142],"offer":[143],"principled,":[145],"versatile,":[146],"theoretically":[148],"grounded":[149],"strategy":[150],"efficient":[152],"updates,":[155],"accelerating":[156],"stabilizing":[158],"large-scale":[162],"The":[164],"code":[165],"publicly":[167],"available":[168],"at":[169],"https://github.com/MaeChd/MGUP.":[170]},"counts_by_year":[],"updated_date":"2026-07-28T07:46:37.118299","created_date":"2026-06-18T00:00:00"}