{"id":"https://openalex.org/W4404399462","doi":"https://doi.org/10.48550/arxiv.2410.21676","title":"How Does Critical Batch Size Scale in Pre-training?","display_name":"How Does Critical Batch Size Scale in Pre-training?","publication_year":2024,"publication_date":"2024-10-29","ids":{"openalex":"https://openalex.org/W4404399462","doi":"https://doi.org/10.48550/arxiv.2410.21676"},"language":"en","primary_location":{"id":"pmh:oai:arXiv.org:2410.21676","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2410.21676","pdf_url":"https://arxiv.org/pdf/2410.21676","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":"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/2410.21676","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5101598769","display_name":"Hanlin Zhang","orcid":"https://orcid.org/0000-0001-8869-6863"},"institutions":[],"countries":[],"is_corresponding":true,"raw_author_name":"Zhang, Hanlin","raw_affiliation_strings":[],"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":[],"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":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5025073572","display_name":"J. Wu","orcid":"https://orcid.org/0000-0002-0395-514X"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wu, Jingfeng","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5085848346","display_name":"Difan Zou","orcid":"https://orcid.org/0000-0002-6544-2593"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zou, Difan","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5017544719","display_name":"Udaya Ghai","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Ghai, Udaya","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5112767220","display_name":"Dean P. Foster","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Foster, Dean","raw_affiliation_strings":[],"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":[],"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":8,"corresponding_author_ids":["https://openalex.org/A5101598769"],"corresponding_institution_ids":[],"apc_list":null,"apc_paid":null,"fwci":null,"has_fulltext":true,"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/T11994","display_name":"Human Resource Development and Performance Evaluation","score":0.3418999910354614,"subfield":{"id":"https://openalex.org/subfields/3202","display_name":"Applied Psychology"},"field":{"id":"https://openalex.org/fields/32","display_name":"Psychology"},"domain":{"id":"https://openalex.org/domains/2","display_name":"Social Sciences"}},"topics":[{"id":"https://openalex.org/T11994","display_name":"Human Resource Development and Performance Evaluation","score":0.3418999910354614,"subfield":{"id":"https://openalex.org/subfields/3202","display_name":"Applied Psychology"},"field":{"id":"https://openalex.org/fields/32","display_name":"Psychology"},"domain":{"id":"https://openalex.org/domains/2","display_name":"Social Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/training","display_name":"Training (meteorology)","score":0.6457779407501221},{"id":"https://openalex.org/keywords/scale","display_name":"Scale (ratio)","score":0.5972475409507751},{"id":"https://openalex.org/keywords/computer-science","display_name":"Computer science","score":0.39166587591171265},{"id":"https://openalex.org/keywords/geography","display_name":"Geography","score":0.13056889176368713},{"id":"https://openalex.org/keywords/cartography","display_name":"Cartography","score":0.06819471716880798},{"id":"https://openalex.org/keywords/meteorology","display_name":"Meteorology","score":0.043395668268203735}],"concepts":[{"id":"https://openalex.org/C2777211547","wikidata":"https://www.wikidata.org/wiki/Q17141490","display_name":"Training (meteorology)","level":2,"score":0.6457779407501221},{"id":"https://openalex.org/C2778755073","wikidata":"https://www.wikidata.org/wiki/Q10858537","display_name":"Scale (ratio)","level":2,"score":0.5972475409507751},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.39166587591171265},{"id":"https://openalex.org/C205649164","wikidata":"https://www.wikidata.org/wiki/Q1071","display_name":"Geography","level":0,"score":0.13056889176368713},{"id":"https://openalex.org/C58640448","wikidata":"https://www.wikidata.org/wiki/Q42515","display_name":"Cartography","level":1,"score":0.06819471716880798},{"id":"https://openalex.org/C153294291","wikidata":"https://www.wikidata.org/wiki/Q25261","display_name":"Meteorology","level":1,"score":0.043395668268203735}],"mesh":[],"locations_count":2,"locations":[{"id":"pmh:oai:arXiv.org:2410.21676","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2410.21676","pdf_url":"https://arxiv.org/pdf/2410.21676","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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"text"},{"id":"doi:10.48550/arxiv.2410.21676","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2410.21676","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":"article"}],"best_oa_location":{"id":"pmh:oai:arXiv.org:2410.21676","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2410.21676","pdf_url":"https://arxiv.org/pdf/2410.21676","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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"text"},"sustainable_development_goals":[],"awards":[],"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":false,"pdf":true},"content_urls":{"pdf":"https://content.openalex.org/works/W4404399462.pdf"},"referenced_works_count":0,"referenced_works":[],"related_works":["https://openalex.org/W4391375266","https://openalex.org/W2899084033","https://openalex.org/W2748952813","https://openalex.org/W230091440","https://openalex.org/W2390279801","https://openalex.org/W2233261550","https://openalex.org/W4391913857","https://openalex.org/W2358668433","https://openalex.org/W4396701345","https://openalex.org/W2810751659"],"abstract_inverted_index":{"Training":[0],"large-scale":[1,165],"models":[2],"under":[3],"given":[4],"resources":[5],"requires":[6],"careful":[7,75],"design":[8],"of":[9,17,48,54,77,96,139,142,157],"parallelism":[10,36],"strategies.":[11],"In":[12],"particular,":[13],"the":[14,23,30,67,94,137,155],"efficiency":[15],"notion":[16],"critical":[18],"batch":[19,81],"size":[20,126],"(CBS),":[21],"concerning":[22],"compromise":[24],"between":[25],"time":[26],"and":[27,50,74,84,109,145,161],"compute,":[28],"marks":[29],"threshold":[31],"beyond":[32,167],"which":[33],"greater":[34],"data":[35,110,125],"leads":[37],"to":[38,62,107,112],"diminishing":[39],"returns.":[40],"To":[41],"operationalize":[42],"it,":[43],"we":[44,91,101,133,153],"propose":[45],"a":[46,52,131],"measure":[47],"CBS":[49,121],"pre-train":[51],"series":[53],"auto-regressive":[55],"language":[56],"models,":[57],"ranging":[58],"from":[59],"85":[60],"million":[61],"1.2":[63],"billion":[64],"parameters,":[65],"on":[66,98],"C4":[68],"dataset.":[69],"Through":[70],"extensive":[71],"hyper-parameter":[72,159],"sweeps":[73],"control":[76],"factors":[78],"such":[79],"as":[80],"size,":[82,130],"momentum,":[83],"learning":[85],"rate":[86],"along":[87],"with":[88,105,124],"its":[89],"scheduling,":[90],"systematically":[92],"investigate":[93],"impact":[95],"scale":[97],"CBS.":[99],"Then":[100],"fit":[102],"scaling":[103],"laws":[104],"respect":[106],"model":[108,129],"sizes":[111],"decouple":[113],"their":[114],"effects.":[115],"Overall,":[116],"our":[117],"results":[118],"demonstrate":[119],"that":[120],"scales":[122],"primarily":[123],"rather":[127],"than":[128],"finding":[132],"justify":[134],"theoretically":[135],"through":[136],"analysis":[138],"infinite-width":[140],"limits":[141],"neural":[143],"networks":[144],"infinite-dimensional":[146],"least":[147],"squares":[148],"regression.":[149],"Of":[150],"independent":[151],"interest,":[152],"highlight":[154],"importance":[156],"common":[158],"choices":[160],"strategies":[162],"for":[163],"studying":[164],"pre-training":[166],"fixed":[168],"training":[169],"durations.":[170]},"counts_by_year":[],"updated_date":"2026-04-09T08:11:56.329763","created_date":"2025-10-10T00:00:00"}