Wenhao Li
Author directoryOther people with similar names: Wenhao Li, Wenhao Li
Unverified author pages with similar names: Wenhao Li
2025
Training Long-Context LLMs Efficiently via Chunk-wise Optimization
Wenhao Li | Yuxin Zhang | Gen Luo | Daohai Yu | Rongrong Ji
Findings of the Association for Computational Linguistics: ACL 2025
Wenhao Li | Yuxin Zhang | Gen Luo | Daohai Yu | Rongrong Ji
Findings of the Association for Computational Linguistics: ACL 2025
While long-context large language models (LLMs) exhibit remarkable document processing capabilities, their prohibitively high training costs often hinder customized applications. To mitigate this issue, we propose Sequential Chunk-wise Optimization (SeCO), a memory-efficient training paradigm that partitions lengthy inputs into manageable chunks. Each chunk independently constructs its computational graph and performs localized backpropagation, ensuring that only one chunk’s forward activations are stored in memory. Building on SeCO, we further introduce Sparse Chunk-wise Optimization (SpaCO), which reduces computational overhead by selectively propagating gradients to specific chunks and incorporates a carefully designed compensation factor to ensure unbiased gradient estimation. SpaCO decouples the computational cost of backpropagation from the context length, enabling training time to gradually converge to inference time as sequences become longer. Implemented as lightweight training wrappers, both SeCO and SpaCO offer substantial practical benefits. For example, when fine-tuning an 8B model with LoRA on a single RTX 3090 GPU, SeCO expands maximum sequence length from 1K to 16K tokens, while SpaCO demonstrates accelerated training speed—achieving up to 3× faster than SeCO under the same experimental setup. These innovations provide new insights into optimizing long-context models, making them more accessible for practical applications. We have open-sourced the code at https://anonymous.4open.science/r/seco-CCBD.
RECALL: REpresentation-aligned Catastrophic-forgetting ALLeviation via Hierarchical Model Merging
Bowen Wang | Haiyuan Wan | Liwen Shi | Chen Yang | Peng He | Yue Ma | Haochen Han | Wenhao Li | Tiao Tan | Yongjian Li | Fangming Liu | Gong Yifan | Sheng Zhang
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing
Bowen Wang | Haiyuan Wan | Liwen Shi | Chen Yang | Peng He | Yue Ma | Haochen Han | Wenhao Li | Tiao Tan | Yongjian Li | Fangming Liu | Gong Yifan | Sheng Zhang
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing
We unveil that internal representations in large language models (LLMs) serve as reliable proxies of learned knowledge, and propose RECALL, a novel representation-aware model merging framework for continual learning without access to historical data. RECALL computes inter-model similarity from layer-wise hidden representations over clustered typical samples, and performs adaptive, hierarchical parameter fusion to align knowledge across models. This design enables the preservation of domain-general features in shallow layers while allowing task-specific adaptation in deeper layers. Unlike prior methods that require task labels or incur performance trade-offs, RECALL achieves seamless multi-domain integration and strong resistance to catastrophic forgetting. Extensive experiments across five NLP tasks and multiple continual learning scenarios show that RECALL outperforms baselines in both knowledge retention and generalization, providing a scalable and data-free solution for evolving LLMs.