{"id":"https://openalex.org/W4403707245","doi":"https://doi.org/10.48550/arxiv.2409.11321","title":"SOAP: Improving and Stabilizing Shampoo using Adam","display_name":"SOAP: Improving and Stabilizing Shampoo using Adam","publication_year":2024,"publication_date":"2024-09-17","ids":{"openalex":"https://openalex.org/W4403707245","doi":"https://doi.org/10.48550/arxiv.2409.11321"},"language":"en","primary_location":{"id":"pmh:oai:arXiv.org:2409.11321","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2409.11321","pdf_url":"https://arxiv.org/pdf/2409.11321","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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"text"},"type":"preprint","indexed_in":["arxiv","datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://arxiv.org/pdf/2409.11321","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5102775428","display_name":"Nikhil Vyas","orcid":"https://orcid.org/0000-0002-4055-7693"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Vyas, Nikhil","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5086429101","display_name":"Depen Morwani","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Morwani, Depen","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5088011506","display_name":"Rosie Zhao","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhao, Rosie","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":null,"display_name":"Kwun, Mujin","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Kwun, Mujin","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5018766326","display_name":"Itai Shapira","orcid":"https://orcid.org/0000-0003-1635-8464"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Shapira, Itai","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5037978510","display_name":"David Brandfonbrener","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Brandfonbrener, David","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5099582937","display_name":"Lucas Janson","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Janson, Lucas","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5108381794","display_name":"Sham M. Kakade","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Kakade, Sham","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":true,"cited_by_count":2,"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/T12729","display_name":"Tree Root and Stability Studies","score":0.5296000242233276,"subfield":{"id":"https://openalex.org/subfields/2210","display_name":"Mechanical Engineering"},"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/T12729","display_name":"Tree Root and Stability Studies","score":0.5296000242233276,"subfield":{"id":"https://openalex.org/subfields/2210","display_name":"Mechanical Engineering"},"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/T11349","display_name":"Music Technology and Sound Studies","score":0.5041999816894531,"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/T11665","display_name":"Animal Vocal Communication and Behavior","score":0.49160000681877136,"subfield":{"id":"https://openalex.org/subfields/1309","display_name":"Developmental Biology"},"field":{"id":"https://openalex.org/fields/13","display_name":"Biochemistry, Genetics and Molecular Biology"},"domain":{"id":"https://openalex.org/domains/1","display_name":"Life Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/shampoo","display_name":"Shampoo","score":0.9483025074005127},{"id":"https://openalex.org/keywords/soap","display_name":"SOAP","score":0.6656633615493774},{"id":"https://openalex.org/keywords/computer-science","display_name":"Computer science","score":0.4029749929904938},{"id":"https://openalex.org/keywords/chemistry","display_name":"Chemistry","score":0.18851953744888306},{"id":"https://openalex.org/keywords/world-wide-web","display_name":"World Wide Web","score":0.09053152799606323},{"id":"https://openalex.org/keywords/organic-chemistry","display_name":"Organic chemistry","score":0.03861144185066223}],"concepts":[{"id":"https://openalex.org/C2781171460","wikidata":"https://www.wikidata.org/wiki/Q180204","display_name":"Shampoo","level":2,"score":0.9483025074005127},{"id":"https://openalex.org/C17881449","wikidata":"https://www.wikidata.org/wiki/Q189620","display_name":"SOAP","level":2,"score":0.6656633615493774},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.4029749929904938},{"id":"https://openalex.org/C185592680","wikidata":"https://www.wikidata.org/wiki/Q2329","display_name":"Chemistry","level":0,"score":0.18851953744888306},{"id":"https://openalex.org/C136764020","wikidata":"https://www.wikidata.org/wiki/Q466","display_name":"World Wide Web","level":1,"score":0.09053152799606323},{"id":"https://openalex.org/C178790620","wikidata":"https://www.wikidata.org/wiki/Q11351","display_name":"Organic chemistry","level":1,"score":0.03861144185066223}],"mesh":[],"locations_count":2,"locations":[{"id":"pmh:oai:arXiv.org:2409.11321","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2409.11321","pdf_url":"https://arxiv.org/pdf/2409.11321","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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"text"},{"id":"doi:10.48550/arxiv.2409.11321","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2409.11321","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":"pmh:oai:arXiv.org:2409.11321","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2409.11321","pdf_url":"https://arxiv.org/pdf/2409.11321","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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"text"},"sustainable_development_goals":[],"awards":[{"id":"https://openalex.org/G1264789558","display_name":null,"funder_award_id":"W911NF2010021","funder_id":"https://openalex.org/F4320332180","funder_display_name":"Defense Advanced Research Projects Agency"},{"id":"https://openalex.org/G3521671702","display_name":"A Theory of Learned Representations in Artificial and Natural Neural Networks","funder_award_id":"2134157","funder_id":"https://openalex.org/F4320306076","funder_display_name":"National Science Foundation"},{"id":"https://openalex.org/G5877837315","display_name":null,"funder_award_id":"DE-SC0022199","funder_id":"https://openalex.org/F4320332180","funder_display_name":"Defense Advanced Research Projects Agency"},{"id":"https://openalex.org/G6645399916","display_name":"MACHINE LEARNING FOR ACCELERATED UNDERSTANDING OF DYNAMIC CATALYSIS","funder_award_id":"DESC0022199","funder_id":"https://openalex.org/F4320306084","funder_display_name":"U.S. Department of Energy"},{"id":"https://openalex.org/G826810440","display_name":"AI Institute for Societal Decision Making (AI-SDM)","funder_award_id":"2229881","funder_id":"https://openalex.org/F4320306076","funder_display_name":"National Science Foundation"}],"funders":[{"id":"https://openalex.org/F4320306076","display_name":"National Science Foundation","ror":"https://ror.org/021nxhr62"},{"id":"https://openalex.org/F4320306084","display_name":"U.S. Department of Energy","ror":"https://ror.org/01bj3aw27"},{"id":"https://openalex.org/F4320332180","display_name":"Defense Advanced Research Projects Agency","ror":"https://ror.org/02caytj08"},{"id":"https://openalex.org/F4320337345","display_name":"Office of Naval Research","ror":"https://ror.org/00rk2pe57"}],"has_content":{"grobid_xml":true,"pdf":true},"content_urls":{"pdf":"https://content.openalex.org/works/W4403707245.pdf","grobid_xml":"https://content.openalex.org/works/W4403707245.grobid-xml"},"referenced_works_count":0,"referenced_works":[],"related_works":["https://openalex.org/W4391375266","https://openalex.org/W2899084033","https://openalex.org/W2748952813","https://openalex.org/W2364209800","https://openalex.org/W1521615856","https://openalex.org/W4390921525","https://openalex.org/W2554164948","https://openalex.org/W2262030258","https://openalex.org/W2351848437","https://openalex.org/W2800716326"],"abstract_inverted_index":{"There":[0],"is":[1,68,165,238],"growing":[2],"evidence":[3],"of":[4,7,38,62,76,85,146,209,236],"the":[5,53,74,83,96,107,143,147,156,201,207],"effectiveness":[6],"Shampoo,":[8],"a":[9,46,59,86,171],"higher-order":[10],"preconditioning":[11,181],"method,":[12],"over":[13,212,219],"Adam":[14,63,152,169],"in":[15,73,95,155,170,228],"deep":[16],"learning":[17],"optimization":[18],"tasks.":[19],"However,":[20],"Shampoo's":[21,77,104,116],"drawbacks":[22],"include":[23],"additional":[24,178],"hyperparameters":[25],"and":[26,40,56,88,196,214],"computational":[27,105],"overhead":[28],"when":[29],"compared":[30,183,221,231],"to":[31,70,82,102,113,128,167,184,222,232],"Adam,":[32],"which":[33],"only":[34,176],"updates":[35],"running":[36,71,144,168],"averages":[37],"first-":[39],"second-moment":[41],"quantities.":[42],"This":[43,79],"work":[44],"establishes":[45],"formal":[47],"connection":[48],"between":[49],"Shampoo":[50,67],"(implemented":[51],"with":[52,93,133,194,224],"1/2":[54],"power)":[55],"Adafactor":[57,72],"--":[58,64],"memory-efficient":[60],"approximation":[61],"showing":[65],"that":[66,131],"equivalent":[69,166],"eigenbasis":[75,98],"preconditioner.":[78],"insight":[80],"leads":[81,127],"design":[84],"simpler":[87],"computationally":[89],"efficient":[90],"algorithm:":[91],"$\\textbf{S}$hampo$\\textbf{O}$":[92],"$\\textbf{A}$dam":[94],"$\\textbf{P}$reconditioner's":[97],"(SOAP).":[99],"With":[100],"regards":[101],"improving":[103],"efficiency,":[106],"most":[108],"straightforward":[109],"approach":[110],"would":[111],"be":[112],"simply":[114],"compute":[115],"eigendecomposition":[117],"less":[118],"frequently.":[119],"Unfortunately,":[120],"as":[121,151],"our":[122],"empirical":[123],"results":[124],"show,":[125],"this":[126,134,138],"performance":[129],"degradation":[130,139],"worsens":[132],"frequency.":[135],"SOAP":[136,164,189,205,237],"mitigates":[137],"by":[140,211,218],"continually":[141],"updating":[142],"average":[145],"second":[148],"moment,":[149],"just":[150],"does,":[153],"but":[154],"current":[157],"(slowly":[158],"changing)":[159],"coordinate":[160],"basis.":[161],"Furthermore,":[162],"since":[163],"rotated":[172],"space,":[173],"it":[174],"introduces":[175],"one":[177],"hyperparameter":[179],"(the":[180],"frequency)":[182],"Adam.":[185],"We":[186],"empirically":[187],"evaluate":[188],"on":[190],"language":[191],"model":[192],"pre-training":[193],"360m":[195],"660m":[197],"sized":[198],"models.":[199],"In":[200],"large":[202],"batch":[203],"regime,":[204],"reduces":[206],"number":[208],"iterations":[210],"40%":[213],"wall":[215],"clock":[216],"time":[217],"35%":[220],"AdamW,":[223],"approximately":[225],"20%":[226],"improvements":[227],"both":[229],"metrics":[230],"Shampoo.":[233],"An":[234],"implementation":[235],"available":[239],"at":[240],"https://github.com/nikhilvyas/SOAP.":[241]},"counts_by_year":[{"year":2026,"cited_by_count":1},{"year":2025,"cited_by_count":1}],"updated_date":"2026-08-26T07:47:46.906454","created_date":"2025-10-10T00:00:00"}