{"id":"https://openalex.org/W7134934673","doi":"https://doi.org/10.48550/arxiv.2603.09697","title":"Mousse: Rectifying the Geometry of Muon with Curvature-Aware Preconditioning","display_name":"Mousse: Rectifying the Geometry of Muon with Curvature-Aware Preconditioning","publication_year":2026,"publication_date":"2026-03-10","ids":{"openalex":"https://openalex.org/W7134934673","doi":"https://doi.org/10.48550/arxiv.2603.09697"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2603.09697","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2603.09697","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.2603.09697","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5128714816","display_name":"Yechen Zhang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhang, Yechen","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5128742437","display_name":"Shuhao Xing","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Xing, Shuhao","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5102627515","display_name":"Junhao Huang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Huang, Junhao","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5092105570","display_name":"Kai Lv","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Lv, Kai","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5102710198","display_name":"Yunhua Zhou","orcid":"https://orcid.org/0000-0002-6907-7167"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhou, Yunhua","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5125493585","display_name":"Xipeng Qiu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Qiu, Xipeng","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5030290474","display_name":"Qipeng Guo","orcid":"https://orcid.org/0000-0001-7113-651X"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Guo, Qipeng","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5128784222","display_name":"Kai rong Chen","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Chen, Kai","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.10320000350475311,"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.10320000350475311,"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/T12579","display_name":"Muon and positron interactions and applications","score":0.09369999915361404,"subfield":{"id":"https://openalex.org/subfields/2211","display_name":"Mechanics of Materials"},"field":{"id":"https://openalex.org/fields/22","display_name":"Engineering"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T11948","display_name":"Machine Learning in Materials Science","score":0.09269999712705612,"subfield":{"id":"https://openalex.org/subfields/2505","display_name":"Materials Chemistry"},"field":{"id":"https://openalex.org/fields/25","display_name":"Materials Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/gradient-descent","display_name":"Gradient descent","score":0.5741000175476074},{"id":"https://openalex.org/keywords/isotropy","display_name":"Isotropy","score":0.46059998869895935},{"id":"https://openalex.org/keywords/orthogonalization","display_name":"Orthogonalization","score":0.42309999465942383},{"id":"https://openalex.org/keywords/limiting","display_name":"Limiting","score":0.38449999690055847},{"id":"https://openalex.org/keywords/spectral-method","display_name":"Spectral method","score":0.36629998683929443},{"id":"https://openalex.org/keywords/unitary-state","display_name":"Unitary state","score":0.3598000109195709},{"id":"https://openalex.org/keywords/constraint","display_name":"Constraint (computer-aided design)","score":0.3578000068664551},{"id":"https://openalex.org/keywords/manifold","display_name":"Manifold (fluid mechanics)","score":0.35679998993873596},{"id":"https://openalex.org/keywords/leverage","display_name":"Leverage (statistics)","score":0.3465000092983246},{"id":"https://openalex.org/keywords/curvature","display_name":"Curvature","score":0.34599998593330383}],"concepts":[{"id":"https://openalex.org/C153258448","wikidata":"https://www.wikidata.org/wiki/Q1199743","display_name":"Gradient descent","level":3,"score":0.5741000175476074},{"id":"https://openalex.org/C184050105","wikidata":"https://www.wikidata.org/wiki/Q273163","display_name":"Isotropy","level":2,"score":0.46059998869895935},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.4375},{"id":"https://openalex.org/C47559304","wikidata":"https://www.wikidata.org/wiki/Q1702189","display_name":"Orthogonalization","level":2,"score":0.42309999465942383},{"id":"https://openalex.org/C121332964","wikidata":"https://www.wikidata.org/wiki/Q413","display_name":"Physics","level":0,"score":0.39169999957084656},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.39160001277923584},{"id":"https://openalex.org/C188198153","wikidata":"https://www.wikidata.org/wiki/Q1613840","display_name":"Limiting","level":2,"score":0.38449999690055847},{"id":"https://openalex.org/C23463724","wikidata":"https://www.wikidata.org/wiki/Q2308831","display_name":"Spectral method","level":2,"score":0.36629998683929443},{"id":"https://openalex.org/C67820243","wikidata":"https://www.wikidata.org/wiki/Q179164","display_name":"Unitary state","level":2,"score":0.3598000109195709},{"id":"https://openalex.org/C2776036281","wikidata":"https://www.wikidata.org/wiki/Q48769818","display_name":"Constraint (computer-aided design)","level":2,"score":0.3578000068664551},{"id":"https://openalex.org/C529865628","wikidata":"https://www.wikidata.org/wiki/Q1790740","display_name":"Manifold (fluid mechanics)","level":2,"score":0.35679998993873596},{"id":"https://openalex.org/C2524010","wikidata":"https://www.wikidata.org/wiki/Q8087","display_name":"Geometry","level":1,"score":0.3538999855518341},{"id":"https://openalex.org/C153083717","wikidata":"https://www.wikidata.org/wiki/Q6535263","display_name":"Leverage (statistics)","level":2,"score":0.3465000092983246},{"id":"https://openalex.org/C195065555","wikidata":"https://www.wikidata.org/wiki/Q214881","display_name":"Curvature","level":2,"score":0.34599998593330383},{"id":"https://openalex.org/C2780598303","wikidata":"https://www.wikidata.org/wiki/Q65921492","display_name":"Flexibility (engineering)","level":2,"score":0.34220001101493835},{"id":"https://openalex.org/C152822103","wikidata":"https://www.wikidata.org/wiki/Q7575207","display_name":"Spectral shape analysis","level":3,"score":0.3375000059604645},{"id":"https://openalex.org/C191795146","wikidata":"https://www.wikidata.org/wiki/Q3878446","display_name":"Norm (philosophy)","level":2,"score":0.3361000120639801},{"id":"https://openalex.org/C121864883","wikidata":"https://www.wikidata.org/wiki/Q677916","display_name":"Statistical physics","level":1,"score":0.33149999380111694},{"id":"https://openalex.org/C85725439","wikidata":"https://www.wikidata.org/wiki/Q466686","display_name":"Anisotropy","level":2,"score":0.32199999690055847},{"id":"https://openalex.org/C60718061","wikidata":"https://www.wikidata.org/wiki/Q1414747","display_name":"Momentum (technical analysis)","level":2,"score":0.32109999656677246},{"id":"https://openalex.org/C112972136","wikidata":"https://www.wikidata.org/wiki/Q7595718","display_name":"Stability (learning theory)","level":2,"score":0.31360000371932983},{"id":"https://openalex.org/C168110828","wikidata":"https://www.wikidata.org/wiki/Q1331626","display_name":"Spectral density","level":2,"score":0.30480000376701355},{"id":"https://openalex.org/C137800194","wikidata":"https://www.wikidata.org/wiki/Q11713455","display_name":"Interpolation (computer graphics)","level":3,"score":0.3000999987125397},{"id":"https://openalex.org/C612670","wikidata":"https://www.wikidata.org/wiki/Q7616373","display_name":"Stiefel manifold","level":2,"score":0.29789999127388},{"id":"https://openalex.org/C62799726","wikidata":"https://www.wikidata.org/wiki/Q190056","display_name":"Hilbert space","level":2,"score":0.29679998755455017},{"id":"https://openalex.org/C137836250","wikidata":"https://www.wikidata.org/wiki/Q984063","display_name":"Optimization problem","level":2,"score":0.2921000123023987},{"id":"https://openalex.org/C134306372","wikidata":"https://www.wikidata.org/wiki/Q7754","display_name":"Mathematical analysis","level":1,"score":0.29179999232292175},{"id":"https://openalex.org/C175291020","wikidata":"https://www.wikidata.org/wiki/Q1156822","display_name":"Offset (computer science)","level":2,"score":0.29120001196861267},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.29010000824928284},{"id":"https://openalex.org/C135889238","wikidata":"https://www.wikidata.org/wiki/Q46276","display_name":"Kinetic energy","level":2,"score":0.2854999899864197},{"id":"https://openalex.org/C198082294","wikidata":"https://www.wikidata.org/wiki/Q3399648","display_name":"Position (finance)","level":2,"score":0.28119999170303345},{"id":"https://openalex.org/C2776562905","wikidata":"https://www.wikidata.org/wiki/Q306610","display_name":"Hypersphere","level":2,"score":0.2809000015258789},{"id":"https://openalex.org/C156778621","wikidata":"https://www.wikidata.org/wiki/Q1365748","display_name":"Spectrum (functional analysis)","level":2,"score":0.2761000096797943},{"id":"https://openalex.org/C74050887","wikidata":"https://www.wikidata.org/wiki/Q848368","display_name":"Rotation (mathematics)","level":2,"score":0.26570001244544983},{"id":"https://openalex.org/C14961307","wikidata":"https://www.wikidata.org/wiki/Q5377176","display_name":"Energy minimization","level":2,"score":0.26499998569488525},{"id":"https://openalex.org/C147764199","wikidata":"https://www.wikidata.org/wiki/Q6865248","display_name":"Minification","level":2,"score":0.262800008058548},{"id":"https://openalex.org/C158847443","wikidata":"https://www.wikidata.org/wiki/Q1997812","display_name":"Method of steepest descent","level":2,"score":0.25949999690055847},{"id":"https://openalex.org/C55660270","wikidata":"https://www.wikidata.org/wiki/Q5164377","display_name":"Constrained optimization","level":2,"score":0.2587999999523163},{"id":"https://openalex.org/C126255220","wikidata":"https://www.wikidata.org/wiki/Q141495","display_name":"Mathematical optimization","level":1,"score":0.25529998540878296},{"id":"https://openalex.org/C111335779","wikidata":"https://www.wikidata.org/wiki/Q3454686","display_name":"Reduction (mathematics)","level":2,"score":0.2517000138759613}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2603.09697","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2603.09697","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.2603.09697","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2603.09697","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":{"Recent":[0],"advances":[1],"in":[2,72,79,124,190],"spectral":[3,35,103,145],"optimization,":[4],"notably":[5],"Muon,":[6,185],"have":[7],"demonstrated":[8],"that":[9,43,97,181],"constraining":[10],"update":[11,36,158],"steps":[12,192],"to":[13,59,118,143,177],"the":[14,54,99,106,119,141,156,162,166],"Stiefel":[15],"manifold":[16],"can":[17],"significantly":[18],"accelerate":[19],"training":[20,191],"and":[21,63],"improve":[22],"generalization.":[23],"However,":[24],"Muon":[25,68],"implicitly":[26],"assumes":[27],"an":[28,151],"isotropic":[29],"optimization":[30],"landscape,":[31],"enforcing":[32],"a":[33,94,125,144],"uniform":[34],"norm":[37],"across":[38,171],"all":[39],"eigen-directions.":[40],"We":[41],"argue":[42],"this":[44,83],"\"egalitarian\"":[45],"constraint":[46],"is":[47,57,159],"suboptimal":[48],"for":[49],"Deep":[50],"Neural":[51],"Networks,":[52],"where":[53,155],"curvature":[55],"spectrum":[56],"known":[58],"be":[60],"highly":[61],"heavy-tailed":[62],"ill-conditioned.":[64],"In":[65,82],"such":[66],"landscapes,":[67],"risks":[69],"amplifying":[70],"instabilities":[71],"high-curvature":[73],"directions":[74],"while":[75],"limiting":[76],"necessary":[77],"progress":[78],"flat":[80],"directions.":[81],"work,":[84],"we":[85,137],"propose":[86],"\\textbf{Mousse}":[87],"(\\textbf{M}uon":[88],"\\textbf{O}ptimization":[89],"\\textbf{U}tilizing":[90],"\\textbf{S}hampoo's":[91],"\\textbf{S}tructural":[92],"\\textbf{E}stimation),":[93],"novel":[95],"optimizer":[96],"reconciles":[98],"structural":[100],"stability":[101],"of":[102,109,113,165],"methods":[104],"with":[105,193],"geometric":[107],"adaptivity":[108],"second-order":[110],"preconditioning.":[111],"Instead":[112],"applying":[114],"Newton-Schulz":[115],"orthogonalization":[116],"directly":[117],"momentum":[120],"matrix,":[121],"Mousse":[122,139,182],"operates":[123],"whitened":[126,167],"coordinate":[127],"system":[128],"induced":[129],"by":[130,150],"Kronecker-factored":[131],"statistics":[132],"(derived":[133],"from":[134,175],"Shampoo).":[135],"Mathematically,":[136],"formulate":[138],"as":[140],"solution":[142],"steepest":[146],"descent":[147],"problem":[148],"constrained":[149],"anisotropic":[152],"trust":[153],"region,":[154],"optimal":[157],"derived":[160],"via":[161],"polar":[163],"decomposition":[164],"gradient.":[168],"Empirical":[169],"results":[170],"language":[172],"models":[173],"ranging":[174],"160M":[176],"800M":[178],"parameters":[179],"demonstrate":[180],"consistently":[183],"outperforms":[184],"achieving":[186],"around":[187],"$\\sim$12\\%":[188],"reduction":[189],"negligible":[194],"computational":[195],"overhead.":[196]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-03-12T00:00:00"}