Joint Continual Learning of Local Language Models and Cloud Offloading Decisions with Budget Constraints
Not on the map. Why?
Abstract (arXiv)
Locally deployed Small Language Models (SLMs) must continually support diverse tasks under strict memory and computation constraints, making selective reliance on cloud Large Language Models (LLMs) unavoidable. Regulating cloud assistance during continual learning is challenging, as naive reward-based reinforcement learning often yields unstable offloading behavior and exacerbates catastrophic forgetting as task distributions shift. We propose DA-GRPO, a dual-advantage extension of Group Relative Policy Optimization that incorporates cloud-usage constraints directly into advantage computation, avoiding fixed reward shaping and external routing models. This design enables the local model to jointly learn task competence and collaboration behavior, allowing cloud requests to emerge naturally during post-training while respecting a prescribed assistance budget. Experiments on mathematical reasoning and code generation benchmarks show that DA-GRPO improves post-switch accuracy, substantially reduces forgetting, and maintains stable cloud usage compared to prior collaborative and routing-based approaches.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 25 of its 28 references that are in the map. First posted on arXiv 29 Jan 2026.
Builds on
25 of 28 references.
References · 25
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
- 2020Measuring Massive Multitask Language Understandingcited by 7,768
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2023Qwen Technical Reportcited by 3,273
- 2025Understanding R1-Zero-Like Training: A Critical Perspectivecited by 1,276
- 2017Constrained Policy Optimizationcited by 928
- 2023FrugalGPT: How to Use Large Language Models While Reducing Cost and Improving Performancecited by 791
- 2024RouteLLM: Learning to Route LLMs with Preference Datacited by 633
- 2024TinyLlama: An Open-Source Small Language Modelcited by 604
- 2018Reward Constrained Policy Optimizationcited by 397
- 2024Hybrid LLM: Cost-Efficient and Quality-Aware Query Routingcited by 354
13 more
- 2023Iterative Preference Learning from Human Feedback: Bridging Theory and Practice for RLHF under KL-Constraintcited by 346
- 2022Teaching Small Language Models to Reasoncited by 279
- 2025Small Language Models are the Future of Agentic AIcited by 254
- 2024MobileLLM: Optimizing Sub-billion Parameter Language Models for On-Device Use Casescited by 216
- 2023TACO: Topics in Algorithmic COde generation datasetcited by 150
- 2024LLaVA-Phi: Efficient Multi-Modal Assistant with Small Language Modelcited by 128
- 2024On-Device Language Models: A Comprehensive Reviewcited by 101
- 2022Enforcing Hard Constraints with Soft Barriers: Safe Reinforcement Learning in Unknown Stochastic Environmentscited by 54
- 2025GVPO: Group Variance Policy Optimization for Large Language Model Post-Trainingcited by 32
- 2018Resource Constrained Deep Reinforcement Learningcited by 23
- 2025RePIC: Reinforced Post-Training for Personalizing Multi-Modal Language Modelscited by 13
- 2025Explicit Preference Optimization: No Need for an Implicit Reward Modelcited by 6
- 2025Concise Reasoning in the Lens of Lagrangian Optimizationcited by 3
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2025Bridging On-Device and Cloud LLMs for Collaborative Reasoning: A Unified Methodology for Local Routing and Post-Training
- 2025Group-Relative REINFORCE Is Secretly an Off-Policy Algorithm: Demystifying Some Myths About GRPO and Its Friends
- 2026GRPO Training Dynamics for Small Language Models
- 2026CoDistill-GRPO: A Co-Distillation Recipe for Efficient Group Relative Policy Optimization
- 2026Hera: Learning Long-Horizon Coordination for Device-Cloud Collaborative LLM Agents
- 2025Large-Small Model Collaboration for Enhancing Edge-Deployed Small Models
- 2024AdaSwitch: Adaptive Switching between Small and Large Agents for Effective Cloud-Local Collaborative Learning
- 2025Collaborative Inference and Learning between Edge SLMs and Cloud LLMs: A Survey of Algorithms, Execution, and Open Challenges