{"id":"https://openalex.org/W7162077425","doi":"https://doi.org/10.48550/arxiv.2605.22297","title":"One LR Doesn't Fit All: Heavy-Tail Guided Layerwise Learning Rates for LLMs","display_name":"One LR Doesn't Fit All: Heavy-Tail Guided Layerwise Learning Rates for LLMs","publication_year":2026,"publication_date":"2026-05-21","ids":{"openalex":"https://openalex.org/W7162077425","doi":"https://doi.org/10.48550/arxiv.2605.22297"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2605.22297","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.22297","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.2605.22297","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5136748910","display_name":"Di He","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"He, Di","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5136469202","display_name":"Songjun Tu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Tu, Songjun","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5136780519","display_name":"Keyu Wang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Keyu","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5136745845","display_name":"Lu Yin","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yin, Lu","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5136771253","display_name":"Shiwei Liu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Liu, Shiwei","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/T10036","display_name":"Advanced Neural Network Applications","score":0.14710000157356262,"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"}},"topics":[{"id":"https://openalex.org/T10036","display_name":"Advanced Neural Network Applications","score":0.14710000157356262,"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/T11948","display_name":"Machine Learning in Materials Science","score":0.1429000049829483,"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"}},{"id":"https://openalex.org/T10181","display_name":"Natural Language Processing Techniques","score":0.1363999992609024,"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/limiting","display_name":"Limiting","score":0.5568000078201294},{"id":"https://openalex.org/keywords/speedup","display_name":"Speedup","score":0.542900025844574},{"id":"https://openalex.org/keywords/convergence","display_name":"Convergence (economics)","score":0.46630001068115234},{"id":"https://openalex.org/keywords/key","display_name":"Key (lock)","score":0.4595000147819519},{"id":"https://openalex.org/keywords/code","display_name":"Code (set theory)","score":0.4138000011444092},{"id":"https://openalex.org/keywords/rate-of-convergence","display_name":"Rate of convergence","score":0.4088999927043915},{"id":"https://openalex.org/keywords/scheme","display_name":"Scheme (mathematics)","score":0.39590001106262207}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.6057000160217285},{"id":"https://openalex.org/C188198153","wikidata":"https://www.wikidata.org/wiki/Q1613840","display_name":"Limiting","level":2,"score":0.5568000078201294},{"id":"https://openalex.org/C68339613","wikidata":"https://www.wikidata.org/wiki/Q1549489","display_name":"Speedup","level":2,"score":0.542900025844574},{"id":"https://openalex.org/C2777303404","wikidata":"https://www.wikidata.org/wiki/Q759757","display_name":"Convergence (economics)","level":2,"score":0.46630001068115234},{"id":"https://openalex.org/C26517878","wikidata":"https://www.wikidata.org/wiki/Q228039","display_name":"Key (lock)","level":2,"score":0.4595000147819519},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.43970000743865967},{"id":"https://openalex.org/C2776760102","wikidata":"https://www.wikidata.org/wiki/Q5139990","display_name":"Code (set theory)","level":3,"score":0.4138000011444092},{"id":"https://openalex.org/C57869625","wikidata":"https://www.wikidata.org/wiki/Q1783502","display_name":"Rate of convergence","level":3,"score":0.4088999927043915},{"id":"https://openalex.org/C77618280","wikidata":"https://www.wikidata.org/wiki/Q1155772","display_name":"Scheme (mathematics)","level":2,"score":0.39590001106262207},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.3953000009059906},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.38109999895095825},{"id":"https://openalex.org/C108583219","wikidata":"https://www.wikidata.org/wiki/Q197536","display_name":"Deep learning","level":2,"score":0.36480000615119934},{"id":"https://openalex.org/C66322947","wikidata":"https://www.wikidata.org/wiki/Q11658","display_name":"Transformer","level":3,"score":0.35350000858306885},{"id":"https://openalex.org/C117220453","wikidata":"https://www.wikidata.org/wiki/Q5172842","display_name":"Correlation","level":2,"score":0.31709998846054077},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.27730000019073486},{"id":"https://openalex.org/C126255220","wikidata":"https://www.wikidata.org/wiki/Q141495","display_name":"Mathematical optimization","level":1,"score":0.27489998936653137},{"id":"https://openalex.org/C2777027219","wikidata":"https://www.wikidata.org/wiki/Q1284190","display_name":"Constant (computer programming)","level":2,"score":0.27410000562667847},{"id":"https://openalex.org/C98045186","wikidata":"https://www.wikidata.org/wiki/Q205663","display_name":"Process (computing)","level":2,"score":0.25839999318122864},{"id":"https://openalex.org/C2778915421","wikidata":"https://www.wikidata.org/wiki/Q3643177","display_name":"Performance improvement","level":2,"score":0.25049999356269836}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2605.22297","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.22297","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.2605.22297","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.22297","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":[{"score":0.4749317169189453,"display_name":"Quality Education","id":"https://metadata.un.org/sdg/4"}],"awards":[],"funders":[],"has_content":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Learning":[0,47],"rate":[1,19],"configuration":[2],"is":[3,64,201,219],"a":[4,16],"fundamental":[5],"aspect":[6],"of":[7,14,27,36,78,158,180,189,199],"modern":[8],"deep":[9],"learning.":[10],"The":[11],"prevailing":[12],"practice":[13],"applying":[15],"uniform":[17,216],"learning":[18,56,92,104,108],"across":[20,118,129],"all":[21],"layers":[22,98],"overlooks":[23],"the":[24,34,73,156,176,215],"structural":[25],"heterogeneity":[26],"Transformers,":[28],"potentially":[29],"limiting":[30],"their":[31],"effectiveness":[32,157],"as":[33],"backbone":[35],"Large":[37],"Language":[38],"Models":[39],"(LLMs).":[40],"In":[41,172],"this":[42,111],"paper,":[43],"we":[44],"introduce":[45],"Layerwise":[46],"Rate":[48],"(LLR),":[49],"an":[50],"adaptive":[51],"scheme":[52],"that":[53,188],"assigns":[54],"distinct":[55],"rates":[57,93,109],"to":[58,82,94,121,134,146,151,163,185,194],"individual":[59],"Transformer":[60],"layers.":[61],"Our":[62],"method":[63],"grounded":[65],"in":[66,110],"Heavy-Tailed":[67],"Self-Regularization":[68],"(HT-SR)":[69],"theory,":[70],"which":[71],"characterizes":[72],"empirical":[74],"spectral":[75],"density":[76],"(ESD)":[77],"weight":[79],"correlation":[80],"matrices":[81],"quantify":[83],"heavy-tailedness.":[84],"Layers":[85],"with":[86,99,149],"weaker":[87],"heavy-tailedness":[88,101],"are":[89],"assigned":[90],"larger":[91],"accelerate":[95],"training,":[96],"while":[97],"stronger":[100],"receive":[102],"smaller":[103],"rates.":[105],"By":[106],"tailoring":[107],"manner,":[112],"LLR":[113,160,200],"promotes":[114],"more":[115],"balanced":[116],"training":[117,153,165],"layers,":[119],"leading":[120],"faster":[122],"convergence":[123],"and":[124,139,141,167,187],"improved":[125],"generalization.":[126],"Extensive":[127],"experiments":[128],"architectures":[130],"ranging":[131],"from":[132,144,183,192,214],"LLaMA":[133],"GPT-nano,":[135],"optimizers":[136],"including":[137],"AdamW":[138],"Muon,":[140],"model":[142],"scales":[143],"60M":[145],"3B":[147,190],"parameters":[148],"up":[150,162],"100B":[152],"tokens":[154],"demonstrate":[155],"LLR.":[159],"achieves":[161],"1.5x":[164],"speedup":[166],"consistently":[168],"outperforms":[169],"uniform-learning-rate":[170],"baselines.":[171],"particular,":[173],"it":[174,206],"improves":[175],"average":[177],"zero-shot":[178],"accuracy":[179],"1B":[181],"models":[182,191],"47.09%":[184],"49.02%,":[186],"48.58%":[193],"50.61%.":[195],"A":[196],"key":[197],"advantage":[198],"its":[202],"low":[203],"tuning":[204],"overhead:":[205],"can":[207],"transfer":[208],"nearly":[209],"optimal":[210],"learning-rate":[211],"settings":[212],"directly":[213],"baseline.":[217],"Code":[218],"available":[220],"at":[221],"https://github.com/hed-ucas/Layer-wise-Learning-Rate.":[222]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-05-23T00:00:00"}