Less Is More in Federated Continual Learning: RieSelect for Conflict-Aware Layer Selection in LLMs
Abstract
Lay Summary
Large AI models often need to keep learning from new data on many user devices, such as phones or edge machines, without collecting private data in one place. A common approach is to let each device send its model changes to a central server, but sending more changes is not always better: it can waste communication resources and may even cause the model to forget useful abilities learned earlier. In this work, we find that this problem is often caused by a small number of model parts whose changes conflict with past knowledge. We propose RieSelect, a method that lets each device send only the changes that are both useful for learning new tasks and unlikely to damage old knowledge. It also carefully limits risky changes before they are shared. Experiments show that this selective strategy helps large models learn new tasks while better retaining previous skills, using much less communication than existing methods. This makes continual improvement of large AI models more practical for bandwidth-limited and privacy-sensitive settings.