Seongwan Kim
2026
Memory-Efficient Structured Backpropagation for On-Device LLM Fine-Tuning
JuneYoung Park | Yuri Hong | Seongwan Kim | Jaeho Lee
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (ACL 2026)
JuneYoung Park | Yuri Hong | Seongwan Kim | Jaeho Lee
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (ACL 2026)
On-device fine-tuning enables privacy-preserving personalization of large language models, but mobile devices impose severe memory constraints, typically 6–12GB shared across all workloads. Existing approaches force a trade-off between exact gradients with high memory (MeBP) and low memory with noisy estimates (MeZO). We propose Memory-efficient Structured Backpropagation (MeSP), which bridges this gap by manually deriving backward passes that exploit LoRA’s low-rank structure. Our key insight is that the intermediate projection h = xA can be recomputed during backward at minimal cost since rank r ≪ din, eliminating the need to store it. MeSP achieves 49% average memory reduction compared to MeBP on Qwen2.5 models (0.5B–3B) while computing mathematically identical gradients. Our analysis also reveals that MeZO’s gradient estimates show near-zero correlation with true gradients (cosine similarity ≈0.001), explaining its slow convergence. MeSP reduces peak memory from 361MB to 136MB for Qwen2.5-0.5B, enabling fine-tuning scenarios previously infeasible on memory-constrained devices.
2025
Riemannian Optimization for LoRA on the Stiefel Manifold
JuneYoung Park | Minjae Kang | Seongbae Lee | Haegang Lee | Seongwan Kim | Jaeho Lee
Findings of the Association for Computational Linguistics: EMNLP 2025
JuneYoung Park | Minjae Kang | Seongbae Lee | Haegang Lee | Seongwan Kim | Jaeho Lee
Findings of the Association for Computational Linguistics: EMNLP 2025
While powerful, large language models (LLMs) present significant fine-tuning challenges due to their size. Parameter-efficient fine-tuning (PEFT) methods like LoRA provide solutions, yet suffer from critical optimizer inefficiencies; notably basis redundancy in LoRA’s B matrix when using AdamW, which fundamentally limits performance. We address this by optimizing the B matrix on the Stiefel manifold, imposing explicit orthogonality constraints that achieve near-perfect orthogonality and full effective rank. This geometric approach dramatically enhances parameter efficiency and representational capacity. Our Stiefel optimizer consistently outperforms AdamW across benchmarks with both LoRA and DoRA, demonstrating that geometric constraints are the key to unlocking LoRA’s full potential for effective LLM fine-tuning.