ChunkWise LoRA: Adaptive Sequence Partitioning for Memory-Efficient Low-Rank Adaptation and Accelerated LLM Inference
Not on the map. Why?
Abstract (arXiv)
Recent advances in low-rank adaptation (LoRA) have enabled efficient fine-tuning of large language models (LLMs) with minimal additional parameters. However, existing LoRA methods apply static rank configurations uniformly across all input tokens, ignoring variation in token complexity and computational requirements. In this work, we propose ChunkWise LoRA, a dynamic and adaptive approach that partitions sequences into variable-length chunks based on token complexity and assigns each chunk a tailored low-rank configuration. Our system introduces a runtime scheduler that estimates token difficulty, performs adaptive chunking, and selects per-chunk LoRA rank and scaling using a rank-ladder mechanism. To preserve output consistency, we further introduce a boundary-safe composition module and integrate policy-driven KV-cache strategies. Experiments on benchmark datasets such as Wikitext-103 and SQuAD demonstrate that ChunkWise LoRA achieves up to 34% lower latency and 38% memory reduction compared to baseline LoRA, while maintaining or improving task performance metrics like BLEU, EM, and perplexity. The proposed framework remains fully compatible with existing transformer architectures and inference frameworks, providing a practical solution for real-world deployment of parameter-efficient LLMs.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 22 of its 23 references that are in the map. First posted on arXiv 28 Jan 2026.
Builds on
22 of 23 references.
References · 22
- 2017Attention Is All You Needcited by 60,361
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
- 2019Parameter-Efficient Transfer Learning for NLPcited by 4,236
- 2021Prefix-Tuning: Optimizing Continuous Prompts for Generationcited by 3,915
- 2021The Power of Scale for Parameter-Efficient Prompt Tuningcited by 3,795
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
- 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,418
- 2020Longformer: The Long-Document Transformercited by 3,413
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
- 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,868
- 2020Big Bird: Transformers for Longer Sequencescited by 1,684
10 more
- 2022Fast Inference from Transformers via Speculative Decodingcited by 1,653
- 2019ZeRO: Memory Optimizations Toward Training Trillion Parameter Modelscited by 1,301
- 2022Token Merging: Your ViT But Fastercited by 1,021
- 2022LLM.int8(): 8-bit Matrix Multiplication for Transformers at Scalecited by 906
- 2020AdapterFusion: Non-Destructive Task Composition for Transfer Learningcited by 788
- 2018Universal Transformerscited by 647
- 2016Adaptive Computation Time for Recurrent Neural Networkscited by 506
- 2020DeeBERT: Dynamic Early Exiting for Accelerating BERT Inferencecited by 332
- 2020FastBERT: a Self-distilling BERT with Adaptive Inference Timecited by 226
- 2024Are Triggers Needed for Document-Level Event Extraction?cited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2024LoRA Land: 310 Fine-tuned LLMs that Rival GPT-4, A Technical Report
- 2025LoRA Is Slower Than You Think
- 2026LoRA-Drop: Temporal LoRA Decoding for Efficient LLM Inference
- 2026Strategic Over-Parameterization for Generalizable Low-Rank Adaptation
- 2025Less is More: Resource-Efficient Low-Rank Adaptation
- 2023S-LoRA: Serving Thousands of Concurrent LoRA Adapters
- 2026Learning Rate Matters: Vanilla LoRA May Suffice for LLM Fine-tuning
- 2026Bayesian-LoRA: Probabilistic Low-Rank Adaptation of Large Language Models