{"id":"https://openalex.org/W4400611281","doi":"https://doi.org/10.48550/arxiv.2407.07972","title":"Deconstructing What Makes a Good Optimizer for Language Models","display_name":"Deconstructing What Makes a Good Optimizer for Language Models","publication_year":2024,"publication_date":"2024-07-10","ids":{"openalex":"https://openalex.org/W4400611281","doi":"https://doi.org/10.48550/arxiv.2407.07972"},"language":"en","primary_location":{"id":"pmh:oai:arXiv.org:2407.07972","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2407.07972","pdf_url":"https://arxiv.org/pdf/2407.07972","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/2407.07972","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","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":"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/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/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":"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/T10181","display_name":"Natural Language Processing Techniques","score":0.91839998960495,"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/T10181","display_name":"Natural Language Processing Techniques","score":0.91839998960495,"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/computer-science","display_name":"Computer science","score":0.5241608619689941},{"id":"https://openalex.org/keywords/linguistics","display_name":"Linguistics","score":0.39054644107818604},{"id":"https://openalex.org/keywords/philosophy","display_name":"Philosophy","score":0.15725472569465637}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.5241608619689941},{"id":"https://openalex.org/C41895202","wikidata":"https://www.wikidata.org/wiki/Q8162","display_name":"Linguistics","level":1,"score":0.39054644107818604},{"id":"https://openalex.org/C138885662","wikidata":"https://www.wikidata.org/wiki/Q5891","display_name":"Philosophy","level":0,"score":0.15725472569465637}],"mesh":[],"locations_count":2,"locations":[{"id":"pmh:oai:arXiv.org:2407.07972","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2407.07972","pdf_url":"https://arxiv.org/pdf/2407.07972","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.2407.07972","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2407.07972","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:2407.07972","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2407.07972","pdf_url":"https://arxiv.org/pdf/2407.07972","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/G4720003262","display_name":null,"funder_award_id":"N00014-22","funder_id":"https://openalex.org/F4320337345","funder_display_name":"Office of Naval Research"},{"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"},{"id":"https://openalex.org/G8876996369","display_name":null,"funder_award_id":"N00014","funder_id":"https://openalex.org/F4320337345","funder_display_name":"Office of Naval Research"}],"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/W4400611281.pdf","grobid_xml":"https://content.openalex.org/works/W4400611281.grobid-xml"},"referenced_works_count":0,"referenced_works":[],"related_works":["https://openalex.org/W4391375266","https://openalex.org/W2748952813","https://openalex.org/W2390279801","https://openalex.org/W2358668433","https://openalex.org/W4396701345","https://openalex.org/W2376932109","https://openalex.org/W2001405890","https://openalex.org/W4396696052","https://openalex.org/W2382290278","https://openalex.org/W4395014643"],"abstract_inverted_index":{"Training":[0],"language":[1,57],"models":[2],"becomes":[3],"increasingly":[4],"expensive":[5],"with":[6],"scale,":[7],"prompting":[8],"numerous":[9],"attempts":[10],"to":[11,27,40,104,139,183,200,224],"improve":[12],"optimization":[13,43],"efficiency.":[14],"Despite":[15],"these":[16,76,148],"efforts,":[17],"the":[18,22,34,53,107,165,185,194,201,209],"Adam":[19,171,179],"optimizer":[20,110],"remains":[21],"most":[23,35],"widely":[24],"used,":[25],"due":[26],"a":[28,60,95,130,175],"prevailing":[29],"view":[30],"that":[31,106],"it":[32],"is":[33],"effective":[36],"approach.":[37],"We":[38],"aim":[39],"compare":[41],"several":[42],"algorithms,":[44],"including":[45],"SGD,":[46,75],"Adafactor,":[47],"Adam,":[48],"Lion,":[49],"and":[50,66,86,120,167,172,212,222],"Sophia":[51],"in":[52,82,88,133,215],"context":[54],"of":[55,62,90,98,109,122,135,154,170,178,193],"autoregressive":[56],"modeling":[58],"across":[59,94],"range":[61,97],"model":[63],"sizes,":[64],"hyperparameters,":[65],"architecture":[67],"variants.":[68],"Our":[69,101],"findings":[70],"indicate":[71],"that,":[72,203],"except":[73],"for":[74,190,219],"algorithms":[77],"all":[78],"perform":[79],"comparably":[80],"both":[81,164,208],"their":[83],"optimal":[84],"performance":[85,136,166,221],"also":[87],"terms":[89,134],"how":[91],"they":[92],"fare":[93],"wide":[96],"hyperparameter":[99,140,168],"choices.":[100],"results":[102],"suggest":[103],"practitioners":[105],"choice":[108],"can":[111],"be":[112],"guided":[113],"by":[114],"practical":[115],"considerations":[116],"like":[117],"memory":[118],"constraints":[119],"ease":[121],"implementation,":[123],"as":[124,129],"no":[125],"single":[126],"algorithm":[127],"emerged":[128],"clear":[131],"winner":[132],"or":[137],"stability":[138,169,223],"misspecification.":[141],"Given":[142],"our":[143],"findings,":[144],"we":[145,161,181],"further":[146],"dissect":[147],"approaches,":[149],"examining":[150],"two":[151],"simplified":[152],"versions":[153],"Adam:":[155],"a)":[156],"signed":[157],"momentum":[158],"(Signum)":[159],"which":[160,180],"see":[162],"recovers":[163],"b)":[173],"Adalayer,":[174],"layerwise":[176],"variant":[177],"introduce":[182],"study":[184],"impact":[186],"on":[187,207],"Adam's":[188],"preconditioning":[189],"different":[191],"layers":[192],"network.":[195],"Examining":[196],"Adalayer":[197],"leads":[198],"us":[199],"conclusion":[202],"perhaps":[204],"surprisingly,":[205],"adaptivity":[206],"last":[210],"layer":[211],"LayerNorm":[213],"parameters":[214],"particular":[216],"are":[217],"necessary":[218],"retaining":[220],"learning":[225],"rate.":[226]},"counts_by_year":[{"year":2025,"cited_by_count":2}],"updated_date":"2026-08-26T07:47:46.906454","created_date":"2025-10-10T00:00:00"}