arXiv Machine Learning

Ranking and Partitioning of Attention Heads for Sparse and Continual Reinforcement Learning Fine-tuning

The paper investigates how to fine‑tune language models for sparse and continual reinforcement learning by ranking and partitioning attention heads. For sparse learning, heads are ranked by gradient norm to identify a small, effective subset for training. For continual learning, heads are partitioned across tasks to reduce interference and preserve prior knowledge. Empirical results on math, instruction‑following, and coding tasks show that tuning higher‑ranked heads outperforms lower‑ranked ones and matches denser baselines, while head partitioning improves retention without sacrificing adaptation to new tasks.

arXiv Machine Learning
Sep 24

Attention Routing Stabilizes Early: Working-Set Inference for Recurrent Language Models

The paper investigates how attention dynamics evolve across recurrent depth in language models, finding that attention support stabilizes early while hidden states and outputs take longer. It proposes WISE, a training‑free method that uses full attention in early steps and then reuses the discovered sparse working set for later steps, preserving performance on multi‑hop QA tasks. Experiments show that WISE maintains quality up to 2K context, offers measurable speedups, and highlights the importance of recurrent discovery of attention support.

By Ke Wan, Chen Chen
arXiv Machine Learning
Jun 2

Simple Recipe Works: Vision-Language-Action Models are Natural Continual Learners with Reinforcement Learning

arXiv:2603. 11653v2 Announce Type: replace Abstract: Continual Reinforcement Learning (CRL) for Vision-Language-Action (VLA) models is a promising direction toward self-improving embodied agents that can adapt in openended, evolving environments.

By Jiaheng Hu, Jay Shim, Chen Tang, Yoonchang Sung, Bo Liu, Peter Stone, Roberto Martin-Martin
arXiv AI
Oct 16

CRAFT: Forgetting-Aware Intervention-Based Adaptation for Continual Learning

CRAFT is a continual learning framework that freezes model weights and learns low‑rank interventions on hidden representations to mitigate catastrophic forgetting. Tasks that are similar share an intervention, determined by how they change the model’s output distribution, and a KL penalty keeps new tasks close to existing behavior. The approach also allows task identification at test time without explicit labels and shows improved performance and reduced forgetting across benchmarks compared to LoRA‑based methods.

By Md Anwar Hossen, Fatema Siddika, Juan Pablo Munoz, Tanya Roosta, Ali Jannesari