← Search

Xunhao Lai

3 accepted papers

2025

Fira: Can We Achieve Full-rank Training of LLMs Under Low-rank Constraint?

NeurIPS 2025poster

Low-rank training has emerged as a promising approach for reducing memory usage in training Large Language Models (LLMs). Previous methods either rely on decomposing weight matrices (e.g., LoRA), or seek to decompose gradient matrices (e.g., GaLore) to ensure reduced memory consumption. However, bot…

Cited by 0SourcecodeScholar
2025

FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference

ICLR 2025oral

Large language models (LLMs) encounter computational challenges during long-sequence inference, especially in the attention pre-filling phase, where the complexity grows quadratically with the prompt length. Previous efforts to mitigate these challenges have relied on fixed sparse attention patterns…

2025

Model Merging in Pre-training of Large Language Models

NeurIPS 2025poster

Model merging has emerged as a promising technique for enhancing large language models, though its application in large-scale pre-training remains relatively unexplored. In this paper, we present a comprehensive investigation of model merging techniques during the pre-training process. Through exten…

Cited by 0SourceScholar