{"id":"https://openalex.org/W7131837684","doi":"https://doi.org/10.48550/arxiv.2602.22681","title":"Accelerating LLM Pre-Training through Flat-Direction Dynamics Enhancement","display_name":"Accelerating LLM Pre-Training through Flat-Direction Dynamics Enhancement","publication_year":2026,"publication_date":"2026-02-26","ids":{"openalex":"https://openalex.org/W7131837684","doi":"https://doi.org/10.48550/arxiv.2602.22681"},"language":null,"primary_location":{"id":"pmh:doi:10.48550/arxiv.2602.22681","is_oa":true,"landing_page_url":null,"pdf_url":null,"source":{"id":"https://openalex.org/S4406922384","display_name":"Open MIND","issn_l":null,"issn":null,"is_oa":false,"is_in_doaj":false,"is_core":false,"host_organization":null,"host_organization_name":null,"host_organization_lineage":[],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Article"},"type":"preprint","indexed_in":["datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":null,"any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5042734572","display_name":"Shuchen Zhu","orcid":"https://orcid.org/0000-0002-1240-2002"},"institutions":[],"countries":[],"is_corresponding":true,"raw_author_name":"Zhu, Shuchen","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5121322693","display_name":"Rizhen Hu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Hu, Rizhen","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5041887782","display_name":"Mingze Wang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Mingze","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5113859272","display_name":"Mou Sun","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Sun, Mou","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5127467895","display_name":"Xue Wang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Xue","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5127367795","display_name":"Kun Yuan","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yuan, Kun","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5114744021","display_name":"ZaiWen Wen","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wen, Zaiwen","raw_affiliation_strings":[],"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":7,"corresponding_author_ids":["https://openalex.org/A5042734572"],"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/T13650","display_name":"Computational Physics and Python Applications","score":0.21570000052452087,"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/T13650","display_name":"Computational Physics and Python Applications","score":0.21570000052452087,"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.10670000314712524,"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/T12579","display_name":"Muon and positron interactions and applications","score":0.05299999937415123,"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"}}],"keywords":[{"id":"https://openalex.org/keywords/leverage","display_name":"Leverage (statistics)","score":0.6105999946594238},{"id":"https://openalex.org/keywords/hessian-matrix","display_name":"Hessian matrix","score":0.6017000079154968},{"id":"https://openalex.org/keywords/curvature","display_name":"Curvature","score":0.5105000138282776},{"id":"https://openalex.org/keywords/acceleration","display_name":"Acceleration","score":0.4945000112056732},{"id":"https://openalex.org/keywords/code","display_name":"Code (set theory)","score":0.4189999997615814},{"id":"https://openalex.org/keywords/isotropy","display_name":"Isotropy","score":0.41440001130104065},{"id":"https://openalex.org/keywords/momentum","display_name":"Momentum (technical analysis)","score":0.39820000529289246},{"id":"https://openalex.org/keywords/preconditioner","display_name":"Preconditioner","score":0.38440001010894775},{"id":"https://openalex.org/keywords/convergence","display_name":"Convergence (economics)","score":0.35830000042915344}],"concepts":[{"id":"https://openalex.org/C153083717","wikidata":"https://www.wikidata.org/wiki/Q6535263","display_name":"Leverage (statistics)","level":2,"score":0.6105999946594238},{"id":"https://openalex.org/C203616005","wikidata":"https://www.wikidata.org/wiki/Q620495","display_name":"Hessian matrix","level":2,"score":0.6017000079154968},{"id":"https://openalex.org/C195065555","wikidata":"https://www.wikidata.org/wiki/Q214881","display_name":"Curvature","level":2,"score":0.5105000138282776},{"id":"https://openalex.org/C117896860","wikidata":"https://www.wikidata.org/wiki/Q11376","display_name":"Acceleration","level":2,"score":0.4945000112056732},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.48420000076293945},{"id":"https://openalex.org/C2776760102","wikidata":"https://www.wikidata.org/wiki/Q5139990","display_name":"Code (set theory)","level":3,"score":0.4189999997615814},{"id":"https://openalex.org/C184050105","wikidata":"https://www.wikidata.org/wiki/Q273163","display_name":"Isotropy","level":2,"score":0.41440001130104065},{"id":"https://openalex.org/C126255220","wikidata":"https://www.wikidata.org/wiki/Q141495","display_name":"Mathematical optimization","level":1,"score":0.41269999742507935},{"id":"https://openalex.org/C60718061","wikidata":"https://www.wikidata.org/wiki/Q1414747","display_name":"Momentum (technical analysis)","level":2,"score":0.39820000529289246},{"id":"https://openalex.org/C167431342","wikidata":"https://www.wikidata.org/wiki/Q1754327","display_name":"Preconditioner","level":3,"score":0.38440001010894775},{"id":"https://openalex.org/C28826006","wikidata":"https://www.wikidata.org/wiki/Q33521","display_name":"Applied mathematics","level":1,"score":0.365200012922287},{"id":"https://openalex.org/C2777303404","wikidata":"https://www.wikidata.org/wiki/Q759757","display_name":"Convergence (economics)","level":2,"score":0.35830000042915344},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.35260000824928284},{"id":"https://openalex.org/C58581272","wikidata":"https://www.wikidata.org/wiki/Q12741163","display_name":"Workspace","level":3,"score":0.335999995470047},{"id":"https://openalex.org/C121332964","wikidata":"https://www.wikidata.org/wiki/Q413","display_name":"Physics","level":0,"score":0.3181999921798706},{"id":"https://openalex.org/C192939610","wikidata":"https://www.wikidata.org/wiki/Q188444","display_name":"Differential geometry","level":2,"score":0.31220000982284546},{"id":"https://openalex.org/C185798385","wikidata":"https://www.wikidata.org/wiki/Q1161707","display_name":"Benchmark (surveying)","level":2,"score":0.3073999881744385},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.30640000104904175},{"id":"https://openalex.org/C132459708","wikidata":"https://www.wikidata.org/wiki/Q744069","display_name":"Extrapolation","level":2,"score":0.305400013923645},{"id":"https://openalex.org/C85725439","wikidata":"https://www.wikidata.org/wiki/Q466686","display_name":"Anisotropy","level":2,"score":0.30149999260902405},{"id":"https://openalex.org/C48044578","wikidata":"https://www.wikidata.org/wiki/Q727490","display_name":"Scalability","level":2,"score":0.30000001192092896},{"id":"https://openalex.org/C26517878","wikidata":"https://www.wikidata.org/wiki/Q228039","display_name":"Key (lock)","level":2,"score":0.28949999809265137},{"id":"https://openalex.org/C2776359362","wikidata":"https://www.wikidata.org/wiki/Q2145286","display_name":"Representation (politics)","level":3,"score":0.27709999680519104},{"id":"https://openalex.org/C114410712","wikidata":"https://www.wikidata.org/wiki/Q973321","display_name":"Hypersurface","level":2,"score":0.2768999934196472},{"id":"https://openalex.org/C181104567","wikidata":"https://www.wikidata.org/wiki/Q761383","display_name":"Riemannian geometry","level":2,"score":0.2712000012397766},{"id":"https://openalex.org/C2524010","wikidata":"https://www.wikidata.org/wiki/Q8087","display_name":"Geometry","level":1,"score":0.27090001106262207},{"id":"https://openalex.org/C207390915","wikidata":"https://www.wikidata.org/wiki/Q1230525","display_name":"Divergence (linguistics)","level":2,"score":0.2662000060081482},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.262800008058548},{"id":"https://openalex.org/C459310","wikidata":"https://www.wikidata.org/wiki/Q117801","display_name":"Computational science","level":1,"score":0.2615000009536743},{"id":"https://openalex.org/C74650414","wikidata":"https://www.wikidata.org/wiki/Q11397","display_name":"Classical mechanics","level":1,"score":0.25769999623298645},{"id":"https://openalex.org/C134306372","wikidata":"https://www.wikidata.org/wiki/Q7754","display_name":"Mathematical analysis","level":1,"score":0.2554999887943268},{"id":"https://openalex.org/C126780896","wikidata":"https://www.wikidata.org/wiki/Q899871","display_name":"Distortion (music)","level":4,"score":0.2547999918460846},{"id":"https://openalex.org/C39920418","wikidata":"https://www.wikidata.org/wiki/Q11476","display_name":"Kinematics","level":2,"score":0.25459998846054077}],"mesh":[],"locations_count":2,"locations":[{"id":"pmh:doi:10.48550/arxiv.2602.22681","is_oa":true,"landing_page_url":null,"pdf_url":null,"source":{"id":"https://openalex.org/S4406922384","display_name":"Open MIND","issn_l":null,"issn":null,"is_oa":false,"is_in_doaj":false,"is_core":false,"host_organization":null,"host_organization_name":null,"host_organization_lineage":[],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Article"},{"id":"doi:10.48550/arxiv.2602.22681","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2602.22681","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":"article"}],"best_oa_location":{"id":"pmh:doi:10.48550/arxiv.2602.22681","is_oa":true,"landing_page_url":null,"pdf_url":null,"source":{"id":"https://openalex.org/S4406922384","display_name":"Open MIND","issn_l":null,"issn":null,"is_oa":false,"is_in_doaj":false,"is_core":false,"host_organization":null,"host_organization_name":null,"host_organization_lineage":[],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Article"},"sustainable_development_goals":[],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Pre-training":[0],"Large":[1],"Language":[2],"Models":[3],"requires":[4],"immense":[5],"computational":[6],"resources,":[7],"making":[8],"optimizer":[9],"efficiency":[10],"essential.":[11],"The":[12,179],"optimization":[13],"landscape":[14],"is":[15,181],"highly":[16],"anisotropic,":[17],"with":[18],"loss":[19],"reduction":[20],"driven":[21],"predominantly":[22],"by":[23,104,118],"progress":[24],"along":[25,51,57,127,165],"flat":[26,52,128,166],"directions.":[27],"While":[28],"matrix-based":[29],"optimizers":[30],"such":[31],"as":[32,95],"Muon":[33,138],"and":[34,124,139,152],"SOAP":[35,140],"leverage":[36],"fine-grained":[37],"curvature":[38],"information":[39],"to":[40,175],"outperform":[41],"AdamW,":[42],"their":[43],"updates":[44],"tend":[45],"toward":[46],"isotropy":[47],"--":[48],"relatively":[49],"conservative":[50],"directions":[53,167],"yet":[54],"potentially":[55],"aggressive":[56],"sharp":[58],"ones.":[59],"To":[60],"address":[61],"this":[62],"limitation,":[63],"we":[64,107],"first":[65],"establish":[66],"a":[67,86,96,110,172],"unified":[68],"Riemannian":[69,87,97],"Ordinary":[70],"Differential":[71],"Equation":[72],"(ODE)":[73],"framework":[74],"that":[75,89,100,114,133,160],"elucidates":[76],"how":[77],"common":[78],"adaptive":[79],"algorithms":[80],"operate":[81],"synergistically:":[82],"the":[83],"preconditioner":[84],"induces":[85],"geometry":[88],"mitigates":[90],"ill-conditioning,":[91],"while":[92],"momentum":[93],"serves":[94],"damping":[98,122],"term":[99],"promotes":[101],"convergence.":[102],"Guided":[103],"these":[105],"insights,":[106],"propose":[108],"LITE,":[109],"generalized":[111],"acceleration":[112],"strategy":[113],"enhances":[115],"training":[116],"dynamics":[117],"applying":[119],"larger":[120],"Hessian":[121],"coefficients":[123],"learning":[125],"rates":[126],"trajectories.":[129],"Extensive":[130],"experiments":[131],"demonstrate":[132],"LITE":[134,161],"significantly":[135],"accelerates":[136],"both":[137],"across":[141],"diverse":[142],"architectures":[143],"(Dense,":[144],"MoE),":[145],"parameter":[146],"scales":[147],"(130M--1.3B),":[148],"datasets":[149],"(C4,":[150],"Pile),":[151],"learning-rate":[153],"schedules":[154],"(cosine,":[155],"warmup-stable-decay).":[156],"Theoretical":[157],"analysis":[158],"confirms":[159],"facilitates":[162],"faster":[163],"convergence":[164],"in":[168],"anisotropic":[169],"landscapes,":[170],"providing":[171],"principled":[173],"approach":[174],"efficient":[176],"LLM":[177],"pre-training.":[178],"code":[180],"available":[182],"at":[183],"https://github.com/SHUCHENZHU/LITE.":[184]},"counts_by_year":[],"updated_date":"2026-04-04T16:13:02.066488","created_date":"2026-02-28T00:00:00"}