{"id":"https://openalex.org/W7148703782","doi":"https://doi.org/10.48550/arxiv.2604.01472","title":"The Newton-Muon Optimizer","display_name":"The Newton-Muon Optimizer","publication_year":2026,"publication_date":"2026-04-01","ids":{"openalex":"https://openalex.org/W7148703782","doi":"https://doi.org/10.48550/arxiv.2604.01472"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2604.01472","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.01472","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":"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.2604.01472","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5101264788","display_name":"Zhehang Du","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Du, Zhehang","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5080575294","display_name":"Weijie Su","orcid":"https://orcid.org/0000-0003-1787-1219"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Su, Weijie","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/T12579","display_name":"Muon and positron interactions and applications","score":0.20479999482631683,"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"}},"topics":[{"id":"https://openalex.org/T12579","display_name":"Muon and positron interactions and applications","score":0.20479999482631683,"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/T13650","display_name":"Computational Physics and Python Applications","score":0.11949999630451202,"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/T11948","display_name":"Machine Learning in Materials Science","score":0.048700001090765,"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/orthogonalization","display_name":"Orthogonalization","score":0.5803999900817871},{"id":"https://openalex.org/keywords/hessian-matrix","display_name":"Hessian matrix","score":0.4627000093460083},{"id":"https://openalex.org/keywords/matrix","display_name":"Matrix (chemical analysis)","score":0.43790000677108765},{"id":"https://openalex.org/keywords/diagonal","display_name":"Diagonal","score":0.39160001277923584},{"id":"https://openalex.org/keywords/surrogate-model","display_name":"Surrogate model","score":0.3885999917984009},{"id":"https://openalex.org/keywords/quadratic-equation","display_name":"Quadratic equation","score":0.3752000033855438},{"id":"https://openalex.org/keywords/function","display_name":"Function (biology)","score":0.3695000112056732},{"id":"https://openalex.org/keywords/isotropy","display_name":"Isotropy","score":0.358599990606308}],"concepts":[{"id":"https://openalex.org/C47559304","wikidata":"https://www.wikidata.org/wiki/Q1702189","display_name":"Orthogonalization","level":2,"score":0.5803999900817871},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.48910000920295715},{"id":"https://openalex.org/C126255220","wikidata":"https://www.wikidata.org/wiki/Q141495","display_name":"Mathematical optimization","level":1,"score":0.4740999937057495},{"id":"https://openalex.org/C203616005","wikidata":"https://www.wikidata.org/wiki/Q620495","display_name":"Hessian matrix","level":2,"score":0.4627000093460083},{"id":"https://openalex.org/C106487976","wikidata":"https://www.wikidata.org/wiki/Q685816","display_name":"Matrix (chemical analysis)","level":2,"score":0.43790000677108765},{"id":"https://openalex.org/C28826006","wikidata":"https://www.wikidata.org/wiki/Q33521","display_name":"Applied mathematics","level":1,"score":0.4074999988079071},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.4041000008583069},{"id":"https://openalex.org/C130367717","wikidata":"https://www.wikidata.org/wiki/Q189791","display_name":"Diagonal","level":2,"score":0.39160001277923584},{"id":"https://openalex.org/C131675550","wikidata":"https://www.wikidata.org/wiki/Q7646884","display_name":"Surrogate model","level":2,"score":0.3885999917984009},{"id":"https://openalex.org/C129844170","wikidata":"https://www.wikidata.org/wiki/Q41299","display_name":"Quadratic equation","level":2,"score":0.3752000033855438},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.37139999866485596},{"id":"https://openalex.org/C14036430","wikidata":"https://www.wikidata.org/wiki/Q3736076","display_name":"Function (biology)","level":2,"score":0.3695000112056732},{"id":"https://openalex.org/C184050105","wikidata":"https://www.wikidata.org/wiki/Q273163","display_name":"Isotropy","level":2,"score":0.358599990606308},{"id":"https://openalex.org/C22789450","wikidata":"https://www.wikidata.org/wiki/Q420904","display_name":"Singular value decomposition","level":2,"score":0.3540000021457672},{"id":"https://openalex.org/C61797465","wikidata":"https://www.wikidata.org/wiki/Q1188986","display_name":"Term (time)","level":2,"score":0.34610000252723694},{"id":"https://openalex.org/C195065555","wikidata":"https://www.wikidata.org/wiki/Q214881","display_name":"Curvature","level":2,"score":0.32499998807907104},{"id":"https://openalex.org/C137836250","wikidata":"https://www.wikidata.org/wiki/Q984063","display_name":"Optimization problem","level":2,"score":0.32100000977516174},{"id":"https://openalex.org/C60718061","wikidata":"https://www.wikidata.org/wiki/Q1414747","display_name":"Momentum (technical analysis)","level":2,"score":0.3050999939441681},{"id":"https://openalex.org/C109282560","wikidata":"https://www.wikidata.org/wiki/Q4166054","display_name":"Singular value","level":3,"score":0.29190000891685486},{"id":"https://openalex.org/C203233044","wikidata":"https://www.wikidata.org/wiki/Q5264358","display_name":"Design matrix","level":3,"score":0.29100000858306885},{"id":"https://openalex.org/C84507854","wikidata":"https://www.wikidata.org/wiki/Q952222","display_name":"Diagonally dominant matrix","level":3,"score":0.2896000146865845}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2604.01472","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.01472","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":"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.2604.01472","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2604.01472","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":"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":{"The":[0],"Muon":[1,166,199],"optimizer":[2],"has":[3],"received":[4],"considerable":[5],"attention":[6],"for":[7,200],"its":[8,21],"strong":[9],"performance":[10],"in":[11,107,209],"training":[12,217],"large":[13],"language":[14],"models,":[15],"yet":[16],"the":[17,42,55,59,64,67,74,85,94,100,117,121,140,177,182,191,205],"design":[18,43],"principle":[19],"behind":[20],"matrix-gradient":[22],"orthogonalization":[23],"remains":[24],"largely":[25],"elusive.":[26],"In":[27,54],"this":[28,105],"paper,":[29],"we":[30,119,158],"introduce":[31],"a":[32,51,70,77,112,148,188],"surrogate":[33,65,106],"model":[34],"that":[35,98,164,175],"not":[36],"only":[37,82],"sheds":[38],"new":[39,52,154],"light":[40],"on":[41,116,187],"of":[44,61,73,190],"Muon,":[45],"but":[46],"more":[47],"importantly":[48],"leads":[49],"to":[50,76,126,160],"optimizer.":[53],"same":[56],"spirit":[57],"as":[58,69,161,170],"derivation":[60],"Newton's":[62],"method,":[63,156],"approximates":[66],"loss":[68,208],"quadratic":[71],"function":[72],"perturbation":[75],"weight":[78,129],"matrix":[79,91,96],"$W$":[80],"using":[81,198],"three":[83],"matrices:":[84],"gradient":[86],"$G$,":[87],"an":[88,171],"output-space":[89],"curvature":[90],"$H$,":[92],"and":[93,110,128,143,214],"data":[95],"$Z$":[97],"stacks":[99],"layer":[101],"inputs.":[102],"By":[103],"minimizing":[104],"one":[108],"step":[109],"adopting":[111],"certain":[113],"isotropic":[114],"assumption":[115],"weights,":[118],"obtain":[120],"closed-form":[122],"update":[123],"rule":[124],"(up":[125],"momentum":[127],"decay)":[130],"$W":[131],"\\leftarrow":[132],"W":[133],"-":[134],"\u03b7\\cdot":[135],"\\mathrm{msgn}(G(ZZ^\\top)^{-1})$,":[136],"where":[137],"$\u03b7$":[138],"is":[139,147],"learning":[141],"rate":[142],"$\\mathrm{msgn}(X)=UV^\\top$":[144],"if":[145],"$X=USV^\\top$":[146],"compact":[149],"singular":[150],"value":[151],"decomposition.":[152],"This":[153],"optimization":[155],"which":[157],"refer":[159],"Newton-Muon,":[162],"shows":[163],"standard":[165],"can":[167],"be":[168],"interpreted":[169],"implicit":[172],"Newton-type":[173],"method":[174],"neglects":[176],"right":[178],"preconditioning":[179],"induced":[180],"by":[181,219],"input":[183],"second":[184],"moment.":[185],"Empirically,":[186],"reproduction":[189],"earliest":[192],"publicly":[193],"released":[194],"Modded-NanoGPT":[195],"speedrun":[196],"configuration":[197],"GPT-2":[201],"pretraining,":[202],"Newton-Muon":[203],"reaches":[204],"target":[206],"validation":[207],"6\\%":[210],"fewer":[211],"iteration":[212],"steps":[213],"reduces":[215],"wall-clock":[216],"time":[218],"about":[220],"4\\%.":[221]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-04-04T00:00:00"}
